Update app.py

app.py

@@ -2,81 +2,95 @@ import streamlit as st
 from prompt_builder import build_prompt 
 from model_api import query_model 
 
-st.set_page_config(page_title="FitPlan AI", layout="wide")
+st.set_page_config(page_title="FitPlan AI | Dashboard", layout="wide")
 
-# Optimized CSS for Unified Daily Cards
+# Professional Dashboard & Card UI
 st.markdown("""
     <style>
+    header {visibility: hidden;}
     .stApp { background: linear-gradient(135deg, #f6d365 0%, #fda085 100%); background-attachment: fixed; }
     section[data-testid="stSidebar"] { background-color: #000000 !important; }
     section[data-testid="stSidebar"] h1, section[data-testid="stSidebar"] p { color: #FFFFFF !important; }
     
-    /* Assessment Card */
-    .assessment-card { 
-        background-color: white !important; padding: 25px; border-radius: 15px; 
-        border-left: 10px solid #004d57; color: #000000 !important; margin-bottom: 30px;
+    /* Dashboard Stat Cards */
+    .stat-card {
+        background: white;
+        padding: 20px;
+        border-radius: 15px;
+        box-shadow: 0 4px 15px rgba(0,0,0,0.05);
+        text-align: center;
+        border-top: 5px solid #004d57;
     }
+    .stat-value { font-size: 24px; font-weight: bold; color: #004d57; }
+    .stat-label { font-size: 14px; color: #636e72; text-transform: uppercase; }
 
-    /* Single Unified White Box Per Day */
+    /* Unified Day Box Per Day */
     .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);
+        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; 
-        display: flex; 
-        align-items: center;
-        border-bottom: 2px solid #f0f2f6;
-        padding-bottom: 15px;
+        color: #004d57; font-size: 26px; font-weight: bold; margin-bottom: 20px; 
+        display: flex; align-items: center; border-bottom: 2px solid #f0f2f6; padding-bottom: 15px;
     }
-
-    /* Clean Exercise Rows inside the Unified Box */
     .exercise-row { 
-        display: flex; 
-        align-items: center; 
-        justify-content: space-between; 
-        padding: 18px 0;
-        border-bottom: 1px solid #f1f1f1;
+        display: flex; align-items: center; justify-content: space-between; 
+        padding: 18px 0; border-bottom: 1px solid #f1f1f1;
     }
     .exercise-row:last-child { border-bottom: none; }
-
-    .ex-name { flex: 2; font-weight: 600; color: #2d3436; font-size: 18px; display: flex; align-items: center; }
-    
+    .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;
-    }
-    
+    .stat-pill { text-align: center; min-width: 75px; background: #f8f9fa; padding: 6px; border-radius: 12px; }
     .stat-val { color: #004d57; font-weight: bold; font-size: 15px; display: block; }
-    .stat-label { font-size: 10px; color: #636e72; text-transform: uppercase; display: block; margin-top: 2px; }
+    .stat-label-inner { font-size: 10px; color: #636e72; text-transform: uppercase; display: block; }
     </style>
     """, unsafe_allow_html=True)
 
-if 'page' not in st.session_state: st.session_state.page = 'input'
+# Initialize Session States
+if 'page' not in st.session_state: st.session_state.page = 'dashboard'
 if 'workout_plan' not in st.session_state: st.session_state.workout_plan = None
+if 'user_data' not in st.session_state: st.session_state.user_data = None
 
-# --- SIDEBAR ---
+# --- SIDEBAR NAVIGATION ---
 with st.sidebar:
