Update src/streamlit_app.py

src/streamlit_app.py

@@ -73,71 +73,59 @@ with st.form("fitness_form"):
 # -------------------------
 # SUBMIT BUTTON
 
-if st.button(" Submit Profile"):
- 
-    if not name:
+# -------------------------
+# HANDLE SUBMISSION
+# -------------------------
+
+if submit:
 
+    if not name:
         st.error("Please enter your name.")
- 
-    elif height <= 0 or weight <= 0:
 
+    elif height <= 0 or weight <= 0:
         st.error("Please enter valid height and weight.")
- 
-    elif not equipment:
 
+    elif not equipment:
         st.error("Please select at least one equipment option.")
- 
+
     else:
+        st.success("Profile Submitted Successfully!")
 
-        st.success(" Profile Submitted Successfully!")
- 
-        bmi_status = bmi_category(bmi)
+        # ✅ Calculate BMI
+        bmi = calculate_bmi(weight, height)
+        bmi_status = get_category(bmi)
 
         equipment_list = ", ".join(equipment)
- 
-        prompt = f"""
 
+        prompt = f"""
 You are a certified professional fitness trainer.
- 
-Create a detailed 5-day workout plan.
- 
-User Information:
-
-- Gender: {gender}
 
-- BMI: {bmi:.2f} ({bmi_status})
+Create a detailed 5-day workout plan.
 
+User Information:
+- Name: {name}
+- BMI: {bmi} ({bmi_status})
 - Goal: {goal}
-
-- Fitness Level: {fitness_level}
-
+- Fitness Level: {level}
 - Equipment Available: {equipment_list}
- 
+
 Start directly with:
- 
-Day 1:
 
+Day 1:
 """
- 
+
         with st.spinner("Generating your AI workout plan..."):
- 
+
             inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
- 
-            outputs = model.generate(
 
+            outputs = model.generate(
                 **inputs,
-
-                max_new_tokens=900,
-
+                max_new_tokens=600,
                 temperature=0.7,
-
                 do_sample=True
-
             )
- 
+
             result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
- 
-        st.subheader(" Your Personalized Workout Plan")
 
-        st.write(result)
- 
+        st.subheader("Your Personalized Workout Plan")
+        st.write(result)