Update app.py

app.py

@@ -1,59 +1,58 @@
 # app.py
 import gradio as gr
-from genesis.pipeline import research_once
-from genesis.pipeline import DEMO_QUERIES
+from genesis.pipeline import research_once, DEMO_QUERIES
 
-# --------- UI Callback ---------
 def run_pipeline_ui(query):
-    try:
-        report = research_once(query)
-        return (
-            report["summary"],
-            report["citations"],
-            report["structures"],
-            report["visual_image_url"] or "No image generated",
-            report["audio_url"] or None
-        )
-    except Exception as e:
-        return f"Error: {e}", [], [], None, None
+    if not query.strip():
+        return "Please enter a research topic.", "", None, None
+    report = research_once(query)
+    summary = report["summary"]
+    citations_md = "\n".join(f"- [{c['type']}]({c['url']})" for c in report["citations"]) or "No citations found."
+    return summary, citations_md, report["visual_image_url"], report["audio_url"]
 
-# --------- Gradio UI ---------
-with gr.Blocks(title="GENESIS-AI — Synthetic Biology Research Engine") as demo:
-    gr.Markdown("## 🧬 GENESIS-AI — Synthetic Biology Research Engine")
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="green", secondary_hue="blue"), css="""
+    .demo-btn {border-radius: 25px; padding: 8px 14px; font-weight: bold;}
+    .demo-btn:hover {background-color: #28a745 !important; color: white !important;}
+""") as demo:
+    
     gr.Markdown(
-        "Ask about synthetic biology, drug design, CRISPR, biosensors, and more."
+        """
+        # 🧬 GENESIS-AI — Synthetic Biology Research Engine
+        Ask about **CRISPR**, drug design, biosensors, metabolic engineering, and more.  
+        _Your AI co-pilot for disruptive biotech discovery._
+        """,
+        elem_id="title"
     )
-
+    
     with gr.Row():
-        query_input = gr.Textbox(
-            label="Enter your research topic",
-            placeholder="e.g., Design a CRISPR-based living therapeutic for triple-negative breast cancer",
-            lines=2
-        )
-        run_button = gr.Button("🚀 Run Research", variant="primary")
-
-    output_summary = gr.Textbox(label="Summary", lines=8)
-    output_citations = gr.JSON(label="Citations")
-    output_structures = gr.JSON(label="Molecular Structures")
-    output_image = gr.Image(label="Generated Diagram / Image")
-    output_audio = gr.Audio(label="Narration", type="filepath")
-
-    # Demo Queries Section
-    gr.Markdown("### Demo Queries")
+        query_box = gr.Textbox(label="Enter Your Research Topic", placeholder="e.g., Design a CRISPR-based living therapeutic for triple-negative breast cancer", scale=4)
+        run_btn = gr.Button("🚀 Run Research", variant="primary", scale=1)
+    
+    with gr.Row():
+        gr.Markdown("**Or try one of these expert-curated demo queries:**")
+    
     with gr.Row():
         for q in DEMO_QUERIES:
-            gr.Button(q).click(
+            gr.Button(q, elem_classes="demo-btn").click(
                 fn=run_pipeline_ui,
-                inputs=[gr.Textbox(value=q, visible=False)],
-                outputs=[output_summary, output_citations, output_structures, output_image, output_audio]
+                inputs=gr.Textbox(value=q, visible=False),
+                outputs=["summary_out", "cites_out", "img_out", "audio_out"]
             )
 
-    # Main run button
-    run_button.click(
+    with gr.Tab("📄 Summary"):
+        summary_out = gr.Textbox(label="Executive Summary", lines=15, interactive=False)
+    with gr.Tab("🔗 Citations"):
+        cites_out = gr.Markdown()
+    with gr.Tab("🖼 Diagram"):
+        img_out = gr.Image(type="filepath", label="Generated Research Diagram")
+    with gr.Tab("🔊 Narration"):
+        audio_out = gr.Audio(type="filepath", label="AI Narration")
+
+    # Main run button event
+    run_btn.click(
         fn=run_pipeline_ui,
-        inputs=[query_input],
-        outputs=[output_summary, output_citations, output_structures, output_image, output_audio]
+        inputs=query_box,
+        outputs=[summary_out, cites_out, img_out, audio_out]
     )
 
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch()