Update app.py

app.py

@@ -20,6 +20,20 @@ import hashlib
 from langchain.tools.retriever import create_retriever_tool
 from datetime import datetime
 
+# ------------------------------
+# Set Streamlit Page Config Early
+# ------------------------------
+st.set_page_config(
+    page_title="AI Research Assistant Pro",
+    layout="wide",
+    initial_sidebar_state="expanded",
+    menu_items={
+        'Get Help': 'https://example.com/docs',
+        'Report a bug': 'https://example.com/issues',
+        'About': "v2.1 | Enhanced Research Assistant"
+    }
+)
+
 # ------------------------------
 # Data Definitions
 # ------------------------------
@@ -346,7 +360,7 @@ workflow.add_edge("rewrite", "agent")
 app = workflow.compile()
 
 # ------------------------------
-# Streamlit UI
+# Streamlit UI & Theme
 # ------------------------------
 class UITheme:
     primary_color = "#2E86C1"
@@ -392,17 +406,6 @@ class UITheme:
 def main():
     UITheme.apply()
     
-    st.set_page_config(
-        page_title="AI Research Assistant Pro",
-        layout="wide",
-        initial_sidebar_state="expanded",
-        menu_items={
-            'Get Help': 'https://example.com/docs',
-            'Report a bug': 'https://example.com/issues',
-            'About': "v2.1 | Enhanced Research Assistant"
-        }
-    )
-    
     with st.sidebar:
         st.header("📂 Knowledge Bases")
         with st.expander("Research Database", expanded=True):