Update app.py

app.py

@@ -1,43 +1,52 @@
 import streamlit as st
-import os
 from landing import show_landing
 from agent_manager import AgentManager
 from dashboard.logs import show_logs
 from stripe_checkout import create_stripe_session
 
-# Initialize session state for page navigation
+# Initialize default page if not already set
 if "page" not in st.session_state:
     st.session_state.page = "Home"
 
-# Sidebar navigation
+# Define the list of pages
+PAGES = ["Home", "Launch", "Logs", "Settings"]
+
+# Create a radio widget in the sidebar, keyed to 'page'
+# This widget *updates* st.session_state.page automatically
 st.sidebar.title("AutoExec AI")
-st.session_state.page = st.sidebar.radio(
+st.sidebar.radio(
     "Go to",
-    ["Home", "Launch", "Logs", "Settings"],
-    index=["Home", "Launch", "Logs", "Settings"].index(st.session_state.page),
+    PAGES,
+    index=PAGES.index(st.session_state.page),
     key="page",
 )
 
-# Route to the correct page
-if st.session_state.page == "Home":
+# Read the current page (streamlit will have set session_state.page for us)
+page = st.session_state.page
+
+# Route to the correct content
+if page == "Home":
     show_landing()
 
-elif st.session_state.page == "Launch":
+elif page == "Launch":
     st.header("🚀 Launch a New AI Business")
     niche = st.text_input("Niche (e.g., fitness)")
-    business_type = st.selectbox("Business Type", ["Dropshipping", "Print-on-Demand", "Newsletter", "Course"])
+    business_type = st.selectbox(
+        "Business Type",
+        ["Dropshipping", "Print-on-Demand", "Newsletter", "Course"]
+    )
     if st.button("Generate & Deploy"):
         manager = AgentManager(niche, business_type)
         result = manager.run_all()
         st.success("✅ Business Launched!")
         st.json(result)
 
-elif st.session_state.page == "Logs":
+elif page == "Logs":
     show_logs()
 
-elif st.session_state.page == "Settings":
+elif page == "Settings":
     st.header("⚙️ Settings & Billing")
     if st.button("Create Stripe Checkout Session"):
         url = create_stripe_session()
         st.markdown(f"[Pay & Activate]({url})")
-    st.markdown("Manage API keys and subscriptions here.")
+    st.markdown("Manage your API keys and subscriptions here.")