Update app.py

app.py

@@ -1725,7 +1725,8 @@ def create_app():
                     show_copy_button=True, 
                     show_label=False,
                     avatar_images=(None, "https://upload.wikimedia.org/wikipedia/commons/0/04/ChatGPT_logo.svg"),
-                    elem_id="chat-window"
+                    elem_id="chat-window",
+                    type="messages"  # use the new format
                 )
                 
                 with gr.Row():
@@ -2299,12 +2300,6 @@ def create_app():
                 return gr.update(choices=all_models, value=default_model)
 
             
-        def refresh_groq_models_list():
-            """Refresh the list of Groq models"""
-            global GROQ_MODELS
-            GROQ_MODELS = fetch_groq_models()
-            return gr.update(choices=list(GROQ_MODELS.keys()))
-        
         def get_current_model(provider, openrouter_model, openai_model, hf_model, groq_model, cohere_model, together_model, ovh_model, cerebras_model, googleai_model):
             """Get the currently selected model based on provider"""
             if provider == "OpenRouter":
@@ -2327,6 +2322,7 @@ def create_app():
                 return googleai_model
             return None
         
+        
         # Process uploaded images
         image_upload_btn.upload(
             fn=lambda files: files,
@@ -2338,17 +2334,17 @@ def create_app():
         provider_choice.change(
             fn=toggle_model_dropdowns,
             inputs=provider_choice,
-            outputs={
-                openrouter_model: None,
-                openai_model: None,
-                hf_model: None,
-                groq_model: None,
-                cohere_model: None,
-                together_model: None,
-                ovh_model: None,
-                cerebras_model: None,
-                googleai_model: None
-            }
+            outputs=[
+                openrouter_model,
+                openai_model,
+                hf_model,
+                groq_model,
+                cohere_model,
+                together_model,
+                ovh_model,
+                cerebras_model,
+                googleai_model
+            ]
         ).then(
             fn=update_context_for_provider,
             inputs=[provider_choice, openrouter_model, openai_model, hf_model, groq_model, cohere_model, together_model, ovh_model, cerebras_model, googleai_model],
@@ -2373,20 +2369,19 @@ def create_app():
             outputs=image_upload_container
         )
         
-        # Set up model search event - FIXED VERSION
-        # We'll now have the correct function that returns model dropdown updates
+        # Set up model search event - return model dropdown updates
         model_search.change(
-            fn=lambda provider, search: {
-                "OpenRouter": search_openrouter_models(search),
-                "OpenAI": search_openai_models(search),
-                "HuggingFace": search_hf_models(search),
-                "Groq": search_groq_models(search),
-                "Cohere": search_cohere_models(search),
-                "Together": search_together_models(search),
-                "OVH": search_ovh_models(search),
-                "Cerebras": search_cerebras_models(search),
-                "GoogleAI": search_googleai_models(search)
-            }[provider],
+            fn=lambda provider, search: [
+                search_openrouter_models(search) if provider == "OpenRouter" else gr.update(),
+                search_openai_models(search) if provider == "OpenAI" else gr.update(),
+                search_hf_models(search) if provider == "HuggingFace" else gr.update(),
+                search_groq_models(search) if provider == "Groq" else gr.update(),
+                search_cohere_models(search) if provider == "Cohere" else gr.update(),
+                search_together_models(search) if provider == "Together" else gr.update(),
+                search_ovh_models(search) if provider == "OVH" else gr.update(),
+                search_cerebras_models(search) if provider == "Cerebras" else gr.update(),
+                search_googleai_models(search) if provider == "GoogleAI" else gr.update()
+            ],
             inputs=[provider_choice, model_search],
             outputs=[
                 openrouter_model, openai_model, hf_model, groq_model, 
@@ -2556,6 +2551,28 @@ def create_app():
             inputs=googleai_model,
             outputs=image_upload_container
         )
+
+        def handle_search(provider, search_term):
+            """Handle search based on provider"""
+            if provider == "OpenRouter":
+                return search_openrouter_models(search_term)
+            elif provider == "OpenAI":
+                return search_openai_models(search_term)
+            elif provider == "HuggingFace":
+                return search_hf_models(search_term)
+            elif provider == "Groq":
+                return search_groq_models(search_term)
+            elif provider == "Cohere":
+                return search_cohere_models(search_term)
+            elif provider == "Together":
+                return search_together_models(search_term)
+            elif provider == "OVH":
+                return search_ovh_models(search_term)
+            elif provider == "Cerebras":
+                return search_cerebras_models(search_term)
+            elif provider == "GoogleAI":
+                return search_googleai_models(search_term)
+            return None
         
         # Set up submission event
         def submit_message(message, history, provider, openrouter_model, openai_model, hf_model, groq_model, cohere_model,