feat: add functions to detect available API keys and set default provider

app.py

@@ -394,6 +394,39 @@ def get_default_model_for_provider(provider: str) -> str:
     )
 
 
+def get_available_providers():
+    """Detect which API keys are available in environment variables."""
+    env_vars = {
+        "Anthropic": "ANTHROPIC_API_KEY",
+        "OpenAI": "OPENAI_API_KEY",
+        "Hugging Face": "HUGGINGFACE_API_KEY",
+        "SambaNova": "SAMBANOVA_API_KEY",
+        "Mistral": "MISTRAL_API_KEY",
+    }
+
+    available_providers = []
+    for provider, env_var in env_vars.items():
+        if os.getenv(env_var):
+            available_providers.append(provider)
+
+    return available_providers
+
+
+def get_default_provider():
+    """Get the default provider based on available API keys."""
+    available = get_available_providers()
+
+    # Priority order for default selection
+    priority_order = ["Anthropic", "OpenAI", "Mistral", "SambaNova", "Hugging Face"]
+
+    for provider in priority_order:
+        if provider in available:
+            return provider
+
+    # If no keys are available, default to Anthropic
+    return "Anthropic"
+
+
 def save_api_key(provider, api_key):
     """Save API key to environment variable and update model accordingly."""
     if not api_key.strip():
@@ -428,8 +461,11 @@ def save_api_key(provider, api_key):
         return f"❌ Unknown provider: {provider}"
 
 
-def get_api_key_status(selected_llm_provider="Anthropic"):
-    """Get the status of Hugging Face and selected LLM provider API keys."""
+def get_api_key_status(selected_llm_provider=None):
+    """Get the status of all API keys, highlighting the selected provider."""
+    if selected_llm_provider is None:
+        selected_llm_provider = get_default_provider()
+
     env_vars = {
         "Hugging Face": "HUGGINGFACE_API_KEY",
         "Anthropic": "ANTHROPIC_API_KEY",
@@ -439,31 +475,37 @@ def get_api_key_status(selected_llm_provider="Anthropic"):
     }
 
     status = []
-
-    # Always show Hugging Face status
-    hf_env_var = env_vars["Hugging Face"]
-    if os.getenv(hf_env_var):
-        key = os.getenv(hf_env_var)
-        masked_key = f"{key[:8]}...{key[-4:]}" if len(key) > 12 else "***"
-        status.append(f"✅ Hugging Face: {masked_key}")
-    else:
-        status.append("❌ Hugging Face: Not set")
-
-    # Show selected LLM provider status
-    llm_env_var = env_vars.get(selected_llm_provider)
-    if llm_env_var and os.getenv(llm_env_var):
-        key = os.getenv(llm_env_var)
-        masked_key = f"{key[:8]}...{key[-4:]}" if len(key) > 12 else "***"
-        model = get_default_model_for_provider(selected_llm_provider)
-        status.append(f"✅ {selected_llm_provider}: {masked_key} (Model: {model})")
-    else:
-        model = get_default_model_for_provider(selected_llm_provider)
-        status.append(f"❌ {selected_llm_provider}: Not set (Would use: {model})")
+    available_providers = get_available_providers()
+
+    # Show status for all providers
+    for provider, env_var in env_vars.items():
+        if os.getenv(env_var):
+            key = os.getenv(env_var)
+            masked_key = f"{key[:8]}...{key[-4:]}" if len(key) > 12 else "***"
+            model = get_default_model_for_provider(provider)
+
+            # Highlight the selected provider
+            if provider == selected_llm_provider:
+                status.append(f"🎯 {provider}: {masked_key} (Model: {model}) [ACTIVE]")
+            else:
+                status.append(f"✅ {provider}: {masked_key} (Model: {model})")
+        else:
+            model = get_default_model_for_provider(provider)
+            if provider == selected_llm_provider:
+                status.append(f"❌ {provider}: Not set (Would use: {model}) [SELECTED]")
+            else:
+                status.append(f"❌ {provider}: Not set (Would use: {model})")
 
     # Show current active model
     current_model = os.getenv("MODEL_ID", "Qwen/Qwen2.5-Coder-32B-Instruct")
     status.append(f"🤖 Current Active Model: {current_model}")
 
