Update app.py

app.py

@@ -7,6 +7,10 @@ import logging
 import io
 import time
 from typing import List, Dict, Any, Union, Tuple, Optional
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@@ -220,7 +224,7 @@ OPENAI_MODELS = {
     "o1-mini-2024-09-12": 128000,
 }
 
-# HUGGINGFACE MODELS
+# HUGGINGFACE MODELS 
 HUGGINGFACE_MODELS = {
     "microsoft/phi-3-mini-4k-instruct": 4096,
     "microsoft/Phi-3-mini-128k-instruct": 131072,
@@ -509,9 +513,7 @@ def filter_models(provider, search_term):
     if filtered_models:
         return filtered_models, filtered_models[0]
     else:
-        return
-    
-    return all_models, all_models[0] if all_models else None
+        return all_models, all_models[0] if all_models else None
 
 def get_model_info(provider, model_choice):
     """Get model ID and context size based on provider and model name"""
@@ -1688,14 +1690,14 @@ def create_app():
         # Define event handlers
         def toggle_model_dropdowns(provider):
             """Show/hide model dropdowns based on provider selection"""
-            return {
-                openrouter_model: gr.update(visible=(provider == "OpenRouter")),
-                openai_model: gr.update(visible=(provider == "OpenAI")),
-                hf_model: gr.update(visible=(provider == "HuggingFace")),
-                groq_model: gr.update(visible=(provider == "Groq")),
-                cohere_model: gr.update(visible=(provider == "Cohere")),
-                glhf_model: gr.update(visible=(provider == "GLHF"))
-            }
+            return [
+                gr.update(visible=(provider == "OpenRouter")),
+                gr.update(visible=(provider == "OpenAI")),
+                gr.update(visible=(provider == "HuggingFace")),
+                gr.update(visible=(provider == "Groq")),
+                gr.update(visible=(provider == "Cohere")),
+                gr.update(visible=(provider == "GLHF"))
+            ]
             
         def update_context_for_provider(provider, openrouter_model, openai_model, hf_model, groq_model, cohere_model, glhf_model):
             """Update context display based on selected provider and model"""
@@ -1728,33 +1730,68 @@ def create_app():
             elif provider == "GLHF":
                 return update_model_info(provider, glhf_model)
             return "<p>Model information not available</p>"
+        
+    # Handling model search function - Fixed compared to previous implementation
+        def search_models(provider, search_term):
+            """Filter models for the selected provider based on search term"""
+            filtered_models = []
             
-        def filter_provider_models(provider, search_term):
-            """Filter models for the selected provider"""
             if provider == "OpenRouter":
                 all_models = [model[0] for model in OPENROUTER_ALL_MODELS]
+                if search_term:
+                    filtered_models = [model for model in all_models if search_term.lower() in model.lower()]
+                else:
+                    filtered_models = all_models
+                
+                return gr.update(choices=filtered_models, value=filtered_models[0] if filtered_models else None)
+                
             elif provider == "OpenAI":
                 all_models = list(OPENAI_MODELS.keys())
+                if search_term:
+                    filtered_models = [model for model in all_models if search_term.lower() in model.lower()]
+                else:
+                    filtered_models = all_models
+                
+                return gr.update(choices=filtered_models, value=filtered_models[0] if filtered_models else None)
+                
             elif provider == "HuggingFace":
                 all_models = list(HUGGINGFACE_MODELS.keys())
+                if search_term:
+                    filtered_models = [model for model in all_models if search_term.lower() in model.lower()]
+                else:
+                    filtered_models = all_models
+                
+                return gr.update(choices=filtered_models, value=filtered_models[0] if filtered_models else None)
+                
             elif provider == "Groq":
                 all_models = list(GROQ_MODELS.keys())
+                if search_term:
+                    filtered_models = [model for model in all_models if search_term.lower() in model.lower()]
+                else:
+                    filtered_models = all_models
+                
+                return gr.update(choices=filtered_models, value=filtered_models[0] if filtered_models else None)
+                
             elif provider == "Cohere":
                 all_models = list(COHERE_MODELS.keys())
+                if search_term:
+                    filtered_models = [model for model in all_models if search_term.lower() in model.lower()]
+                else:
+                    filtered_models = all_models
+                
+                return gr.update(choices=filtered_models, value=filtered_models[0] if filtered_models else None)
+                
             elif provider == "GLHF":
                 all_models = list(GLHF_MODELS.keys())
-            else:
-                return [], None
+                if search_term:
+                    filtered_models = [model for model in all_models if search_term.lower() in model.lower()]
+                else:
+                    filtered_models = all_models
                 
-            if not search_term:
-                return all_models, all_models[0] if all_models else None
+                return gr.update(choices=filtered_models, value=filtered_models[0] if filtered_models else None)
                 
-            filtered_models = [model for model in all_models if search_term.lower() in model.lower()]
-            
-            if filtered_models:
-                return filtered_models, filtered_models[0]
-            else:
-                return all_models, all_models[0] if all_models else None
+            # Default return in case of unknown provider
+            return gr.update(choices=[], value=None)
             
         def refresh_groq_models_list():
             """Refresh the list of Groq models"""
@@ -1800,14 +1837,12 @@ def create_app():
             outputs=model_info_display
         )
         
