Update app.py

app.py

@@ -1,79 +1,59 @@
 # app.py
 import gradio as gr
-from genesis.pipeline import research_once, DEMO_QUERIES
+from genesis.pipeline import research_once
+from genesis.pipeline import DEMO_QUERIES
 
+# --------- UI Callback ---------
 def run_pipeline_ui(query):
-    """Run the full GENESIS-AI pipeline for the given query."""
     try:
-        result = research_once(query)
-
-        summary_md = f"### Research Summary\n{result['summary']}"
-        cites_md = "### Citations\n"
-        if result["citations"]:
-            for c in result["citations"]:
-                cites_md += f"- [{c['type']} {c['id'] or ''}]({c['url']})\n"
-        else:
-            cites_md += "_No citations found._"
-
-        struct_md = "### Molecular Structures\n"
-        if result["structures"]:
-            for s in result["structures"]:
-                struct_md += f"- **{s['name']}** — [View]({s['url']})\n"
-        else:
-            struct_md += "_No structures found._"
-
-        img_md = ""
-        if result["visual_image_url"]:
-            img_md = f"### Visual Diagram\n![]({result['visual_image_url']})"
-
-        audio_component = None
-        if result["audio_url"]:
-            audio_component = result["audio_url"]
-
-        return summary_md, cites_md, struct_md, img_md, audio_component
-
+        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
-
-
-with gr.Blocks(title="GENESIS-AI — Synthetic Biology Research Engine", theme="default") as demo:
-    gr.Markdown("# 🧬 GENESIS-AI — Synthetic Biology Research Engine")
-    gr.Markdown("Ask about synthetic biology, drug design, CRISPR, biosensors, and more.")
+        return f"Error: {e}", [], [], None, None
 
-    with gr.Row():
-        query_input = gr.Textbox(label="Enter your research topic", placeholder="e.g., AI-driven biosensor design for early cancer detection")
-        run_btn = gr.Button("Run Research", variant="primary")
+# --------- Gradio UI ---------
+with gr.Blocks(title="GENESIS-AI — Synthetic Biology Research Engine") as demo:
+    gr.Markdown("## 🧬 GENESIS-AI — Synthetic Biology Research Engine")
+    gr.Markdown(
+        "Ask about synthetic biology, drug design, CRISPR, biosensors, and more."
+    )
 
     with gr.Row():
-        summary_output = gr.Markdown()
+        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")
     with gr.Row():
-        cites_output = gr.Markdown()
-    with gr.Row():
-        struct_output = gr.Markdown()
-    with gr.Row():
-        img_output = gr.Markdown()
-    with gr.Row():
-        audio_output = gr.Audio()
-
-    run_btn.click(
-        run_pipeline_ui,
-        inputs=[query_input],
-        outputs=[summary_output, cites_output, struct_output, img_output, audio_output]
-    )
-
-    gr.Markdown("## Demo Queries")
-    demo_gallery = gr.Dataset(
-        components=[gr.Textbox()],
-        samples=[[q] for q in DEMO_QUERIES],
-        type="values",
-        label="Click a demo query to run it"
-    )
-
-    demo_gallery.click(
-        run_pipeline_ui,
+        for q in DEMO_QUERIES:
+            gr.Button(q).click(
+                fn=run_pipeline_ui,
+                inputs=[gr.Textbox(value=q, visible=False)],
+                outputs=[output_summary, output_citations, output_structures, output_image, output_audio]
+            )
+
+    # Main run button
+    run_button.click(
+        fn=run_pipeline_ui,
         inputs=[query_input],
-        outputs=[summary_output, cites_output, struct_output, img_output, audio_output]
+        outputs=[output_summary, output_citations, output_structures, output_image, output_audio]
     )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
+    demo.launch(server_name="0.0.0.0", server_port=7860)