Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,177 +3,141 @@ import pandas as pd
 
 # --- Page Config ---
 st.set_page_config(
-    page_title="FitPulse Pro | Dashboard",
+    page_title="FitPulse Pro",
     page_icon="⚡",
-    layout="wide",
-    initial_sidebar_state="expanded",
+    layout="wide"
 )
 
-# --- Custom CSS for Modern UI ---
+# --- Modern UI Styling ---
 st.markdown("""
     <style>
-    /* Main Background */
-    .stApp {
-        background-color: #f4f7f6;
+    .stApp { background-color: #F8FAFC; }
+    [data-testid="stSidebar"] { background-color: #111827; }
+    .main-card {
+        background: white;
+        padding: 2rem;
+        border-radius: 15px;
+        box-shadow: 0 10px 15px -3px rgba(0,0,0,0.1);
     }
-    
-    /* Sidebar Styling */
-    section[data-testid="stSidebar"] {
-        background-color: #1E1E2F !important;
-        color: white;
-    }
-    
-    /* Card Styling */
-    div.stButton > button:first-child {
-        background-color: #624BFF;
-        color: white;
-        border-radius: 8px;
-        border: none;
-        padding: 0.6rem 2rem;
-        transition: all 0.3s ease;
-    }
-    
-    div.stButton > button:hover {
-        background-color: #4B39C1;
-        transform: translateY(-2px);
-    }
-
-    /* Custom Metric Box */
-    .metric-card {
-        background-color: white;
-        padding: 20px;
+    .metric-box {
+        background: #ffffff;
+        padding: 1.5rem;
         border-radius: 12px;
-        box-shadow: 0 4px 6px rgba(0,0,0,0.05);
-        border-left: 5px solid #624BFF;
+        border: 1px solid #E2E8F0;
+        text-align: center;
     }
     </style>
     """, unsafe_allow_html=True)
 
-# --- Session State Initialization ---
-if 'profile_created' not in st.session_state:
-    st.session_state.profile_created = False
+# --- Session State (The App's Memory) ---
+if 'profile_complete' not in st.session_state:
+    st.session_state.profile_complete = False
+if 'user_data' not in st.session_state:
     st.session_state.user_data = {}
 
-# --- Helper Functions ---
+# --- Logic Functions ---
 def calculate_bmi(w, h_cm):
-    h_m = h_cm / 100
-    return round(w / (h_m ** 2), 2)
+    return round(w / ((h_cm / 100) ** 2), 2)
 
-def get_bmi_status(bmi):
-    if bmi < 18.5: return "Underweight", "🔵", "Consider a nutrient-rich diet."
-    if bmi < 25: return "Normal", "🟢", "Great job! Maintain your current lifestyle."
-    if bmi < 30: return "Overweight", "🟠", "Incorporate more cardio and portion control."
-    return "Obese", "🔴", "Consult with a healthcare provider for a plan."
+def get_bmi_info(bmi):
+    if bmi < 18.5: return "Underweight", "🔵", "Needs Caloric Surplus"
+    if bmi < 25: return "Normal", "🟢", "Maintain & Tone"
+    if bmi < 30: return "Overweight", "🟠", "Caloric Deficit Advised"
+    return "Obese", "🔴", "Medical Consultation Advised"
 
-# --- Sidebar Navigation ---
+# --- Sidebar Navigation Control ---
 with st.sidebar:
-    st.markdown("<h1 style='color: white;'>⚡ FitPulse Pro</h1>", unsafe_allow_html=True)
+    st.markdown("<h1 style='color: white;'>⚡ FitPulse</h1>", unsafe_allow_html=True)
     st.markdown("---")
-    menu = st.radio(
-        "NAVIGATION",
-        ["🏠 Dashboard", "👤 User Profile", "📊 Health Metrics", "🥗 Meal Plan (Beta)"],
-        index=0 if not st.session_state.profile_created else 0
-    )
+    
+    # If profile isn't done, lock the navigation to Profile only
+    if not st.session_state.profile_complete:
+        menu = st.radio("GET STARTED", ["👤 Setup Profile"])
+        st.warning("Please complete setup to unlock Dashboard.")
+    else:
+        menu = st.radio("MENU", ["🏠 Dashboard", "📊 Analytics", "⚙️ Edit Profile"])
+    
     st.markdown("---")
