Update app.py

app.py

@@ -8,6 +8,12 @@ import os
 st.set_page_config(page_title="FitPlan-AI", page_icon="💪")
 st.title("💪 FitPlan-AI: Personalized Fitness Profile")
 
+# -------------------------
+# SESSION STATE FOR PAGE NAVIGATION
+# -------------------------
+if "page" not in st.session_state:
+    st.session_state.page = "form"
+
 # -------------------------
 # LOAD HF CLIENT (CACHED)
 # -------------------------
@@ -41,62 +47,54 @@ def get_category(bmi):
         return "Obese"
 
 # -------------------------
-# PROMPT BUILDER
+# PROMPT BUILDER (AGE ADDED)
 # -------------------------
-def build_prompt(name, height, weight, goal, level, equipment, bmi, bmi_status):
+def build_prompt(name, age, height, weight, goal, level, equipment, bmi, bmi_status):
     equipment_list = ", ".join(equipment)
 
     prompt = f"""
 You are a certified professional fitness trainer.
-
 IMPORTANT:
 - Do NOT write introduction.
 - Do NOT write explanation.
 - Do NOT give nutrition advice.
 - Output ONLY the workout plan.
 - Keep it concise.
-
 Client Details:
 Name: {name}
+Age: {age}
 Height: {height} cm
 Weight: {weight} kg
 BMI: {bmi} ({bmi_status})
 Goal: {goal}
 Fitness Level: {level}
 Available Equipment: {equipment_list}
-
 STRICT FORMAT:
-
 Day 1:
 Warm-up:
 Main Workout (sets x reps):
 Rest:
 Cooldown:
-
 Day 2:
 Warm-up:
 Main Workout (sets x reps):
 Rest:
 Cooldown:
-
 Day 3:
 Warm-up:
 Main Workout (sets x reps):
 Rest:
 Cooldown:
-
 Day 4:
 Warm-up:
 Main Workout (sets x reps):
 Rest:
 Cooldown:
-
 Day 5:
 Warm-up:
 Main Workout (sets x reps):
 Rest:
 Cooldown:
-
 Each day must contain 4–5 exercises only.
 Keep total response under 900 words.
 """
@@ -124,79 +122,118 @@ def query_model(prompt):
     except Exception as e:
         return f"⚠️ Model Error: {str(e)}"
 
-# -------------------------
-# FORM
-# -------------------------
-with st.form("fitness_form"):
+# =========================
+# PAGE 1 → FORM
+# =========================
+if st.session_state.page == "form":
 
-    st.subheader("Personal Information")
+    with st.form("fitness_form"):
 
-    name = st.text_input("Full Name*", placeholder="Enter your name")
+        st.subheader("Personal Information")
 
-    col1, col2 = st.columns(2)
-    with col1:
-        height = st.number_input("Height (cm)*", min_value=1.0, step=0.1)
-    with col2:
-        weight = st.number_input("Weight (kg)*", min_value=1.0, step=0.1)
+        name = st.text_input("Full Name*", placeholder="Enter your name")
 
-    st.subheader("Fitness Details")
+        age = st.number_input("Age*", min_value=18, max_value=65, step=1)
 
-    goal = st.selectbox(
-        "Fitness Goal",
-        ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexibility"]
-    )
+        col1, col2 = st.columns(2)
+        with col1:
+            height = st.number_input("Height (cm)*", min_value=1.0, step=0.1)
+        with col2:
+            weight = st.number_input("Weight (kg)*", min_value=1.0, step=0.1)
 
-    level = st.radio(
-        "Fitness Level",
-        ["Beginner", "Intermediate", "Advanced"]
-    )
+        st.subheader("Fitness Details")
 
-    equipment = st.multiselect(
-        "Available Equipment",
-        ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", 
-         "Kettlebell", "Pull-up Bar"]
-    )
+        goal = st.selectbox(
+            "Fitness Goal",
+            ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexibility"]
+        )
 
-    submit = st.form_submit_button("Submit Profile")
+        level = st.radio(
+            "Fitness Level",
+            ["Beginner", "Intermediate", "Advanced"]
+        )
 
-# -------------------------
-# HANDLE SUBMISSION
-# -------------------------
-if submit:
+        equipment = st.multiselect(
+            "Available Equipment",
+            ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", 
+             "Kettlebell", "Pull-up Bar"]
+        )
 
-    if not name:
-        st.error("Please enter your name.")
+        submit = st.form_submit_button("Submit Profile")
+
+    # -------------------------
+    # HANDLE SUBMISSION
+    # -------------------------
+    if submit:
+
+        if not name:
+            st.error("Please enter your name.")
+
+        elif height <= 0 or weight <= 0:
+            st.error("Please enter valid height and weight.")
+
+        elif not equipment:
+            st.error("Please select at least one equipment option.")
+
+        else:
+            # Store in session
+            st.session_state.name = name
+            st.session_state.age = age
+            st.session_state.height = height
+            st.session_state.weight = weight
+            st.session_state.goal = goal
+            st.session_state.level = level
+            st.session_state.equipment = equipment
+
+            st.session_state.page = "result"
+            st.rerun()
+
+# =========================
+# PAGE 2 → RESULT
+# =========================
+if st.session_state.page == "result":
+
+    name = st.session_state.name
+    age = st.session_state.age
+    height = st.session_state.height
+    weight = st.session_state.weight
+    goal = st.session_state.goal
+    level = st.session_state.level
+    equipment = st.session_state.equipment
+
+    st.success("Profile Submitted Successfully!")
+
+    bmi = calculate_bmi(weight, height)
+    bmi_status = get_category(bmi)
+
+    st.write("## 📋 Your Personal Information")
+    st.write(f"**Name:** {name}")
+    st.write(f"**Age:** {age}")
+    st.write(f"**Height:** {height} cm")
+    st.write(f"**Weight:** {weight} kg")
+    st.write(f"**BMI:** {bmi} ({bmi_status})")
+    st.write(f"**Goal:** {goal}")
+    st.write(f"**Fitness Level:** {level}")
+    st.write(f"**Equipment:** {', '.join(equipment)}")
+
+    with st.spinner("Generating your 5-day workout plan..."):
+        prompt = build_prompt(
+            name=name,
+            age=age,
+            height=height,
+            weight=weight,
+            goal=goal,
+            level=level,
+            equipment=equipment,
+            bmi=bmi,
+            bmi_status=bmi_status
+        )
 
-    elif height <= 0 or weight <= 0:
-        st.error("Please enter valid height and weight.")
+        full_plan = query_model(prompt)
 
-    elif not equipment:
-        st.error("Please select at least one equipment option.")
+    st.subheader("🏋️ Your Personalized 5-Day Workout Plan")
+    st.write(full_plan)
 
-    else:
-        st.success("Profile Submitted Successfully!")
-
-        # BMI Calculation
-        bmi = calculate_bmi(weight, height)
-        bmi_status = get_category(bmi)
-
-        st.write(f"### 📊 Your BMI: {bmi} ({bmi_status})")
-
-        # Generate Plan
-        with st.spinner("Generating your 5-day workout plan..."):
-            prompt = build_prompt(
-                name=name,
-                height=height,
-                weight=weight,
-                goal=goal,
-                level=level,
-                equipment=equipment,
-                bmi=bmi,
-                bmi_status=bmi_status
-            )
-
-            full_plan = query_model(prompt)
-
-        # Display Plan
-        st.subheader("🏋️ Your Personalized 5-Day Workout Plan")
-        st.write(full_plan)
+    if st.button("⬅️ Back"):
+        st.session_state.page = "form"
+        st.rerun()