+    # Show summary
+    if available_providers:
+        status.append(f"📊 Available providers: {', '.join(available_providers)}")
+    else:
+        status.append("⚠️ No API keys detected in environment")
+
     return "\n".join(status)
 
 
@@ -645,39 +687,28 @@ class GradioUI:
                             label="Current API Key Status",
                             value=get_api_key_status(),
                             interactive=False,
-                            lines=4,
-                            max_lines=4,
+                            lines=6,
+                            max_lines=8,
                         )
 
-                        # with gr.Row():
-                        #     refresh_status_btn = gr.Button(
-                        #         "🔄 Refresh Status", size="sm"
-                        #     )
-
-                        gr.Markdown("---")
-
-                        # Hugging Face Token
-                        with gr.Row():
-                            hf_token = gr.Textbox(
-                                label="Hugging Face Token",
-                                placeholder="hf_...",
-                                type="password",
-                                scale=4,
-                            )
-                            hf_save_btn = gr.Button("Save", size="sm", scale=1)
-
                         gr.Markdown("---")
 
-                        # LLM Token with Provider Selection
+                        # LLM Token with Provider Selection (now includes Hugging Face)
                         with gr.Row():
                             llm_provider = gr.Dropdown(
                                 label="LLM Provider",
-                                choices=["Anthropic", "OpenAI", "Mistral", "SambaNova"],
-                                value="Anthropic",
+                                choices=[
+                                    "Anthropic",
+                                    "OpenAI",
+                                    "Mistral",
+                                    "SambaNova",
+                                    "Hugging Face",
+                                ],
+                                value=get_default_provider(),
                                 scale=1,
                             )
                             llm_token = gr.Textbox(
-                                label="LLM Token",
+                                label="API Key",
                                 placeholder="Enter your API key...",
                                 type="password",
                                 scale=3,
@@ -695,13 +726,11 @@ class GradioUI:
             file_uploads_log = gr.State([])
 
             # Set up event handlers for API key saving
-            def save_and_update_status(
-                provider, api_key, selected_llm_provider="Anthropic", session_state=None
-            ):
+            def save_and_update_status(provider, api_key, session_state=None):
                 message = save_api_key(provider, api_key)
-                status = get_api_key_status(selected_llm_provider)
+                status = get_api_key_status(provider)
 
-                # If this is an LLM provider (not Hugging Face), recreate the agent
+                # For non-Hugging Face providers, recreate the agent
                 if provider != "Hugging Face" and session_state is not None:
                     agent_message = self.recreate_agent_with_new_model(
                         session_state, provider
@@ -711,28 +740,14 @@ class GradioUI:
 
                 return message, status, ""  # Clear the input field
 
-            hf_save_btn.click(
-                lambda key, llm_prov, sess_state: save_and_update_status(
-                    "Hugging Face", key, llm_prov, sess_state
-                ),
-                inputs=[hf_token, llm_provider, session_state],
-                outputs=[api_message, api_status, hf_token],
-            ).then(lambda: gr.Textbox(visible=True), outputs=[api_message])
-
             llm_save_btn.click(
                 lambda provider, key, sess_state: save_and_update_status(
-                    provider, key, provider, sess_state
+                    provider, key, sess_state
                 ),
                 inputs=[llm_provider, llm_token, session_state],
                 outputs=[api_message, api_status, llm_token],
             ).then(lambda: gr.Textbox(visible=True), outputs=[api_message])
 
-            # refresh_status_btn.click(
-            #     fn=lambda llm_prov: get_api_key_status(llm_prov),
-            #     inputs=[llm_provider],
-            #     outputs=[api_status],
-            # )
-
             # Update status when LLM provider dropdown changes
             llm_provider.change(
                 fn=get_api_key_status, inputs=[llm_provider], outputs=[api_status]