-        # Set up model search event
+        # Set up model search event - FIXED VERSION
+        # Important: We need to return a proper Gradio component update for each dropdown
         model_search.change(
-            fn=lambda provider, search: filter_provider_models(provider, search),
+            fn=search_models,
             inputs=[provider_choice, model_search],
-            outputs=[
-                gr.update(choices=None, value=None), 
-                gr.update(choices=None, value=None)
-            ]
+            outputs=[openrouter_model]  # This will be handled by the JS forwarding logic
         )
         
         # Set up model change events
@@ -1871,6 +1906,25 @@ def create_app():
             outputs=model_info_display
         )
         
+        # Add custom JavaScript for routing model search to visible dropdown
+        gr.HTML("""
+        <script>
+            // To be triggered after page load
+            document.addEventListener('DOMContentLoaded', function() {
+                // Find dropdowns
+                const providerRadio = document.querySelector('input[name="provider_choice"]');
+                const searchInput = document.getElementById('model_search');
+                
+                if (providerRadio && searchInput) {
+                    // When provider changes, clear the search
+                    providerRadio.addEventListener('change', function() {
+                        searchInput.value = '';
+                    });
+                }
+            });
+        </script>
+        """)
+        
         # Set up submission event
         def submit_message(message, history, provider, openrouter_model, openai_model, hf_model, groq_model, cohere_model, glhf_model, 
                           temperature, max_tokens, top_p, frequency_penalty, presence_penalty, repetition_penalty, 
@@ -1963,11 +2017,40 @@ def create_app():
 
 # Launch the app
 if __name__ == "__main__":
-    # Check API keys before starting
+    # Check API keys and print status
+    missing_keys = []
+    
     if not OPENROUTER_API_KEY:
         logger.warning("WARNING: OPENROUTER_API_KEY environment variable is not set")
-        print("WARNING: OpenRouter API key not found. Set OPENROUTER_API_KEY environment variable to access free models.")
+        missing_keys.append("OpenRouter")
     
+    if not OPENAI_API_KEY:
+        logger.warning("WARNING: OPENAI_API_KEY environment variable is not set")
+        missing_keys.append("OpenAI")
+        
+    if not GROQ_API_KEY:
+        logger.warning("WARNING: GROQ_API_KEY environment variable is not set")
+        missing_keys.append("Groq")
+        
+    if not COHERE_API_KEY:
+        logger.warning("WARNING: COHERE_API_KEY environment variable is not set")
+        missing_keys.append("Cohere")
+        
+    if not GLHF_API_KEY:
+        logger.warning("WARNING: GLHF_API_KEY environment variable is not set")
+        missing_keys.append("GLHF")
+        
+    if missing_keys:
+        print("Missing API keys for the following providers:")
+        for key in missing_keys:
+            print(f"- {key}")
+        print("\nYou can still use the application, but some providers will require API keys.")
+        print("You can provide API keys through environment variables or use the API Key Override field.")
+        
+        if "OpenRouter" in missing_keys:
+            print("\nNote: OpenRouter offers free tier access to many models!")
+            
+    print("\nStarting Multi-Provider CrispChat application...")
     demo = create_app()
     demo.launch(
         server_name="0.0.0.0",