Update app.py

app.py

@@ -208,8 +208,10 @@ def update_category_models(category):
     for cat in MODELS:
         if cat["category"] == category:
             model_names = [model[0] for model in cat["models"]]
-            return model_names, model_names[0]
-    return [], ""
+            if model_names:
+                return gr.Dropdown.update(choices=model_names, value=model_names[0])
+            return gr.Dropdown.update(choices=[], value=None)
+    return gr.Dropdown.update(choices=[], value=None)
 
 def encode_image_to_base64(image_path):
     """Encode an image file to base64 string"""
@@ -697,7 +699,8 @@ def create_app():
                             [model[0] for model in ALL_MODELS],
                             value=ALL_MODELS[0][0],
                             label="Model",
-                            elem_id="model-choice"  # Add elem_id for debugging
+                            elem_id="model-choice",
+                            allow_custom_value=True  # allow custom values
                         )
                         context_display = gr.Textbox(
                             value=update_context_display(ALL_MODELS[0][0]),
@@ -717,7 +720,8 @@ def create_app():
                         category_models = gr.Dropdown(
                             [model[0] for model in MODELS[0]["models"]],
                             label="Models in Category",
-                            value=MODELS[0]["models"][0][0]
+                            value=MODELS[0]["models"][0][0],
+                            allow_custom_value=True  # allow custom values
                         )
                 
                 with gr.Accordion("Generation Parameters", open=False):
@@ -893,7 +897,7 @@ def create_app():
         model_search.change(
             fn=filter_models,
             inputs=model_search,
-            outputs=[model_choice, model_choice]
+            outputs=[model_choice, model_choice]  # update both values
         )
         
         # Update context display when model changes
@@ -914,12 +918,12 @@ def create_app():
         model_categories.change(
             fn=update_category_models,
             inputs=model_categories,
-            outputs=[category_models, category_models]
+            outputs=category_models
         )
         
         # Update main model choice when category model is selected
         category_models.change(
-            fn=lambda x: x,
+            fn=lambda x: x,  # Simply pass the value through
             inputs=category_models,
             outputs=model_choice
         )