Update app.py

app.py

@@ -20,20 +20,6 @@ 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
 # ------------------------------
@@ -360,7 +346,7 @@ workflow.add_edge("rewrite", "agent")
 app = workflow.compile()
 
 # ------------------------------
-# Streamlit UI & Theme
+# Streamlit UI
 # ------------------------------
 class UITheme:
     primary_color = "#2E86C1"
@@ -406,6 +392,17 @@ 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):
@@ -485,4 +482,4 @@ def main():
         """)
 
 if __name__ == "__main__":
-    main()
+    main()