From 6d1c01c9557577573df84489c08581c405e7bdc6 Mon Sep 17 00:00:00 2001
From: DepFA <35278260+dfaker@users.noreply.github.com>
Date: Thu, 22 Sep 2022 17:26:26 +0100
Subject: [PATCH] Add section splitting to settings ui

---
 modules/ui.py | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/modules/ui.py b/modules/ui.py
index 8925fbcb..0d9f4c09 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -9,6 +9,7 @@ import random
 import sys
 import time
 import traceback
+import itertools
 
 import numpy as np
 import torch
@@ -842,14 +843,22 @@ def create_ui(txt2img, img2img, run_extras, run_pnginfo):
         settings_submit = gr.Button(value="Apply settings", variant='primary')
         result = gr.HTML()
 
-        with gr.Row(elem_id="settings").style(equal_height=False):
-            for colno in range(settings_cols):
-                with gr.Column(variant='panel'):
-                    for rowno in range(items_per_col):
-                        index = rowno + colno * items_per_col
+        sortedKeys  = sorted([(opts.data_labels[key].section,key) for key in keys],key=lambda x:x[0])
+        groupedKeys = itertools.groupby(sortedKeys,key=lambda x:x[0])
 
-                        if index < len(keys):
-                            components.append(create_setting_component(keys[index]))
+        for (sectionNumber,sectionName),sectionGroup in groupedKeys:
+          with gr.Row(elem_id="settings_header_{}".format(sectionNumber)).style(equal_height=False):
+            gr.HTML(elem_id="settings_header_text_{}".format(sectionNumber), value='<h1 class="gr-button-lg">{}</h1>'.format(sectionName))
+
+          with gr.Row(elem_id="settings_{}".format(sectionNumber)).style(equal_height=False):
+              columnLookup = {}
+              for colNum,element in zip(itertools.cycle(range(settings_cols)),sectionGroup):
+                columnLookup.setdefault(colNum,[]).append(element)
+
+              for colno,elements in sorted(columnLookup.items()):
+                  with gr.Column(variant='panel'):
+                      for _,keyElement in elements:
+                        components.append(create_setting_component(keyElement))
 
         settings_submit.click(
             fn=run_settings,