Update app.py

app.py

@@ -1,6 +1,5 @@
 # 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
@@ -9,16 +8,24 @@ from components.graph_view import render_graph
 
 
 def main():
-    st.set_page_config(page_title="AI Research Companion", layout="wide")
+    # Page configuration with theming
+    st.set_page_config(page_title="AI Research Companion", layout="wide", initial_sidebar_state="expanded")
+
+    # Render sidebar for inputs: query, num results, theme
+    query, num_results, theme, search_clicked = render_sidebar()
 
-    # Render sidebar and get user inputs
-    query, search_clicked = render_sidebar()
+    # Dynamically inject simple dark CSS if chosen
+    if theme == "Dark":
+        st.markdown(
+            "<style>body {background-color: #0E1117; color: #E6E1DC;} .stButton>button {background-color: #2563EB; color: white;}</style>",
+            unsafe_allow_html=True
+        )
 
     if search_clicked and query:
         dispatcher = Dispatcher()
 
-        # 1. Search papers
-        papers = dispatcher.search_papers(query)
+        # 1. Search for papers via MCP servers
+        papers = dispatcher.search_papers(query, limit=num_results)
         render_paper_list(papers)
 
         # 2. Show notebook for the first paper
@@ -27,7 +34,7 @@ def main():
             notebook_cells = dispatcher.get_notebook_cells(first_id)
             render_notebook(notebook_cells)
 
-            # 3. Visualize knowledge graph
+            # 3. Visualize a knowledge graph for the paper
             graph_data = dispatcher.get_graph(first_id)
             render_graph(graph_data)