Upload folder using huggingface_hub

app.py

@@ -674,6 +674,7 @@ def run_agent_on_case(
     case: MedicalCase,
     backend: str,
     context_mode: str,
+    user_api_key: str = "",
 ) -> tuple[str, str, str]:
     """
     Run the agent on a case and return formatted markdown outputs.
@@ -685,11 +686,14 @@ def run_agent_on_case(
         model_name = "simulated"
     else:
         try:
-            client = create_client(backend)
+            kwargs = {}
+            if user_api_key and user_api_key.strip():
+                kwargs["api_key"] = user_api_key.strip()
+            client = create_client(backend, **kwargs)
         except Exception as e:
             return (
                 f"**Error creating {backend} client:** {e}\n\n"
-                "Make sure your API key is set in `.env` or environment variables. "
+                "Make sure you enter your API key above, or set it in environment variables. "
                 "Or select **simulated (no API key)** to see a demo trace.",
                 "", "",
             )
@@ -756,13 +760,13 @@ def on_demo_case_selected(case_name: str) -> tuple[str, str]:
     return "", ""
 
 
-def run_demo_case(case_name: str, backend: str, context_mode: str):
+def run_demo_case(case_name: str, backend: str, context_mode: str, user_api_key: str = ""):
     """Run agent on a selected demo case."""
     if case_name not in DEMO_CASES:
         return "Please select a demo case.", "", ""
 
     case = DEMO_CASES[case_name]["case"]()
-    return run_agent_on_case(case, backend, context_mode)
+    return run_agent_on_case(case, backend, context_mode, user_api_key)
 
 
 def run_custom_case(
@@ -772,6 +776,7 @@ def run_custom_case(
     ch3_name: str, ch3_type: str, ch3_value: str,
     uploaded_image,
     backend: str, context_mode: str,
+    user_api_key: str = "",
 ):
     """Run agent on a custom user-defined case."""
     if not scenario.strip():
@@ -784,7 +789,7 @@ def run_custom_case(
         ch3_name, ch3_type, ch3_value,
         uploaded_image,
     )
-    return run_agent_on_case(case, backend, context_mode)
+    return run_agent_on_case(case, backend, context_mode, user_api_key)
 
 
 # ============================================================
@@ -804,9 +809,10 @@ def create_app():
             elem_classes="header-text",
         )
 
-        # Build backend choices: simulation always available, real backends if keys exist
-        backend_choices = ["simulated (no API key)"] + AVAILABLE_BACKENDS
-        default_backend = AVAILABLE_BACKENDS[0] if AVAILABLE_BACKENDS else "simulated (no API key)"
+        # Build backend choices: always show real backends (user can paste their own key)
+        all_backends = ["openai", "anthropic", "together"]
+        backend_choices = ["simulated (no API key)"] + all_backends
+        default_backend = "openai"
 
         with gr.Row():
             backend = gr.Dropdown(
@@ -823,6 +829,13 @@ def create_app():
                 info="How the agent manages conversation history",
                 scale=1,
             )
+            user_api_key = gr.Textbox(
+                label="API Key (optional)",
+                placeholder="sk-... (paste your OpenAI/Anthropic key)",
+                type="password",
+                info="Enter your own API key to use a real VLM backend",
+                scale=1,
+            )
 
         with gr.Tabs():
             # ---- Tab 1: Demo Cases ----
@@ -857,7 +870,7 @@ def create_app():
 
                 run_demo_btn.click(
                     fn=run_demo_case,
-                    inputs=[case_selector, backend, context_mode],
+                    inputs=[case_selector, backend, context_mode, user_api_key],
                     outputs=[demo_steps, demo_entropy, demo_summary],
                 )
 
@@ -931,6 +944,7 @@ def create_app():
                         ch3_name, ch3_type, ch3_value,
                         custom_image,
                         backend, context_mode,
+                        user_api_key,
                     ],
                     outputs=[custom_steps, custom_entropy, custom_summary],
                 )