Update app.py

app.py

@@ -1,77 +1,61 @@
 # app.py
 import asyncio
 import gradio as gr
+from dotenv import load_dotenv
 
 from research_manager import ResearchManager
 
+load_dotenv(override=True)
 
 manager = ResearchManager()
 
 
-async def run_research(topic: str):
+def run(query: str):
     """
-    Runs the deep research pipeline and streams updates.
-    """
-    updates = []
-    async for update in manager.run(topic):
-        updates.append(update)
-        yield "\n\n".join(updates)
-
-
-def run(topic: str):
-    """
-    Gradio does not support async generators directly,
-    so we bridge async → sync safely.
+    Hugging Face Spaces requires sync functions.
+    This safely consumes your async generator.
     """
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
 
-    async_gen = run_research(topic)
-
     outputs = []
+    async_gen = manager.run(query)
+
     try:
         while True:
-            update = loop.run_until_complete(async_gen.__anext__())
-            outputs.append(update)
+            chunk = loop.run_until_complete(async_gen.__anext__())
+            outputs.append(chunk)
     except StopAsyncIteration:
         pass
     finally:
         loop.close()
 
-    return outputs[-1] if outputs else "No output."
+    # Return the final combined output
+    return "\n\n".join(outputs)
 
 
-with gr.Blocks() as demo:
-    gr.Markdown(
-        """
-        # 🔍 Deep Research & Email Agent
+with gr.Blocks(theme=gr.themes.Default(primary_hue="sky")) as ui:
+    gr.Markdown("# 🔍 Deep Research Agent")
 
-        This app:
-        - Plans web searches
-        - Performs parallel research
-        - Writes a long-form report
-        - Sends the report via email
-
-        ⚠️ Email is sent to a fixed recipient configured in the backend.
-        """
-    )
-
-    topic = gr.Textbox(
-        label="Research topic",
+    query_textbox = gr.Textbox(
+        label="What topic would you like to research?",
         placeholder="e.g. Latest AI agent frameworks in 2025",
     )
 
-    run_btn = gr.Button("Run Deep Research")
+    run_button = gr.Button("Run", variant="primary")
 
-    output = gr.Textbox(
-        label="Status / Output",
-        lines=20,
+    report = gr.Markdown(label="Output")
+
+    run_button.click(
+        fn=run,
+        inputs=query_textbox,
+        outputs=report,
     )
 
-    run_btn.click(
+    query_textbox.submit(
         fn=run,
-        inputs=topic,
-        outputs=output,
+        inputs=query_textbox,
+        outputs=report,
     )
 
-demo.launch()
+ui.launch()