Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py (New Secure & Robust Version)
+# app.py (New Secure & Robust Version - Gradio FIX)
 import gradio as gr
 import httpx
 import os
@@ -17,14 +17,13 @@ def call_blaxel_backend(
     sambanova_key: str
 ):
     
-    # --- 1. Validate Keys ---
+    # --- 1. Validate Keys (FIX: All yields are now dictionaries) ---
     if not BLAXEL_BASE_URL or not BLAXEL_API_KEY:
-        yield "Configuration Error: The app's own BLAXEL secrets are not set."
+        yield {status_output: "Configuration Error: The app's own BLAXEL secrets are not set in Hugging Face."}
         return
     
-    # We require at least the Google key
     if not google_key:
-        yield {status_output: "Input Error: Please enter your Google API Key to continue. It is required for the 'Judge' agent."}
+        yield {status_output: "Input Error: Please enter your Google API Key. It is required for the 'Judge' agent."}
         return
 
     # --- 2. Set up connection details ---
@@ -40,12 +39,13 @@ def call_blaxel_backend(
         "problem": user_problem,
         "keys": {
             "google": google_key,
-            "anthropic": anthropic_key or None,  # Send None if empty
-            "sambanova": sambanova_key or None, # Send None if empty
+            "anthropic": anthropic_key or None,
+            "sambanova": sambanova_key or None,
         }
     }
 
-    yield f"Connecting to MudabbirAI (at {full_endpoint_url})..."
+    # FIX: Yield a dictionary
+    yield {status_output: f"Connecting to MudabbirAI (at {full_endpoint_url})..."}
     
     try:
         # Set a long timeout for all the LLM calls
@@ -53,6 +53,7 @@ def call_blaxel_backend(
             
             if response.status_code != 200:
                 error_details = response.read().decode()
+                # FIX: Yield a dictionary
                 yield {status_output: f"HTTP Error: Received status code {response.status_code} from Blaxel.\nDetails: {error_details}"}
                 return 
 
@@ -60,12 +61,11 @@ def call_blaxel_backend(
             full_log = ""
             for chunk in response.iter_text():
                 if chunk.startswith("FINAL:"):
-                    # This is our final package, don't yield it
                     final_json = json.loads(chunk.replace("FINAL:", ""))
                 else:
-                    # This is a live log update
                     full_log += chunk + "\n"
-                    yield {status_output: full_log} # Update log
+                    # FIX: Yield a dictionary to update just the log
+                    yield {status_output: full_log} 
 
             # --- 4. Return final results to all UI components ---
             if final_json:
@@ -75,6 +75,7 @@ def call_blaxel_backend(
                     final_audio_output: final_json.get("audio")
                 }
 
+    # FIX: All except blocks now yield dictionaries
     except httpx.ConnectError as e:
         yield {status_output: f"Connection Error: Could not connect to Blaxel.\nDetails: {str(e)}"}
     except httpx.ReadTimeout:
@@ -118,7 +119,7 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             status_output = gr.Textbox(
                 label="Agent's Internal Monologue (Live Log)",
-                lines=20, # Made log even taller
+                lines=20,
                 interactive=False
             )
             final_text_output = gr.Markdown(label="Final Briefing")
@@ -131,16 +132,17 @@ with gr.Blocks() as demo:
         sambanova_key_input
     ]
     
-    all_outputs = {
+    # FIX: Define outputs as a stable LIST, not an unordered set
+    all_outputs = [
         status_output,
         final_text_output,
         final_audio_output
-    }
+    ]
 
     submit_button.click(
         fn=call_blaxel_backend,
         inputs=all_inputs,
-        outputs=list(all_outputs) # Convert set to list
+        outputs=all_outputs # Pass the list
     )
 
 demo.launch()