Update app.py

app.py

@@ -1,43 +1,61 @@
-# File: app.py
 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():
-    # Page configuration with theming
-    st.set_page_config(page_title="AI Research Companion", layout="wide", initial_sidebar_state="expanded")
+    st.set_page_config(
+        page_title="🚀 MCP Research Companion",
+        layout="wide",
+        initial_sidebar_state="expanded"
+    )
 
-    # Render sidebar for inputs: query, num results, theme
-    query, num_results, theme, search_clicked = render_sidebar()
+    # Sidebar: user query, num results, theme, gemini prompt, etc.
+    query, num_results, theme, search_clicked, gemini_prompt = render_sidebar()
 
-    # Dynamically inject simple dark CSS if chosen
+    # 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
+            """
+            <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()
-
-        # 1. Search for papers via MCP servers
-        papers = dispatcher.search_papers(query, limit=num_results)
+        with st.spinner("Searching MCP servers..."):
+            papers = dispatcher.search_papers(query, limit=num_results)
         render_paper_list(papers)
 
-        # 2. Show notebook for the first paper
         if papers:
-            first_id = papers[0]["id"]
-            notebook_cells = dispatcher.get_notebook_cells(first_id)
+            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)
 
-            # 3. Visualize a knowledge graph for the paper
-            graph_data = dispatcher.get_graph(first_id)
+            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()