Update app.py

app.py

@@ -122,7 +122,84 @@ if not st.session_state.logged_in:
 
     st.stop()
 
-# ------------------ AFTER LOGIN ------------------
+# =====================================================
+# 💪 WORKOUT GENERATOR PAGE
+# =====================================================
 
-st.title("🏋️ AI Personalized 5-Day Workout Planner")
-st.success("You are logged in successfully! Now generate your workout plan.")
+st.markdown('<div class="title">💪 AI Personalized 5-Day Workout Planner</div>', unsafe_allow_html=True)
+
+with st.container():
+
+    st.markdown('<div class="card">', unsafe_allow_html=True)
+
+    name = st.text_input("Name")
+
+    age = st.number_input("Age", 10, 80)
+
+    gender = st.selectbox(
+        "Gender",
+        ["Male", "Female", "Other"]
+    )
+
+    height = st.number_input("Height (cm)", 100, 250)
+
+    weight = st.number_input("Weight (kg)", 30, 200)
+
+    goal = st.selectbox(
+        "Fitness Goal",
+        [
+            "Build Muscle",
+            "Weight Loss",
+            "Strength Gain",
+            "Abs Building",
+            "Flexibility"
+        ]
+    )
+
+    fitness_level = st.selectbox(
+        "Fitness Level",
+        [
+            "Beginner",
+            "Intermediate",
+            "Advanced"
+        ]
+    )
+
+    equipment = st.multiselect(
+        "Available Equipment",
+        [
+            "Dumbbells",
+            "Resistance Band",
+            "Yoga Mat",
+            "Skipping Rope",
+            "Pullups Bar",
+            "Inclined Bench",
+            "Weight Plates"
+        ]
+    )
+
+    if st.button("Generate 5-Day Plan 💪"):
+
+        prompt, bmi, bmi_status = build_prompt(
+            name, age, gender,
+            height, weight,
+            goal, fitness_level,
+            equipment
+        )
+
+        st.subheader(f"📊 BMI: {bmi:.2f} ({bmi_status})")
+
+        with st.spinner("Generating your AI workout plan..."):
+
+            result = query_model(prompt)
+
+        st.markdown("## 🗓️ Your Workout Plan")
+        st.markdown(result)
+
+    st.markdown("</div>", unsafe_allow_html=True)
+
+
+# ---------- LOGOUT ----------
+if st.button("Logout"):
+    st.session_state.logged_in = False
+    st.rerun()