Update app.py

app.py

@@ -2653,19 +2653,19 @@ def create_app():
                 return update_model_info(provider, googleai_model)
             return "<p>Model information not available</p>"
             
-        def update_vision_indicator(provider, model_choice):
+        def update_vision_indicator(provider, model_choice=None):
             """Update the vision capability indicator"""
-            # Safety check for None model
+            # Simplified - don't call get_current_model since it causes issues
             if model_choice is None:
-                return False
+                # Just check if the provider generally supports vision
+                return provider in VISION_MODELS and len(VISION_MODELS[provider]) > 0
+            
             return is_vision_model(provider, model_choice)
             
-        def update_image_upload_visibility(provider, model_choice):
+        def update_image_upload_visibility(provider, model_choice=None):
             """Show/hide image upload based on model vision capabilities"""
-            # Safety check for None model
-            if model_choice is None:
-                return gr.update(visible=False)
-            is_vision = is_vision_model(provider, model_choice)
+            # Simplified
+            is_vision = update_vision_indicator(provider, model_choice)
             return gr.update(visible=is_vision)
         
         # Search model function
@@ -2710,19 +2710,26 @@ def create_app():
                 default_model = "mistralai/Mistral-7B-Instruct-v0.3" if "mistralai/Mistral-7B-Instruct-v0.3" in all_models else all_models[0] if all_models else None
                 return gr.update(choices=all_models, value=default_model)
             
-        def search_poe_models(search_term):
-            """Filter Poe models based on search term"""
-            all_models = list(POE_MODELS.keys())
+        def search_models_generic(search_term, model_dict, default_model=None):
+            """Generic model search function to reduce code duplication"""
+            all_models = list(model_dict.keys())
+            if not all_models:
+                return gr.update(choices=[], value=None)
+                
             if not search_term:
-                return gr.update(choices=all_models, value="chinchilla" if "chinchilla" in all_models else all_models[0] if all_models else None)
+                return gr.update(choices=all_models, value=default_model if default_model in all_models else all_models[0])
                 
             filtered_models = [model for model in all_models if search_term.lower() in model.lower()]
             
             if filtered_models:
                 return gr.update(choices=filtered_models, value=filtered_models[0])
             else:
-                return gr.update(choices=all_models, value="chinchilla" if "chinchilla" in all_models else all_models[0] if all_models else None)
-                
+                return gr.update(choices=all_models, value=default_model if default_model in all_models else all_models[0])
+
+        def search_poe_models(search_term):
+            """Filter Poe models based on search term"""
+            return search_models_generic(search_term, POE_MODELS, "chinchilla")    
+        
         def search_groq_models(search_term):
             """Filter Groq models based on search term"""
             all_models = list(GROQ_MODELS.keys())
@@ -2839,29 +2846,48 @@ def create_app():
                 cohere_model,
                 together_model,
                 anthropic_model,
+                poe_model,
                 googleai_model
             ]
         ).then(
             fn=update_context_for_provider,
-            inputs=[provider_choice, openrouter_model, openai_model, hf_model, groq_model, cohere_model, together_model, anthropic_model, googleai_model],
+            inputs=[
+                provider_choice, 
+                openrouter_model, 
+                openai_model, 
+                hf_model, 
+                groq_model, 
+                cohere_model, 
+                together_model, 
+                anthropic_model, 
+                poe_model, 
+                googleai_model
+            ],
             outputs=context_display
         ).then(
             fn=update_model_info_for_provider,
-            inputs=[provider_choice, openrouter_model, openai_model, hf_model, groq_model, cohere_model, together_model, anthropic_model, googleai_model],
+            inputs=[
+                provider_choice, 
+                openrouter_model, 
+                openai_model, 
+                hf_model, 
+                groq_model, 
+                cohere_model, 
+                together_model, 
+                anthropic_model, 
+                poe_model, 
+                googleai_model
+            ],
             outputs=model_info_display
         ).then(
-            fn=lambda provider, model: update_vision_indicator(
-                provider, 
-                get_current_model(provider, model, None, None, None, None, None, None, None, None)
-            ),
-            inputs=[provider_choice, openrouter_model],
+            # Fix this with correct number of args using a simpler approach
+            fn=lambda provider: update_vision_indicator(provider, None),
+            inputs=provider_choice,
             outputs=is_vision_indicator
         ).then(
-            fn=lambda provider, model: update_image_upload_visibility(
-                provider, 
-                get_current_model(provider, model, None, None, None, None, None, None, None, None)
-            ),
-            inputs=[provider_choice, openrouter_model],
+            # Same here
+            fn=lambda provider: update_image_upload_visibility(provider, None),
+            inputs=provider_choice,
             outputs=image_upload_container
         )
         
@@ -3136,17 +3162,27 @@ def create_app():
                 transforms=transforms,
                 api_key_override=api_key_override
             )
+
+        def clean_message(message):
+            """Clean the message from style tags"""
+            if isinstance(message, str):
+                import re
+                # Remove style tags
+                message = re.sub(r'<userStyle>.*?</userStyle>', '', message)
+            return message
         
         # Submit button click event
         submit_btn.click(
-            fn=submit_message,
+            fn=lambda *args: submit_message(clean_message(args[0]), *args[1:]),
             inputs=[
                 message, chatbot, provider_choice, 
-                openrouter_model, openai_model, hf_model, groq_model, cohere_model, together_model, anthropic_model, poe_model, googleai_model,
+                openrouter_model, openai_model, hf_model, groq_model, cohere_model, 
+                together_model, anthropic_model, poe_model, googleai_model,
                 temperature, max_tokens, top_p, frequency_penalty, presence_penalty, repetition_penalty, 
                 top_k, min_p, seed, top_a, stream_output, response_format,
                 images, documents, reasoning_effort, system_message, transforms,
-                openrouter_api_key, openai_api_key, hf_api_key, groq_api_key, cohere_api_key, together_api_key, anthropic_api_key, poe_api_key, googleai_api_key
+                openrouter_api_key, openai_api_key, hf_api_key, groq_api_key, cohere_api_key, 
+                together_api_key, anthropic_api_key, poe_api_key, googleai_api_key
             ],
             outputs=chatbot,
             show_progress="minimal",
@@ -3161,11 +3197,13 @@ def create_app():
             fn=submit_message,
             inputs=[
                 message, chatbot, provider_choice, 
-                openrouter_model, openai_model, hf_model, groq_model, cohere_model, together_model, anthropic_model, poe_model, googleai_model,
+                openrouter_model, openai_model, hf_model, groq_model, cohere_model, 
+                together_model, anthropic_model, poe_model, googleai_model,
                 temperature, max_tokens, top_p, frequency_penalty, presence_penalty, repetition_penalty, 
                 top_k, min_p, seed, top_a, stream_output, response_format,
                 images, documents, reasoning_effort, system_message, transforms,
-                openrouter_api_key, openai_api_key, hf_api_key, groq_api_key, cohere_api_key, together_api_key, anthropic_api_key, poe_api_key, googleai_api_key
+                openrouter_api_key, openai_api_key, hf_api_key, groq_api_key, cohere_api_key, 
+                together_api_key, anthropic_api_key, poe_api_key, googleai_api_key
             ],
             outputs=chatbot,
             show_progress="minimal",