From ac388b0f4e92b12bada0790b551d343c9862b275 Mon Sep 17 00:00:00 2001
From: Karl <karl@k-world.me.uk>
Date: Fri, 6 Jun 2025 13:10:32 +0100
Subject: [PATCH] Add grouping to model list

---
 ai_frame_image_server.py    |  3 +--
 templates/create_image.html | 18 +++++++++++++-----
 2 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/ai_frame_image_server.py b/ai_frame_image_server.py
index 79bea8e..48d3dc9 100644
--- a/ai_frame_image_server.py
+++ b/ai_frame_image_server.py
@@ -98,7 +98,7 @@ def cancel_job() -> None:
 def create():
     if request.method == "POST":
         prompt = request.form.get("prompt")
-        model = request.form.get("model", "Random")
+        model = request.form.get("model") or "Random"
 
         if not prompt:
             prompt = create_prompt_on_openwebui(user_config["comfyui"]["prompt"])
@@ -130,7 +130,6 @@ def create_image_endpoint() -> str:
     """
 
     models = load_models_from_config()
-    models.insert(0, "Random")	
 
     return render_template(
         "create_image.html", models=models
diff --git a/templates/create_image.html b/templates/create_image.html
index 30dd5f7..4023a68 100644
--- a/templates/create_image.html
+++ b/templates/create_image.html
@@ -128,12 +128,20 @@
 
         <button onclick="randomPrompt()">Random Prompt</button>
 
-
-        <!-- new model selector -->
         <select id="model-select">
-            {% for m in models %}
-            <option value="{{ m }}">{{ m.rsplit('.', 1)[0] }}</option>
-            {% endfor %}
+            <option value="" selected>Random</option>
+            <!-- Group: FLUX -->
+            <optgroup label="FLUX">
+                {% for m in models if 'flux' in m|lower %}
+                <option value="{{ m }}">{{ m.rsplit('.', 1)[0] }}</option>
+                {% endfor %}
+            </optgroup>
+            <!-- Group: SDXL -->
+            <optgroup label="SDXL">
+                {% for m in models if 'flux' not in m|lower %}
+                <option value="{{ m }}">{{ m.rsplit('.', 1)[0] }}</option>
+                {% endfor %}
+            </optgroup>
         </select>
 
     </div>