Update src/streamlit_app.py

src/streamlit_app.py

@@ -21,15 +21,52 @@ def load_model():
 generator = load_model()
 
 # ---------------------------------------------------
-# USER NAME
+# USER DETAILS
 # ---------------------------------------------------
+st.subheader("👤 Personal Information")
+
 name = st.text_input("Enter Your Name")
 
+col1, col2 = st.columns(2)
+
+with col1:
+    height = st.number_input("Height (in cm)", min_value=100, max_value=250, value=170)
+
+with col2:
+    weight = st.number_input("Weight (in kg)", min_value=30, max_value=200, value=70)
+
 # ---------------------------------------------------
-# GOALS
+# BMI CALCULATION
 # ---------------------------------------------------
+bmi = None
+bmi_category = ""
+
+if height > 0 and weight > 0:
+    height_m = height / 100
+    bmi = weight / (height_m ** 2)
+
+    st.write(f"### 📊 Your BMI: {bmi:.2f}")
+
+    if bmi < 18.5:
+        bmi_category = "Underweight"
+        st.info("Category: Underweight")
+    elif 18.5 <= bmi < 24.9:
+        bmi_category = "Normal weight"
+        st.success("Category: Normal weight")
+    elif 25 <= bmi < 29.9:
+        bmi_category = "Overweight"
+        st.warning("Category: Overweight")
+    else:
+        bmi_category = "Obese"
+        st.error("Category: Obese")
+
+# ---------------------------------------------------
+# GOAL
+# ---------------------------------------------------
+st.subheader("🎯 Fitness Goal")
+
 goal = st.selectbox(
-    "Fitness Goal",
+    "",
     [
         "Flexible",
         "Weight Loss",
@@ -42,7 +79,7 @@ goal = st.selectbox(
 # ---------------------------------------------------
 # EQUIPMENT
 # ---------------------------------------------------
-st.subheader("Available Equipment")
+st.subheader("🏋️ Available Equipment")
 
 col1, col2 = st.columns(2)
 
@@ -88,7 +125,7 @@ for item, selected in equipment_map.items():
 # ---------------------------------------------------
 # FITNESS LEVEL
 # ---------------------------------------------------
-st.subheader("Fitness Level")
+st.subheader("📈 Fitness Level")
 
 fitness_level = st.radio(
     "",
@@ -99,16 +136,21 @@ fitness_level = st.radio(
 # ---------------------------------------------------
 # PROMPT BUILDER
 # ---------------------------------------------------
-def build_prompt(name, goal, fitness_level, equipment):
+def build_prompt(name, goal, fitness_level, equipment, bmi, bmi_category):
     equipment_text = ", ".join(equipment)
 
     prompt = f"""
-    Generate a detailed 5-day workout plan for {name}.
+    Generate a detailed 5-day personalized workout plan.
 
+    Name: {name}
     Goal: {goal}
     Fitness Level: {fitness_level}
+    BMI: {bmi:.2f}
+    BMI Category: {bmi_category}
     Available Equipment: {equipment_text}
 
+    Adjust workout intensity according to BMI and fitness level.
+
     Include:
     - Day-wise workout split
     - Exercises
@@ -121,26 +163,37 @@ def build_prompt(name, goal, fitness_level, equipment):
 
 
 # ---------------------------------------------------
-# SUBMIT BUTTON
+# GENERATE PLAN BUTTON
 # ---------------------------------------------------
-if st.button("Generate Personalised Plan"):
+if st.button("🚀 Generate Personalized Plan"):
 
     if not name:
         st.error("Please enter your name.")
     elif not equipment:
         st.error("Please select at least one equipment option.")
+    elif bmi is None:
+        st.error("Please enter valid height and weight.")
     else:
         with st.spinner("Generating your AI workout plan..."):
-            prompt = build_prompt(name, goal, fitness_level, equipment)
+
+            prompt = build_prompt(
+                name,
+                goal,
+                fitness_level,
+                equipment,
+                bmi,
+                bmi_category
+            )
 
             result = generator(
                 prompt,
-                max_length=500,
+                max_length=600,
                 do_sample=True,
                 temperature=0.7
             )
 
             plan = result[0]["generated_text"]
 
-        st.success(f"Plan Generated Successfully, {name}! 💪")
+        st.success(f"✅ Plan Generated Successfully, {name}!")
+        st.markdown("## 📝 Your Personalized Workout Plan")
         st.write(plan)