Update app.py

app.py

@@ -1,98 +1,52 @@
+# app.py (Smoke Test - Now with Secrets)
 import gradio as gr
 import httpx
-import os
+import os  # <-- 1. Import the os module
 import json
 
-# --- Read Secrets ---
-BLAXEL_BACKEND_URL = os.environ.get("BLAXEL_BACKEND_URL", "")
+# --- Config ---
+# !!! 2. Read the URL from Hugging Face secrets (environment variables)
+BLAXEL_BACKEND_URL = os.getenv("BLAXEL_BACKEND_URL")
 
-def call_blaxel_backend(user_problem, run_count, do_calibrate):
-    """
-    This is the NON-STREAMING function, calling the correct public URL.
-    """
+# --- Backend Client ---
+def call_blaxel_backend(user_problem):
+    
+    # 3. Add a check to see if the secret was loaded
     if not BLAXEL_BACKEND_URL:
-        return "ERROR: 'BLAXEL_BACKEND_URL' is not set...", None, None
-
-    yield "[1/2] 🧠 Agent activated. Calling public trigger URL...", gr.skip(), gr.skip()
-
-    request_data = {
-        "problem": user_problem,
-        "run_count": run_count,
-        "do_calibrate": do_calibrate
-    }
+        yield "Configuration Error: BLAXEL_BACKEND_URL secret is not set in Hugging Face."
+        yield "Please add your Blaxel URL to the 'Secrets' section of your Space settings."
+        return # Stop execution if the URL isn't found
 
+    # 1. Show a "thinking" message
+    yield f"Attempting to connect to MudabbirAI (at {BLAXEL_BACKEND_URL})..."
+    
+    # 2. Make a streaming API call
     try:
-        # We call the public URL directly, no auth headers needed.
-        response = httpx.post(
-            BLAXEL_BACKEND_URL,
-            json=request_data, 
-            timeout=30
-        )
-        
-        response.raise_for_status() 
-        final_data = response.json()
-        status_log = "\n".join(final_data.get("status_log", []))
-        
-        yield (
-            status_log, 
-            final_data.get("text", "Error: No text received."), 
-            final_data.get("audio", None)
-        )
-
+        with httpx.stream("POST", BLAXEL_BACKEND_URL, json={"problem": user_problem}, timeout=60) as response:
+            response.raise_for_status() # Raise an exception for bad statuses (4xx, 5xx)
+            for chunk in response.iter_text():
+                # Each chunk is a status update
+                yield chunk
+    except httpx.ConnectError as e:
+        yield f"Connection Error: Could not connect to Blaxel. Is the URL correct and the backend running?\nDetails: {str(e)}"
     except httpx.HTTPStatusError as e:
-        yield f"HTTP Error: {e.response.status_code} - {e.response.text}", None, None
-    except httpx.ConnectError:
-        yield f"ERROR: Could not connect to Blaxel backend at: {BLAXEL_BACKEND_URL}", None, None
+        yield f"HTTP Error: Received status code {e.response.status_code} from Blaxel.\nDetails: {e.response.text}"
     except httpx.ReadTimeout:
-        yield "Error: The agent backend timed out (30 seconds).", None, None
+        yield "Error: The connection timed out."
     except Exception as e:
-        yield f"An unexpected error occurred: {str(e)}", None, None
-
-# --- Gradio UI (No changes needed) ---
-with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
-    gr.Markdown("# MudabbirAI")
-    gr.Markdown("Based on the research by Youssef Hariri (Papers 1 & 2)")
+        yield f"An unknown error occurred: {str(e)}"
+
+# --- Gradio UI ---
+with gr.Blocks() as demo:
+    gr.Markdown("# MudabbirAI - Connection Test")
+    problem_input = gr.Textbox(label="Enter any text to test")
+    submit_button = gr.Button("Test Connection")
+    status_output = gr.Textbox(label="Live Log from Blaxel", interactive=False)
     
-    with gr.Tabs():
-        with gr.TabItem("🚀 Deploy Agent"):
-            gr.Markdown("## 1. Enter Your 'Wicked Problem'")
-            problem_input = gr.Textbox(
-                label="Problem Brief",
-                placeholder="e.g., 'Draft a novel, feasible, and culturally sensitive marketing plan for our new app in Japan...'",
-                lines=5
-            )
-            
-            gr.Markdown("## 2. (Optional) Select Frameworks")
-            
-            gr.Markdown("## 3. (Optional) Advanced Settings")
-            with gr.Accordion("Advanced Run Options", open=False):
-                run_count = gr.Slider(
-                    label="Number of Runs (Leverages AI stochasticity)",
-                    minimum=1, maximum=5, step=1, value=1
-                )
-                do_calibrate = gr.Checkbox(
-                    label="Run Live Ensemble Calibration (Slower, highest accuracy)",
-                    value=False
-                )
-
-            gr.Markdown("## 4. Execute")
-            submit_button = gr.Button("Deploy Agent", variant="primary")
-            
-        with gr.TabItem("📊 Live Log & Final Output"):
-            gr.Markdown("## Agent's Internal Monologue (Live Log)")
-            status_output = gr.Textbox(label="Agent Status Log", interactive=False, lines=10)
-
-            gr.Markdown("## Final Briefing")
-            final_text_output = gr.Markdown()
-            
-            gr.Markdown("## Audio Presentation")
-            final_audio_output = gr.Audio(label="Audio Presentation", type="filepath")
-
     submit_button.click(
         fn=call_blaxel_backend,
-        inputs=[problem_input, run_count, do_calibrate],
-        outputs=[status_output, final_text_output, final_audio_output]
+        inputs=problem_input,
+        outputs=status_output
     )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()