Update app.py

app.py

@@ -1,33 +1,30 @@
-# app.py
-
 import streamlit as st
 import os
 from landing import show_landing
-from dashboard.logs import show_logs
 from agent_manager import AgentManager
+from dashboard.logs import show_logs
 from stripe_checkout import create_stripe_session
 
-st.set_page_config(page_title="AutoExec AI", layout="wide")
-
-# Read any query params (e.g., ?mode=landing) using the new API
-mode = st.query_params.get("mode", ["app"])[0]
+# Initialize session state for page navigation
+if "page" not in st.session_state:
+    st.session_state.page = "Home"
 
 # Sidebar navigation
-st.sidebar.header("AutoExec AI")
-selection = st.sidebar.radio(
+st.sidebar.title("AutoExec AI")
+st.session_state.page = st.sidebar.radio(
     "Go to",
-    ["🏠 Home", "🚀 Launch", "📊 Logs", "⚙️ Settings"],
-    index=0 if mode in ("app", "landing") else None
+    ["Home", "Launch", "Logs", "Settings"],
+    index=["Home", "Launch", "Logs", "Settings"].index(st.session_state.page),
+    key="page",
 )
 
-# Route: Landing or Home
-if mode == "landing" or selection == "🏠 Home":
+# Route to the correct page
+if st.session_state.page == "Home":
     show_landing()
 
-# Route: Launch
-elif selection == "🚀 Launch":
+elif st.session_state.page == "Launch":
     st.header("🚀 Launch a New AI Business")
-    niche = st.text_input("Niche (e.g. fitness)")
+    niche = st.text_input("Niche (e.g., fitness)")
     business_type = st.selectbox("Business Type", ["Dropshipping", "Print-on-Demand", "Newsletter", "Course"])
     if st.button("Generate & Deploy"):
         manager = AgentManager(niche, business_type)
@@ -35,22 +32,12 @@ elif selection == "🚀 Launch":
         st.success("✅ Business Launched!")
         st.json(result)
 
-# Route: Logs
-elif selection == "📊 Logs":
+elif st.session_state.page == "Logs":
     show_logs()
 
-# Route: Settings & Billing
-elif selection == "⚙️ Settings":
+elif st.session_state.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(
-        """
-        **Secrets in use**:
-        - `OPENAI_API_KEY`  
-        - `GEMINI_API_KEY`  
-        - `STRIPE_API_KEY`  
-        - `API_KEY` (for protected API routes)
-        """
-    )
+    st.markdown("Manage API keys and subscriptions here.")