import gradio as gr
from genesis.pipeline import research_once, DEMO_QUERIES

def run_pipeline_ui(query):
    """Run the full synthetic biology research pipeline."""
    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"]

with gr.Blocks(
    theme=gr.themes.Soft(primary_hue="green", secondary_hue="blue"),
    css="""
        #title {text-align: center;}
        .demo-btn {border-radius: 25px; padding: 8px 14px; font-weight: bold;}
        .demo-btn:hover {background-color: #28a745 !important; color: white !important;}
    """
) as demo:

    # Title
    gr.Markdown(
        """
        # 🧬 GENESIS-AI — Synthetic Biology Research Engine
        _Your AI co-pilot for disruptive biotech discovery._
        """,
        elem_id="title"
    )

    # Input row
    with gr.Row():
        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)

    # Demo query buttons
    gr.Markdown("**Or try one of these expert-curated demo queries:**")
    with gr.Row():
        for q in DEMO_QUERIES:
            btn = gr.Button(q, elem_classes="demo-btn")
            btn.click(
                fn=run_pipeline_ui,
                inputs=gr.Textbox(value=q, visible=False),
                outputs=["summary_out", "cites_out", "img_out", "audio_out"]
            )

    # Output tabs
    with gr.Tab("📄 Summary"):
        summary_out = gr.Textbox(label="Executive Summary", lines=15, interactive=False, elem_id="summary_out")
    with gr.Tab("🔗 Citations"):
        cites_out = gr.Markdown(elem_id="cites_out")
    with gr.Tab("🖼 Diagram"):
        img_out = gr.Image(type="filepath", label="Generated Research Diagram", elem_id="img_out")
    with gr.Tab("🔊 Narration"):
        audio_out = gr.Audio(type="filepath", label="AI Narration", elem_id="audio_out")

    # Main run button action
    run_btn.click(
        fn=run_pipeline_ui,
        inputs=query_box,
        outputs=[summary_out, cites_out, img_out, audio_out]
    )

demo.launch()