Update app.py

app.py

@@ -9,12 +9,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 import streamlit as st
 
-st.set_page_config(
-        page_title="Prospira",
-        page_icon="📊",
-        layout="wide",
-        initial_sidebar_state="expanded"
-    )
 # --- Data Processing Class ---
 class DataProcessor:
     def __init__(self):
@@ -431,25 +425,30 @@ def render_chat():
 
 # Replace the existing render_chat() function in your main.py with this implementation
 def main():
+    st.set_page_config(
+        page_title="Prospira",
+        page_icon="📊",
+        layout="wide",
+        initial_sidebar_state="expanded"
+    )
     
+    with st.sidebar:
+        st.title("Prospira")
+        st.subheader("Data-Driven Solutions")
+        
+        page = st.radio(
+            "Navigation",
+            ["Dashboard", "Analytics", "Brainstorm", "Chat"]
+        )
     
-with st.sidebar:
-    st.title("Prospira")
-    st.subheader("Data-Driven Solutions")
-    
-    if 'page' not in st.session_state:
-        st.session_state.page = "Dashboard"  # Default page
-
-    if st.button("Dashboard"):
-        st.session_state.page = "Dashboard"
-    if st.button("Analytics"):
-        st.session_state.page = "Analytics"
-    if st.button("Brainstorm"):
-        st.session_state.page = "Brainstorm"
-    if st.button("Chat"):
-        st.session_state.page = "Chat"
-
-    page = st.session_state.page
+    if page == "Dashboard":
+        render_dashboard()
+    elif page == "Analytics":
+        render_analytics()
+    elif page == "Brainstorm":
+        render_brainstorm_page()
+    elif page == "Chat":
+        render_chat()
 
 if __name__ == "__main__":
     main()