Update app.py

app.py

@@ -2277,7 +2277,7 @@ def create_app():
                             placeholder="Enter Poe API key",
                             label="Poe API Key",
                             type="password",
-                            value=os.environ.get("POE_API_KEY", "")
+                            value=POE_API_KEY if POE_API_KEY else ""
                         )
                         
                         openai_api_key = gr.Textbox(
@@ -2611,7 +2611,7 @@ def create_app():
                 googleai_model: gr.update(visible=(provider == "GoogleAI"))
             }
             
-        def update_context_for_provider(provider, openrouter_model, openai_model, hf_model, groq_model, cohere_model, together_model, anthropic_model, googleai_model):
+        def update_context_for_provider(provider, openrouter_model, openai_model, hf_model, groq_model, cohere_model, together_model, anthropic_model, poe_model, googleai_model):
             """Update context display based on selected provider and model"""
             if provider == "OpenRouter":
                 return update_context_display(provider, openrouter_model)
@@ -2627,6 +2627,8 @@ def create_app():
                 return update_context_display(provider, together_model)
             elif provider == "Anthropic":
                 return update_context_display(provider, anthropic_model)
+            elif provider == "Poe":
+                return update_context_display(provider, poe_model)
             elif provider == "GoogleAI":
                 return update_context_display(provider, googleai_model)
             return "Unknown"
@@ -2707,6 +2709,19 @@ def create_app():
             else:
                 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())
+            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)
+                
+            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)
                 
         def search_groq_models(search_term):
             """Filter Groq models based on search term"""
@@ -2837,14 +2852,14 @@ def create_app():
         ).then(
             fn=lambda provider, model: update_vision_indicator(
                 provider, 
-                get_current_model(provider, model, None, None, None, None, None, None, None)
+                get_current_model(provider, model, None, None, None, None, None, None, None, None)
             ),
             inputs=[provider_choice, openrouter_model],
             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)
+                get_current_model(provider, model, None, None, None, None, None, None, None, None)
             ),
             inputs=[provider_choice, openrouter_model],
             outputs=image_upload_container
@@ -2860,12 +2875,13 @@ def create_app():
                 search_cohere_models(search) if provider == "Cohere" else gr.update(),
                 search_together_models(search) if provider == "Together" else gr.update(),
                 search_anthropic_models(search) if provider == "Anthropic" else gr.update(),
+                search_poe_models(search) if provider == "Poe" 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, 
-                cohere_model, together_model, anthropic_model, googleai_model
+                cohere_model, together_model, anthropic_model, poe_model, googleai_model
             ]
         )
         
@@ -2887,6 +2903,25 @@ def create_app():
             inputs=openrouter_model,
             outputs=image_upload_container
         )
+
+        # Event handler for Poe model change
+        poe_model.change(
+            fn=lambda model: update_context_display("Poe", model),
+            inputs=poe_model,
+            outputs=context_display
+        ).then(
+            fn=lambda model: update_model_info("Poe", model),
+            inputs=poe_model,
+            outputs=model_info_display
+        ).then(
+            fn=lambda model: update_vision_indicator("Poe", model),
+            inputs=poe_model,
+            outputs=is_vision_indicator
+        ).then(
+            fn=lambda model: update_image_upload_visibility("Poe", model),
+            inputs=poe_model,
+            outputs=image_upload_container
+        )
         
         openai_model.change(
             fn=lambda model: update_context_display("OpenAI", model),