-    st.markdown("# 👤 Personalised Fitness Plan")
-    if st.session_state.page == 'result':
-        if st.button("⬅️ Back to Profile"):
-            st.session_state.page = 'input'
-            st.rerun()
+    st.markdown("# 🛡️ FitPlan AI")
+    st.write("---")
+    if st.button("📊 My Dashboard", use_container_width=True):
+        st.session_state.page = 'dashboard'
+        st.rerun()
+    if st.button("🏋️ Plan Generator", 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()
+
+# --- PAGE: DASHBOARD ---
+if st.session_state.page == 'dashboard':
+    st.title("📊 Welcome Back!")
+    
+    if st.session_state.user_data:
+        d = st.session_state.user_data
+        col1, col2, col3, col4 = st.columns(4)
+        with col1:
+            st.markdown(f'<div class="stat-card"><div class="stat-value">{d["name"]}</div><div class="stat-label">User</div></div>', unsafe_allow_html=True)
+        with col2:
+            st.markdown(f'<div class="stat-card"><div class="stat-value">{d["bmi"]:.1f}</div><div class="stat-label">Current BMI</div></div>', unsafe_allow_html=True)
+        with col3:
+            st.markdown(f'<div class="stat-card"><div class="stat-value" style="color:{d["color"]}">{d["status"]}</div><div class="stat-label">Health Status</div></div>', unsafe_allow_html=True)
+        with col4:
+            st.markdown(f'<div class="stat-card"><div class="stat-value">5 Days</div><div class="stat-label">Active Plan</div></div>', unsafe_allow_html=True)
+        
+        st.write("### 📅 Quick Access")
+        if st.session_state.workout_plan:
+            st.success("Your workout plan is ready! Go to 'Plan Generator' to view it.")
+        else:
+            st.info("No active plan found. Click 'Plan Generator' in the sidebar to start.")
+    else:
+        st.warning("Please go to 'Plan Generator' to set up your profile first!")
 
-# --- INPUT PAGE ---
-if st.session_state.page == 'input':
+# --- PAGE: INPUT / GENERATOR ---
+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")
@@ -89,22 +103,20 @@ if st.session_state.page == 'input':
     
     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"])
+    equip = st.multiselect("Equipment", ["Dumbbells", "Kettlebells", "Pull-up Bar", "No Equipment"])
 
     if st.button("Generate Plan"):
         prompt, bmi, bmi_status, status_color = build_prompt(name, gender, age, height, weight, goal, level, equip)
-        with st.spinner("Crafting your unified routine..."):
+        with st.spinner("Analyzing your profile..."):
             raw_workout = query_model(prompt)
             st.session_state.workout_plan = raw_workout
-            st.session_state.assessment_data = {"name": name, "bmi": bmi, "status": bmi_status, "color": status_color, "age": age, "gender": gender}
+            st.session_state.user_data = {"name": name, "bmi": bmi, "status": bmi_status, "color": status_color, "age": age, "gender": gender}
             st.session_state.page = 'result'
             st.rerun()
 
-# --- RESULT PAGE ---
+# --- PAGE: RESULT VIEW ---
 elif st.session_state.page == 'result':
-    d = st.session_state.assessment_data
-    st.markdown(f'<div class="assessment-card"><h2>Hello {d["name"]}</h2><p>{d["gender"]} | {d["age"]} Years Old</p><p>BMI: <strong>{d["bmi"]:.2f}</strong> | <span style="color:{d["color"]}">{d["status"]}</span></p></div>', unsafe_allow_html=True)
-
+    st.title("📋 Your Personalized Routine")
     raw_content = st.session_state.workout_plan
     days = ["Day 1:", "Day 2:", "Day 3:", "Day 4:", "Day 5:"]
     
@@ -114,29 +126,26 @@ elif st.session_state.page == 'result':
             end = raw_content.find(days[i+1]) if i+1 < len(days) else len(raw_content)
             day_text = raw_content[start:end].strip()
             
-            # Open Single White Day Container
             st.markdown(f'<div class="day-container"><div class="day-header">🔷 {days[i][:-1]}</div>', unsafe_allow_html=True)
-            
             lines = day_text.split("\n")
             for line in lines:
                 if "|" in line:
                     parts = [p.strip() for p in line.replace("- ", "").split("|")]
                     if len(parts) == 4:
-                        ex_name, sets, reps, rest = parts
+                        n, s, r, rs = parts
                         st.markdown(f"""
                             <div class="exercise-row">
-                                <div class="ex-name">🔥 {ex_name}</div>
+                                <div class="ex-name">🔥 {n}</div>
                                 <div class="stat-group">
-                                    <div class="stat-pill"><span class="stat-val">{sets}</span><span class="stat-label">Sets</span></div>
-                                    <div class="stat-pill"><span class="stat-val">{reps}</span><span class="stat-label">Reps</span></div>
-                                    <div class="stat-pill" style="background:#fff3e0;"><span class="stat-val" style="color:#e65100;">{rest}</span><span class="stat-label">Rest</span></div>
+                                    <div class="stat-pill"><span class="stat-val">{s}</span><span class="stat-label-inner">Sets</span></div>
+                                    <div class="stat-pill"><span class="stat-val">{r}</span><span class="stat-label-inner">Reps</span></div>
+                                    <div class="stat-pill" style="background:#fff3e0;"><span class="stat-val" style="color:#e65100;">{rs}</span><span class="stat-label-inner">Rest</span></div>
                                 </div>
                             </div>
                         """, unsafe_allow_html=True)
-            
-            if "Rest" in day_text or "Recovery" in day_text:
-                st.markdown('<p style="color:#636e72; font-style:italic; padding-top:10px;">🧘 Recovery Day: Focused on flexibility and rest.</p>', unsafe_allow_html=True)
-            
-            st.markdown('</div>', unsafe_allow_html=True) # Close Day Container
+            st.markdown('</div>', unsafe_allow_html=True)
 
-    st.download_button("📥 Download Full Plan", data=raw_content, file_name=f"{d['name']}_Workout.txt")
+# --- PAGE: DIET ---
+elif st.session_state.page == 'diet':
+    st.title("🥗 Personalized Dietary Plan")
+    st.info("Coming Soon: AI-generated meal plans based on your workout intensity!")