Update app.py

app.py

@@ -1,8 +1,11 @@
+import os
+
 import streamlit as st
 from landing import show_landing_content
 from agent_manager import AgentManager
 from dashboard.logs import show_logs
 from stripe_checkout import create_stripe_session
+from shopify_client import create_shopify_product
 
 # ────────────────────────────────────────────────────────────────────────────────
 # 1. GLOBAL CONFIGURATION
@@ -14,7 +17,9 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-# Define your navigation pages and labels here
+# ────────────────────────────────────────────────────────────────────────────────
+# 2. NAVIGATION SETUP
+# ────────────────────────────────────────────────────────────────────────────────
 PAGES = {
     "Home": "🏠 Home",
     "Launch": "🚀 Launch",
@@ -22,27 +27,19 @@ PAGES = {
     "Settings": "⚙️ Settings"
 }
 
-# Initialize state
 if "current_page" not in st.session_state:
     st.session_state.current_page = "Home"
 
-# ────────────────────────────────────────────────────────────────────────────────
-# 2. SIDEBAR NAVIGATION COMPONENT
-# ────────────────────────────────────────────────────────────────────────────────
 def render_sidebar() -> str:
-    """Renders the sidebar and returns the selected page key."""
     st.sidebar.title("AutoExec AI")
     choice = st.sidebar.radio(
-        label="Navigate to:",
+        "Navigate to:",
         options=list(PAGES.values()),
         index=list(PAGES.values()).index(PAGES[st.session_state.current_page]),
-        key="nav_radio",
+        key="nav_radio"
     )
-    # Map label back to internal key
-    for key, label in PAGES.items():
-        if label == choice:
-            return key
-    return "Home"
+    # reverse‐lookup internal key
+    return next(k for k, v in PAGES.items() if v == choice)
 
 st.session_state.current_page = render_sidebar()
 
@@ -51,7 +48,6 @@ st.session_state.current_page = render_sidebar()
 # ────────────────────────────────────────────────────────────────────────────────
 def main():
     page = st.session_state.current_page
-
     if page == "Home":
         render_home()
     elif page == "Launch":
@@ -67,12 +63,10 @@ def main():
 # 4. PAGE RENDERERS
 # ────────────────────────────────────────────────────────────────────────────────
 def render_home():
-    """Renders the Home (Landing) page with hero and features."""
-    # Hero section
     st.markdown(
         """
         <div style="text-align:center; margin:2rem 0;">
-            <h1 style="font-size:3rem; margin-bottom:0.5rem;">🚀 AutoExec AI</h1>
+            <h1 style="font-size:3rem;">🚀 AutoExec AI</h1>
             <p style="font-size:1.25rem; color:#555;">
                 Launch, manage, and optimize AI‑powered businesses with a single click.
             </p>
@@ -83,38 +77,61 @@ def render_home():
     show_landing_content()
 
 def render_launch():
-    """Renders the Launch page with a form to kick off the agent pipeline."""
     st.markdown("## 🚀 Launch a New AI Business")
-    with st.form("launch_form", clear_on_submit=False):
+    with st.form("launch_form"):
         niche = st.text_input(
-            label="🎯 Niche",
+            "🎯 Niche",
             placeholder="e.g., fitness wear",
             help="Define the market or audience for your business.",
         )
         business_type = st.selectbox(
-            label="📦 Business Type",
+            "📦 Business Type",
             options=["Dropshipping", "Print-on-Demand", "Newsletter", "Course"],
-            help="Select the kind of business model to generate.",
+            help="Select the type of business model to generate.",
         )
         submit = st.form_submit_button("Generate & Deploy")
 
     if submit:
         if not niche.strip():
-            st.warning("Please enter a valid niche to continue.")
+            st.warning("Please enter a valid niche.")
             return
-        _run_agents(niche.strip(), business_type)
+
+        # 1) Run the AI agents
+        with st.spinner("🤖 Running AI agents..."):
+            manager = AgentManager(niche.strip(), business_type)
+            results = manager.run_all()
+        st.success("✅ Business Launched Successfully!")
+        st.json(results)
+
+        # 2) Publish to Shopify
+        st.markdown("### 🛍️ Publishing Product to Shopify")
+        product_title = f"{business_type} in {niche}"
+        product_desc  = results.get("copy", "")
+        price_str     = "49.00"  # default price
+        image_url     = None
+
+        try:
+            shopify_url = create_shopify_product(
+                title=product_title,
+                description=product_desc,
+                price=price_str,
+                image_url=image_url
+            )
+            st.success("🎉 Product published to Shopify!")
+            st.markdown(f"[View your live product →]({shopify_url})")
+        except Exception as e:
+            st.error(f"❌ Shopify publish failed: {e}")
 
 def render_logs():
-    """Renders the Logs dashboard."""
     st.markdown("## 📊 Agent Memory Log Dashboard")
     show_logs()
 
 def render_settings():
-    """Renders the Settings & Billing page."""
     st.markdown("## ⚙️ Settings & Billing")
     st.markdown(
         """
-        **Secrets to configure** (Settings → Secrets):
+        **Ensure the following secrets are set** under  
+        Settings → Secrets:
         - `API_KEY`  
         - `OPENAI_API_KEY`  
         - `GEMINI_API_KEY`  
@@ -126,18 +143,7 @@ def render_settings():
         st.markdown(f"[Proceed to Payment]({session_url})", unsafe_allow_html=True)
 
 # ────────────────────────────────────────────────────────────────────────────────
-# 5. AGENT EXECUTION HELPER
-# ────────────────────────────────────────────────────────────────────────────────
-def _run_agents(niche: str, business_type: str):
-    """Internal helper to run the AgentManager and display results."""
-    with st.spinner("🤖 Running AI agents... please wait"):
-        manager = AgentManager(niche, business_type)
-        results = manager.run_all()
-    st.success("✅ Business Launched Successfully!")
-    st.json(results)
-
-# ────────────────────────────────────────────────────────────────────────────────
-# 6. FOOTER
+# 5. FOOTER
 # ────────────────────────────────────────────────────────────────────────────────
 def render_footer():
     st.markdown("---")