Update app.py

app.py

@@ -5,7 +5,7 @@ from diet_builder import build_diet_prompt
 
 st.set_page_config(page_title="FitPlan AI", layout="wide")
 
-# --- CSS: PRESERVING ALL PREVIOUS STYLES + NEW DASHBOARD ---
+# --- UI STYLING (Preserving your exact original look) ---
 st.markdown("""
     <style>
     header {visibility: hidden;}
@@ -13,146 +13,116 @@ st.markdown("""
     section[data-testid="stSidebar"] { background-color: #000000 !important; }
     section[data-testid="stSidebar"] h1, section[data-testid="stSidebar"] p { color: #FFFFFF !important; }
     
-    /* Login Box */
-    .login-box { background: white; padding: 40px; border-radius: 20px; box-shadow: 0 10px 25px rgba(0,0,0,0.1); text-align: center; margin-top: 10vh; }
-
-    /* Assessment Card (Preserved from your original) */
     .assessment-card { 
         background-color: white !important; padding: 25px; border-radius: 15px; 
         border-left: 10px solid #004d57; color: #000000 !important; margin-bottom: 30px;
     }
-
-    /* Dashboard Metric Cards */
     .stat-card { background: white; padding: 20px; border-radius: 15px; text-align: center; border-top: 5px solid #004d57; box-shadow: 0 4px 15px rgba(0,0,0,0.05); }
-    .stat-val { font-size: 22px; font-weight: bold; color: #004d57; }
-    .stat-lbl { font-size: 12px; color: #636e72; text-transform: uppercase; }
-
-    /* Unified White Box (Preserved from your original) */
-    .day-container { 
-        background-color: white !important; padding: 30px; border-radius: 25px; 
-        margin-bottom: 40px; box-shadow: 0 10px 30px rgba(0,0,0,0.1);
-    }
+    .day-container { background-color: white !important; padding: 30px; border-radius: 25px; margin-bottom: 40px; box-shadow: 0 10px 30px rgba(0,0,0,0.1); }
     .day-header { color: #004d57; font-size: 26px; font-weight: bold; margin-bottom: 20px; border-bottom: 2px solid #f0f2f6; padding-bottom: 15px; }
-    
     .item-row { display: flex; align-items: center; justify-content: space-between; padding: 18px 0; border-bottom: 1px solid #f1f1f1; }
     .item-row:last-child { border-bottom: none; }
-    .item-name { flex: 2; font-weight: 600; color: #2d3436; font-size: 18px; }
-    
+    .ex-name { flex: 2; font-weight: 600; color: #2d3436; font-size: 18px; }
     .stat-group { display: flex; gap: 15px; flex: 1.5; justify-content: flex-end; }
     .stat-pill { text-align: center; min-width: 75px; background: #f8f9fa; padding: 6px; border-radius: 12px; }
-    .pill-val { color: #004d57; font-weight: bold; font-size: 15px; display: block; }
-    .pill-lbl { font-size: 10px; color: #636e72; text-transform: uppercase; display: block; }
+    .stat-val { color: #004d57; font-weight: bold; font-size: 15px; display: block; }
+    .stat-label { font-size: 10px; color: #636e72; text-transform: uppercase; display: block; }
     </style>
     """, unsafe_allow_html=True)
 
-# --- STATE MANAGEMENT ---
-if 'logged_in' not in st.session_state: st.session_state.logged_in = False
+# --- NAVIGATION STATE ---
 if 'page' not in st.session_state: st.session_state.page = 'dashboard'
 if 'user_data' not in st.session_state: st.session_state.user_data = None
 if 'workout_plan' not in st.session_state: st.session_state.workout_plan = None
 if 'diet_plan' not in st.session_state: st.session_state.diet_plan = None
 
-# --- LOGIN PAGE ---
-if not st.session_state.logged_in:
-    _, col, _ = st.columns([1, 2, 1])
-    with col:
-        st.markdown('<div class="login-box">', unsafe_allow_html=True)
-        st.title("🏋️ FitPlan AI Login")
-        email = st.text_input("Email ID")
-        if st.button("Login", use_container_width=True):
-            if "@" in email:
-                st.session_state.logged_in = True
-                st.rerun()
-    st.markdown('</div>', unsafe_allow_html=True)
+# --- SIDEBAR ---
+with st.sidebar:
+    st.markdown("# 👤 FitPlan AI")
+    if st.button("📊 Dashboard", use_container_width=True): st.session_state.page = 'dashboard'; st.rerun()
+    if st.button("🏋️ Workout Plan", use_container_width=True): st.session_state.page = 'input'; st.rerun()
+    if st.button("🥗 Dietary Plan", use_container_width=True): st.session_state.page = 'diet'; st.rerun()
 
-else:
-    # --- SIDEBAR (Preserved Style) ---
-    with st.sidebar:
-        st.markdown("# 👤 Personalised Fitness")
-        if st.button("📊 Dashboard", use_container_width=True): st.session_state.page = 'dashboard'; st.rerun()
-        if st.button("🏋️ Workout Plan", use_container_width=True): st.session_state.page = 'input'; st.rerun()
-        if st.button("🥗 Dietary Plan", use_container_width=True): st.session_state.page = 'diet'; st.rerun()
+# --- 1. DASHBOARD PAGE ---
+if st.session_state.page == 'dashboard':
+    st.title("📊 Your Dashboard")
+    if st.session_state.user_data:
+        d = st.session_state.user_data
+        col1, col2, col3 = st.columns(3)
+        with col1: st.markdown(f'<div class="stat-card"><div style="font-size:24px; font-weight:bold;">{d["name"]}</div><div style="color:#636e72;">USER</div></div>', unsafe_allow_html=True)
+        with col2: st.markdown(f'<div class="stat-card"><div style="font-size:24px; font-weight:bold;">{d["bmi"]:.1f}</div><div style="color:#636e72;">BMI</div></div>', unsafe_allow_html=True)
+        with col3: st.markdown(f'<div class="stat-card"><div style="font-size:24px; font-weight:bold; color:{d["color"]}">{d["status"]}</div><div style="color:#636e72;">HEALTH STATUS</div></div>', unsafe_allow_html=True)
+        
         st.write("---")
-        if st.button("Logout"): st.session_state.logged_in = False; st.rerun()
-
-    # --- DASHBOARD ---
-    if st.session_state.page == 'dashboard':
-        st.title("📊 Your Fitness Dashboard")
-        if st.session_state.user_data:
-            d = st.session_state.user_data
-            c1, c2, c3, c4 = st.columns(4)
-            with c1: st.markdown(f'<div class="stat-card"><div class="stat-val">{d["name"]}</div><div class="stat-lbl">Member</div></div>', unsafe_allow_html=True)
-            with c2: st.markdown(f'<div class="stat-card"><div class="stat-val">{d["bmi"]:.1f}</div><div class="stat-lbl">BMI</div></div>', unsafe_allow_html=True)
-            with c3: st.markdown(f'<div class="stat-card"><div class="stat-val" style="color:{d["color"]}">{d["status"]}</div><div class="stat-lbl">Status</div></div>', unsafe_allow_html=True)
-            with c4: st.markdown(f'<div class="stat-card"><div class="stat-val">{d["goal"]}</div><div class="stat-lbl">Goal</div></div>', unsafe_allow_html=True)
-        else:
-            st.info("No profile found. Please click 'Workout Plan' to generate your routine.")
+        st.subheader("Current Focus")
+        st.info(f"Targeting: **{d['goal']}** at **{d['level']}** intensity.")
+    else:
+        st.warning("No data found. Please go to 'Workout Plan' and generate your profile.")
 
-    # --- INPUT PAGE ---
-    elif st.session_state.page == 'input':
-        st.title("🏋️ Fitness Profile")
-        col1, col2, col3 = st.columns([2, 1, 1])
-        with col1: name = st.text_input("Name")
-        with col2: gender = st.selectbox("Gender", ["Male", "Female", "Other"])
-        with col3: age = st.number_input("Age", min_value=1, value=19)
-        h, w = st.columns(2)
-        with h: height = st.number_input("Height (cm)", min_value=1.0)
-        with w: weight = st.number_input("Weight (kg)", min_value=1.0)
-        goal = st.selectbox("Goal", ["Build Muscle", "Weight Loss", "Strength", "Flexibility"])
-        level = st.selectbox("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
-        equip = st.multiselect("Equipment", ["Dumbbells", "Kettlebells", "Pull-up Bar", "Resistance Bands", "Yoga Mat", "No Equipment"])
+# --- 2. INPUT PAGE (Your Original Logic) ---
+elif st.session_state.page == 'input':
+    st.title("🏋️ Fitness Profile")
+    c1, c2, c3 = st.columns([2, 1, 1])
+    with c1: name = st.text_input("Name")
+    with c2: gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+    with c3: age = st.number_input("Age", min_value=1, value=19)
+    h, w = st.columns(2)
+    with h: height = st.number_input("Height (cm)", min_value=1.0)
+    with w: weight = st.number_input("Weight (kg)", min_value=1.0)
+    goal = st.selectbox("Goal", ["Build Muscle", "Weight Loss", "Strength", "Flexibility"])
+    level = st.selectbox("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
+    equip = st.multiselect("Equipment", ["Dumbbells", "Kettlebells", "Pull-up Bar", "Resistance Bands", "Yoga Mat", "No Equipment"])
 
-        if st.button("Generate Plan"):
-            prompt, bmi, status, color = build_prompt(name, gender, age, height, weight, goal, level, equip)
-            with st.spinner("AI is crafting your routine..."):
-                st.session_state.workout_plan = query_model(prompt)
-                st.session_state.user_data = {"name": name, "bmi": bmi, "status": status, "color": color, "age": age, "gender": gender, "goal": goal, "height": height, "weight": weight}
-                st.session_state.diet_plan = None # Reset diet for new profile
-                st.session_state.page = 'result'
-                st.rerun()
+    if st.button("Generate Complete Plan"):
+        prompt, bmi, status, color = build_prompt(name, gender, age, height, weight, goal, level, equip)
+        with st.spinner("AI is crafting your routine..."):
+            st.session_state.workout_plan = query_model(prompt)
+            st.session_state.user_data = {"name": name, "bmi": bmi, "status": status, "color": color, "age": age, "gender": gender, "goal": goal, "height": height, "weight": weight, "level": level}
+            st.session_state.diet_plan = None # Reset diet so it generates fresh for this user
+            st.session_state.page = 'result'
+            st.rerun()
 
-    # --- RESULT PAGE (Workout) ---
-    elif st.session_state.page == 'result':
-        d = st.session_state.user_data
-        st.markdown(f'<div class="assessment-card"><h2>Hello {d["name"]}</h2><p>BMI: <strong>{d["bmi"]:.2f}</strong> | <span style="color:{d["color"]}">{d["status"]}</span></p></div>', unsafe_allow_html=True)
-
-        raw_content = st.session_state.workout_plan
-        days = ["Day 1:", "Day 2:", "Day 3:", "Day 4:", "Day 5:"]
-        for i in range(len(days)):
-            start = raw_content.find(days[i])
-            if start != -1:
-                end = raw_content.find(days[i+1]) if i+1 < len(days) else len(raw_content)
-                day_text = raw_content[start:end].strip()
-                st.markdown(f'<div class="day-container"><div class="day-header">🔷 {days[i][:-1]}</div>', unsafe_allow_html=True)
-                for line in day_text.split("\n"):
-                    if "|" in line:
-                        parts = [p.strip() for p in line.replace("- ", "").split("|")]
-                        if len(parts) == 4:
-                            n, s, r, rs = parts
-                            st.markdown(f'<div class="item-row"><div class="item-name">🔥 {n}</div><div class="stat-group"><div class="stat-pill"><span class="pill-val">{s}</span><span class="pill-lbl">Sets</span></div><div class="stat-pill"><span class="pill-val">{r}</span><span class="pill-lbl">Reps</span></div><div class="stat-pill" style="background:#fff3e0;"><span class="pill-val" style="color:#e65100;">{rs}</span><span class="pill-lbl">Rest</span></div></div></div>', unsafe_allow_html=True)
-                st.markdown('</div>', unsafe_allow_html=True)
-        
-        # Download Button Preserved
-        st.download_button("📥 Download Full Plan", data=raw_content, file_name=f"{d['name']}_Workout.txt")
-
-    # --- DIET PAGE ---
-    elif st.session_state.page == 'diet':
-        st.title("🥗 Personalized Dietary Plan")
-        if st.session_state.user_data:
-            d = st.session_state.user_data
-            if not st.session_state.diet_plan:
-                diet_prompt = build_diet_prompt(d['name'], d['gender'], d['age'], d['height'], d['weight'], d['goal'])
-                with st.spinner("Preparing meal plan..."):
-                    st.session_state.diet_plan = query_model(diet_prompt)
-            
-            raw_diet = st.session_state.diet_plan
-            st.markdown('<div class="day-container"><div class="day-header">🍽️ Daily Nutrition</div>', unsafe_allow_html=True)
-            for line in raw_diet.split("\n"):
+# --- 3. WORKOUT RESULT (Preserving your exact Row/Pill layout) ---
+elif st.session_state.page == 'result':
+    d = st.session_state.user_data
+    st.markdown(f'<div class="assessment-card"><h2>Hello {d["name"]}</h2><p>BMI: <strong>{d["bmi"]:.2f}</strong> | <span style="color:{d["color"]}">{d["status"]}</span></p></div>', unsafe_allow_html=True)
+    
+    raw = st.session_state.workout_plan
+    days = ["Day 1:", "Day 2:", "Day 3:", "Day 4:", "Day 5:"]
+    for i in range(len(days)):
+        start = raw.find(days[i])
+        if start != -1:
+            end = raw.find(days[i+1]) if i+1 < len(days) else len(raw)
+            day_text = raw[start:end].strip()
+            st.markdown(f'<div class="day-container"><div class="day-header">🔷 {days[i][:-1]}</div>', unsafe_allow_html=True)
+            for line in day_text.split("\n"):
                 if "|" in line:
                     parts = [p.strip() for p in line.replace("- ", "").split("|")]
                     if len(parts) == 4:
-                        m_n, m_t, m_i, m_c = parts
-                        st.markdown(f'<div class="item-row"><div class="item-name">🍴 {m_n}<br><small>{m_i}</small></div><div class="stat-group"><div class="stat-pill"><span class="pill-val">{m_t}</span><span class="pill-lbl">Time</span></div><div class="stat-pill" style="background:#e8f5e9;"><span class="pill-val" style="color:#2e7d32;">{m_c}</span><span class="pill-lbl">Energy</span></div></div></div>', unsafe_allow_html=True)
+                        n, s, r, rs = parts
+                        st.markdown(f'<div class="item-row"><div class="ex-name">🔥 {n}</div><div class="stat-group"><div class="stat-pill"><span class="stat-val">{s}</span><span class="stat-label">Sets</span></div><div class="stat-pill"><span class="stat-val">{r}</span><span class="stat-label">Reps</span></div><div class="stat-pill" style="background:#fff3e0;"><span class="stat-val" style="color:#e65100;">{rs}</span><span class="stat-label">Rest</span></div></div></div>', unsafe_allow_html=True)
             st.markdown('</div>', unsafe_allow_html=True)
-        else:
-            st.warning("Please generate a profile first.")
+    st.download_button("📥 Download Plan", data=raw, file_name=f"{d['name']}_Plan.txt")
+
+# --- 4. DIETARY PAGE (Now Functional) ---
+elif st.session_state.page == 'diet':
+    st.title("🥗 Dietary Plan")
+    if st.session_state.user_data:
+        d = st.session_state.user_data
+        if not st.session_state.diet_plan:
+            d_prompt = build_diet_prompt(d['name'], d['gender'], d['age'], d['height'], d['weight'], d['goal'])
+            with st.spinner("Generating unique meal plan..."):
+                st.session_state.diet_plan = query_model(d_prompt)
+        
+        raw_diet = st.session_state.diet_plan
+        st.markdown('<div class="day-container"><div class="day-header">🍏 Daily Nutrition</div>', unsafe_allow_html=True)
+        for line in raw_diet.split("\n"):
+            if "|" in line:
+                parts = [p.strip() for p in line.replace("- ", "").split("|")]
+                if len(parts) == 4:
+                    m_n, m_t, m_i, m_c = parts
+                    st.markdown(f'<div class="item-row"><div class="ex-name">🍴 {m_n}<br><small>{m_i}</small></div><div class="stat-group"><div class="stat-pill"><span class="stat-val">{m_t}</span><span class="stat-label">Time</span></div><div class="stat-pill" style="background:#e8f5e9;"><span class="stat-val" style="color:#2e7d32;">{m_c}</span><span class="stat-label">Energy</span></div></div></div>', unsafe_allow_html=True)
+        st.markdown('</div>', unsafe_allow_html=True)
+        if st.button("Refresh Meals"): st.session_state.diet_plan = None; st.rerun()
+    else:
+        st.warning("Generate a profile in 'Workout Plan' first.")