import streamlit as st
from orchestrator.dispatcher import Dispatcher
from components.sidebar import render_sidebar
from components.paper_list import render_paper_list
from components.notebook_view import render_notebook
from components.graph_view import render_graph
from orchestrator.gemini import gemini_generate

def main():
    st.set_page_config(
        page_title="🚀 MCP Research Companion",
        layout="wide",
        initial_sidebar_state="expanded"
    )

    # Sidebar: user query, num results, theme, gemini prompt, etc.
    query, num_results, theme, search_clicked, gemini_prompt = render_sidebar()

    # Optional theme
    if theme == "Dark":
        st.markdown(
            """
            <style>
                body {background-color: #0E1117; color: #E6E1DC;}
                .stButton>button {background-color: #2563EB; color: white;}
            </style>
            """,
            unsafe_allow_html=True,
        )

    # Search and display papers
    if search_clicked and query:
        dispatcher = Dispatcher()
        with st.spinner("Searching MCP servers..."):
            papers = dispatcher.search_papers(query, limit=num_results)
        render_paper_list(papers)

        if papers:
            first_paper = papers[0]
            st.subheader("Gemini-Powered Abstract Summarizer")
            if st.button("Summarize Abstract with Gemini"):
                with st.spinner("Gemini is generating summary..."):
                    summary = gemini_generate(first_paper["abstract"])
                    st.success(summary)

            # Show notebook and graph
            notebook_cells = dispatcher.get_notebook_cells(first_paper["id"])
            render_notebook(notebook_cells)

            graph_data = dispatcher.get_graph(first_paper["id"])
            render_graph(graph_data)

    # Gemini research Q&A, even if user hasn't hit 'Search'
    if gemini_prompt:
        st.header("💡 Gemini Research Q&A")
        with st.spinner("Gemini is thinking..."):
            answer = gemini_generate(gemini_prompt)
        st.success(answer)

if __name__ == "__main__":
    main()