Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py (New Secure & Robust Version - Gradio FIX)
+# app.py (New Secure & Robust Version - with Tabs)
 import gradio as gr
 import httpx
 import os
@@ -10,6 +10,8 @@ BLAXEL_BASE_URL = os.getenv("BLAXEL_BACKEND_URL")
 BLAXEL_API_KEY = os.getenv("BLAXEL_API_KEY") 
 
 # --- Backend Client ---
+# This function does not need to change.
+# Its dictionary-based yields will work perfectly with the new tabs.
 def call_blaxel_backend(
     user_problem: str, 
     google_key: str, 
@@ -17,7 +19,6 @@ def call_blaxel_backend(
     sambanova_key: str
 ):
     
-    # --- 1. Validate Keys (FIX: All yields are now dictionaries) ---
     if not BLAXEL_BASE_URL or not BLAXEL_API_KEY:
         yield {status_output: "Configuration Error: The app's own BLAXEL secrets are not set in Hugging Face."}
         return
@@ -26,7 +27,6 @@ def call_blaxel_backend(
         yield {status_output: "Input Error: Please enter your Google API Key. It is required for the 'Judge' agent."}
         return
 
-    # --- 2. Set up connection details ---
     full_endpoint_url = f"{BLAXEL_BASE_URL.rstrip('/')}/solve_problem"
     
     headers = {
@@ -34,7 +34,6 @@ def call_blaxel_backend(
         "Content-Type": "application/json"
     }
     
-    # --- 3. Securely package the user's keys ---
     payload = {
         "problem": user_problem,
         "keys": {
@@ -44,16 +43,13 @@ def call_blaxel_backend(
         }
     }
 
-    # 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
         with httpx.stream("POST", full_endpoint_url, json=payload, headers=headers, timeout=300.0) as response:
             
             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 
 
@@ -64,10 +60,8 @@ def call_blaxel_backend(
                     final_json = json.loads(chunk.replace("FINAL:", ""))
                 else:
                     full_log += chunk + "\n"
-                    # 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:
                 yield {
                     status_output: full_log + "\nDone!",
@@ -75,7 +69,6 @@ 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:
@@ -83,27 +76,26 @@ def call_blaxel_backend(
     except Exception as e:
         yield {status_output: f"An unknown error occurred: {str(e)}"}
 
-# --- Gradio UI (Now with Optional Keys) ---
-with gr.Blocks() as demo:
+# --- Gradio UI (New Tabbed Layout) ---
+with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="neutral")) as demo:
     gr.Markdown("# MudabbirAI: The Self-Calibrating Strategic Selector")
     gr.Markdown("Based on the research by Youssef Hariri (Papers 1 & 2)")
 
-    with gr.Row():
-        with gr.Column(scale=2):
-            problem_input = gr.Textbox(
-                label="Enter Your 'Wicked Problem'",
-                placeholder="e.g., 'Develop a new market entry strategy for Southeast Asia...'",
-                lines=5
-            )
-            
-            gr.Markdown("### Sponsor API Keys")
+    # --- Section 1: Inputs ---
+    with gr.Group():
+        problem_input = gr.Textbox(
+            label="1. Enter Your 'Wicked Problem'",
+            placeholder="e.g., 'Develop a new market entry strategy for Southeast Asia...'",
+            lines=5
+        )
+        
+        # Use an Accordion to hide the keys, matching the "simple on outside" philosophy
+        with gr.Accordion("2. Sponsor API Keys (Expand to Enter)", open=False):
             gr.Markdown("Your keys are sent securely, used only for this request, and never stored.")
-            
             google_key_input = gr.Textbox(
                 label="Google API Key (Required)", type="password",
                 placeholder="Required for 'Judge' and 'default' agent"
             )
-            
             with gr.Row():
                 anthropic_key_input = gr.Textbox(
                     label="Anthropic API Key (Optional)", type="password",
@@ -113,18 +105,23 @@ with gr.Blocks() as demo:
                     label="SambaNova API Key (Optional)", type="password",
                     placeholder="Enter to enable SambaNova models"
                 )
-            
-            submit_button = gr.Button("Deploy Agent", variant="primary")
+    
+    submit_button = gr.Button("Deploy Agent", variant="primary")
 
-        with gr.Column(scale=1):
+    # --- Section 2 & 3: Outputs (Separated by Tabs) ---
+    with gr.Tabs():
+        with gr.TabItem("Agent's Internal Monologue", id="log_tab"):
             status_output = gr.Textbox(
-                label="Agent's Internal Monologue (Live Log)",
+                label="Live Log",
                 lines=20,
-                interactive=False
+                interactive=False,
+                autoscroll=True
             )
-            final_text_output = gr.Markdown(label="Final Briefing")
+        
+        with gr.TabItem("Final Briefing", id="result_tab"):
+            final_text_output = gr.Markdown(label="Final Polished Briefing")
             final_audio_output = gr.Audio(label="Audio Presentation")
-            
+        
     all_inputs = [
         problem_input,
         google_key_input,
@@ -132,7 +129,6 @@ with gr.Blocks() as demo:
         sambanova_key_input
     ]
     
-    # FIX: Define outputs as a stable LIST, not an unordered set
     all_outputs = [
         status_output,
         final_text_output,
@@ -142,7 +138,7 @@ with gr.Blocks() as demo:
     submit_button.click(
         fn=call_blaxel_backend,
         inputs=all_inputs,
-        outputs=all_outputs # Pass the list
+        outputs=all_outputs
     )
 
 demo.launch()