-    st.info("System Status: Online")
+    if st.session_state.profile_complete:
+        if st.button("Logout / Reset"):
+            st.session_state.profile_complete = False
+            st.rerun()
 
-# --- Page Logic ---
+# --- Page Routing ---
 
-if menu == "👤 User Profile":
-    st.header("👤 Personal Information")
-    st.write("Complete your profile to sync your health data.")
+# PAGE 1: SETUP PROFILE (Shown first)
+if not st.session_state.profile_complete or menu == "⚙️ Edit Profile":
+    st.title("👤 Create Your Fitness Identity")
+    st.write("Enter your details to generate a custom fitness roadmap.")
     
     with st.container():
+        st.markdown('<div class="main-card">', unsafe_allow_html=True)
         col1, col2 = st.columns(2)
+        
         with col1:
-            name = st.text_input("Full Name", value=st.session_state.user_data.get('name', ""))
-            height = st.number_input("Height (cm)", min_value=0.0, max_value=250.0, value=float(st.session_state.user_data.get('height', 170.0)))
+            name = st.text_input("Name", value=st.session_state.user_data.get('name', ""))
+            height = st.number_input("Height (cm)", min_value=50.0, max_value=250.0, value=float(st.session_state.user_data.get('height', 170.0)))
+        
         with col2:
-            weight = st.number_input("Weight (kg)", min_value=0.0, max_value=300.0, value=float(st.session_state.user_data.get('weight', 70.0)))
+            weight = st.number_input("Weight (kg)", min_value=10.0, max_value=300.0, value=float(st.session_state.user_data.get('weight', 70.0)))
             level = st.select_slider("Fitness Level", options=["Beginner", "Intermediate", "Advanced"])
