Update app.py

app.py

@@ -203,16 +203,26 @@ def update_model_info(model_name):
             """
     return "<p>Model information not available</p>"
 
-def update_category_models(category):
-    """Update model list based on selected category"""
+def update_category_models_ui(category):
+    """Completely regenerate the models dropdown based on selected category"""
     for cat in MODELS:
         if cat["category"] == category:
             model_names = [model[0] for model in cat["models"]]
             if model_names:
-                # Return the list of choices first, then the default value
-                return model_names, model_names[0]
-            return [], None
-    return [], None
+                # Return a completely new dropdown component
+                return gr.Dropdown(
+                    choices=model_names,
+                    value=model_names[0],
+                    label="Models in Category",
+                    allow_custom_value=True
+                )
+    # Return empty dropdown if no models found
+    return gr.Dropdown(
+        choices=[],
+        value=None,
+        label="Models in Category",
+        allow_custom_value=True
+    )
 
 def encode_image_to_base64(image_path):
     """Encode an image file to base64 string"""
@@ -718,12 +728,14 @@ def create_app():
                             value=MODELS[0]["category"]
                         )
                         
-                        category_models = gr.Dropdown(
-                            [model[0] for model in MODELS[0]["models"]],
-                            label="Models in Category",
-                            value=MODELS[0]["models"][0][0],
-                            allow_custom_value=True
-                        )
+                        # Create a container for the category models dropdown
+                        with gr.Column(visible=True, elem_id="category-models-container") as category_models_container:
+                            category_models = gr.Dropdown(
+                                [model[0] for model in MODELS[0]["models"]],
+                                label="Models in Category",
+                                value=MODELS[0]["models"][0][0],
+                                allow_custom_value=True
+                            )
                 
                 with gr.Accordion("Generation Parameters", open=False):
                     with gr.Group(elem_classes="parameter-grid"):
@@ -917,10 +929,9 @@ def create_app():
         
         # Update model list when category changes
         model_categories.change(
-            fn=update_category_models,
+            fn=update_category_models_ui,
             inputs=model_categories,
-            # The first output updates the choices, the second updates the value
-            outputs=[category_models, category_models]
+            outputs=category_models_container
         )
         
         # Update main model choice when category model is selected