Update app.py

app.py

@@ -1,26 +1,57 @@
 import streamlit as st
-from prompt_builder import build_prompt
 from model_api import query_model
+from prompt_builder import build_prompt
 
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-st.title("🏋️ FitPlan AI - Personalized Workout Generator")
+st.title("🏋️ FitPlan AI – Personalized Workout Generator")
+
+# ---------------- INPUT ---------------- #
+
+name = st.text_input("Enter Your Name")
+
+gender = st.radio("Gender", ["Male", "Female"])
+
+height = st.number_input("Height (cm)", min_value=100.0, max_value=250.0)
+weight = st.number_input("Weight (kg)", min_value=30.0, max_value=200.0)
+
+goal = st.selectbox(
+    "Fitness Goal",
+    ["Build Muscle", "Lose Weight", "Improve Endurance", "General Fitness"]
+)
+
+fitness_level = st.radio(
+    "Fitness Level",
+    ["Beginner", "Intermediate", "Advanced"]
+)
+
+equipment = st.multiselect(
+    "Available Equipment",
+    [
+        "No Equipment", "Dumbbells", "Barbell",
+        "Pull-up Bar", "Resistance Bands",
+        "Treadmill", "Kettlebells", "Full Gym"
+    ]
+)
+
+# ---------------- GENERATE ---------------- #
+
+if st.button("Generate Workout Plan"):
 
-# User Inputs
-goal = st.selectbox("Select Your Goal", 
-                    ["Build Muscle", "Lose Weight", "Improve Endurance"])
+    if height > 0 and weight > 0:
 
-fitness_level = st.selectbox("Fitness Level", 
-                             ["Beginner", "Intermediate", "Advanced"])
+        prompt, bmi, bmi_status = build_prompt(
+            name, gender, height, weight,
+            goal, fitness_level, equipment
+        )
 
-equipment = st.selectbox("Available Equipment",
-                         ["No Equipment", "Dumbbells", "Full Gym"])
+        with st.spinner("Generating your personalized plan..."):
+            response = query_model(prompt)
 
-if st.button("Generate Plan"):
+        st.subheader("📋 Your Personalized Workout Plan")
+        st.write(response)
 
-    with st.spinner("Generating your personalized workout plan..."):
-        prompt = build_prompt(goal, fitness_level, equipment)
-        response = query_model(prompt)
+        st.info(f"Calculated BMI: {bmi:.2f} ({bmi_status})")
 
-    st.subheader("Your Personalized Workout Plan")
-    st.write(response)
+    else:
+        st.warning("Please enter valid height and weight.")