Update app.py

app.py

@@ -60,23 +60,16 @@ def bmi_category(bmi):
 # Submit Button
 # -------------------------
 
-if st.button(" Generate 5-day workout plan"):
-    
-    # Validation
-    if not name or height_cm <= 0 or weight_kg <= 0:
-        st.error("⚠ Please fill all required fields with valid values!")
-    else:
-        bmi = calculate_bmi(weight_kg, height_cm)
-        category = bmi_category(bmi)
+if st.button("Generate Workout Plan"):
+
+    bmi = calculate_bmi(weight_kg, height_cm)
+    category = bmi_category(bmi)
 
-        st.success("✅ Calculation Successful!")
+    st.success(f"BMI: {bmi} ({category})")
 
-        st.write(f"### 👤 Name: {name}")
-        st.write(f"### 📊 Your BMI: {bmi}")
-        st.write(f"### 🏷 BMI Category: {category}")
+    workout_plan = generate_5day_plan(
+        name, bmi, category, fitness_level, equipment
+    )
 
-        st.write("---")
-        st.write("### 🏋 Fitness Summary")
-        st.write(f"**Goal:** {goal}")
-        st.write(f"**Fitness Level:** {fitness_level}")
-        st.write(f"**Equipment Available:** {', '.join(equipment) if equipment else 'None selected'}")
+    st.subheader("🏋️ Your Personalized Workout Plan")
+    st.text(workout_plan)