-        
-        goal = st.selectbox("Primary Fitness Goal", ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
+            
+        goal = st.selectbox("Fitness Goal", ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
         equip = st.multiselect("Available Equipment", ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", "Full Gym"], default=["No Equipment"])
         
-        if st.button("Save & Update Profile"):
+        if st.button("🚀 Generate My Dashboard"):
             if name and height > 0 and weight > 0:
                 st.session_state.user_data = {
                     "name": name, "height": height, "weight": weight,
                     "goal": goal, "level": level, "equip": equip,
                     "bmi": calculate_bmi(weight, height)
                 }
-                st.session_state.profile_created = True
-                st.success("Profile Updated!")
-                st.balloons()
+                st.session_state.profile_complete = True
+                st.success("Analysis Complete! Redirecting to Dashboard...")
+                st.rerun()
             else:
-                st.error("Please fill in all required fields accurately.")
+                st.error("Please fill in all required fields.")
+        st.markdown('</div>', unsafe_allow_html=True)
 
+# PAGE 2: DASHBOARD (Hidden until profile is done)
 elif menu == "🏠 Dashboard":
-    if not st.session_state.profile_created:
-        st.warning("⚠️ Please complete your **User Profile** section first.")
-        st.image("https://illustrations.popsy.co/gray/fitness-stats.svg", width=400)
-    else:
-        ud = st.session_state.user_data
-        st.title(f"Welcome back, {ud['name']}! 👋")
-        
-        # Dashboard Row 1: Key Metrics
-        m1, m2, m3, m4 = st.columns(4)
-        status, icon, advice = get_bmi_status(ud['bmi'])
-        
-        m1.metric("Current BMI", f"{ud['bmi']}")
-        m2.metric("Status", status)
-        m3.metric("Goal", ud['goal'])
-        m4.metric("Level", ud['level'])
-        
-        st.markdown("---")
-        
-        # Dashboard Row 2: Visual Insights
-        c1, c2 = st.columns([2, 1])
-        with c1:
-            st.subheader("🚀 Activity Overview")
-            st.info(f"**Recommended focus:** Based on your goal of **{ud['goal']}**, you should prioritize protein intake and consistent training.")
-            
-            # Simulated Chart
-            chart_data = pd.DataFrame({
-                'Week': ['W1', 'W2', 'W3', 'W4'],
-                'Consistency': [70, 85, 80, 95]
-            })
-            st.line_chart(chart_data.set_index('Week'))
-            
-        with c2:
-            st.subheader("🛠️ Setup")
-            st.write("**Equipment:**")
-            for item in ud['equip']:
-                st.write(f"✅ {item}")
-            
-            st.markdown(f"""
-            <div class="metric-card">
-                <h4>Health Advice</h4>
-                <p>{icon} {advice}</p>
-            </div>
-            """, unsafe_allow_html=True)
+    ud = st.session_state.user_data
+    status, icon, advice = get_bmi_info(ud['bmi'])
+    
+    st.title(f"Welcome, {ud['name']} {icon}")
+    
+    # Top Row Metrics
+    m1, m2, m3, m4 = st.columns(4)
+    with m1:
+        st.markdown(f'<div class="metric-box"><h5>BMI Score</h5><h2>{ud["bmi"]}</h2></div>', unsafe_allow_html=True)
+    with m2:
+        st.markdown(f'<div class="metric-box"><h5>Category</h5><h2 style="color:blue;">{status}</h2></div>', unsafe_allow_html=True)
+    with m3:
+        st.markdown(f'<div class="metric-box"><h5>Goal</h5><h2>{ud["goal"]}</h2></div>', unsafe_allow_html=True)
+    with m4:
+        st.markdown(f'<div class="metric-box"><h5>Level</h5><h2>{ud["level"]}</h2></div>', unsafe_allow_html=True)
 
-elif menu == "📊 Health Metrics":
-    if not st.session_state.profile_created:
-        st.error("No data found. Please enter details in Profile.")
-    else:
-        st.header("📊 Deep Dive Metrics")
-        ud = st.session_state.user_data
-        
-        # Detailed BMI Breakdown
-        st.subheader("BMI Analysis")
-        st.progress(min(ud['bmi']/40, 1.0))
-        st.write(f"Your BMI is **{ud['bmi']}**. Ideal range is 18.5 - 24.9.")
+    st.markdown("### 🎯 Your Action Plan")
+    c1, c2 = st.columns([2, 1])
+    
+    with c1:
+        st.info(f"**Insight:** To reach your goal of **{ud['goal']}**, we've prepared a {ud['level']} track. You are using: {', '.join(ud['equip'])}.")
+        # Progress Chart
+        st.line_chart({"Progress": [10, 25, 45, 60, 85]})
         
-        # Calculations for calories (Simplified BMR)
-        bmr = 10 * ud['weight'] + 6.25 * ud['height'] - 5 * 25 + 5 # Simplified Male BMR for logic demo
-        st.write(f"**Estimated Daily Maintenance Calories:** {int(bmr)} kcal")
+    with c2:
+        st.success(f"**Advice:** {advice}")
+        st.warning("📅 **Next Workout:** Tomorrow, 8:00 AM")
 
-else: # Meal Plan Beta
-    st.header("🥗 Personalized Meal Plan")
-    st.info("This feature is currently in Beta. Based on your goal, we recommend a 40/30/30 Macro split.")
-    st.image("https://illustrations.popsy.co/gray/healthy-food.svg", width=300)
-
-# --- Footer ---
-st.sidebar.markdown("---")
-st.sidebar.caption("FitPulse v2.1.0-Stable")
+# PAGE 3: ANALYTICS
+elif menu == "📊 Analytics":
+    st.title("📊 Health Deep Dive")
+    ud = st.session_state.user_data
+    st.write(f"Detailed breakdown for {ud['name']}")
+    
+    # Metric explanation
+    st.write(f"Current Weight: **{ud['weight']} kg**")
+    st.write(f"Target Zone: **Normal (18.5 - 24.9)**")
+    
+    delta = round(ud['bmi'] - 22, 2)
+    st.metric("Distance from 'Ideal' BMI (22.0)", delta, delta_color="inverse")