Update src/streamlit_app.py

src/streamlit_app.py

@@ -99,44 +99,39 @@ st.markdown("---")
 # =========================
 # Submit Button
 # =========================
-# SUBMIT BUTTON
+# =========================
+# Submit Button
+# =========================
 
-if st.button(" Submit Profile"):
- 
-    if not name:
+if st.button("Submit Profile"):
 
+    if not name:
         st.error("Please enter your name.")
- 
-    elif height <= 0 or weight <= 0:
 
+    elif height_cm <= 0 or weight_kg <= 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)
-
         equipment_list = ", ".join(equipment)
- 
-        prompt = f"""
 
-prompt = f"""
+        prompt = f"""
 You are a certified professional fitness trainer.
- 
+
 Generate a structured 5-day workout plan based on the following user profile.
- 
+
 User Profile:
 - Gender: {gender}
 - BMI: {bmi:.2f} ({bmi_status})
-- Goal: {goal}
+- Goal: {fitness_goal}
 - Fitness Level: {fitness_level}
 - Available Equipment: {equipment_list}
- 
+
 Instructions:
 1. Divide the plan clearly into Day 1, Day 2, Day 3, Day 4, and Day 5.
 2. Under each day, list 4-6 exercises.
@@ -149,37 +144,22 @@ Instructions:
 5. Avoid medical or dietary advice.
 6. Keep the output clean, structured, and easy to read.
 7. Do NOT include explanations outside the workout plan.
- 
-Format Example:
- 
-Day 1: Upper Body Strength
-1. Push-Ups  
-   Sets: 3  
-   Reps: 12  
-   Rest: 60 seconds  
- 
+
 Only return the workout plan.
 """
- 
+
         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=600,
-
                 temperature=0.7,
-
                 do_sample=True
-
             )
- 
+
             result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
- 
-        st.subheader(" Your Personalized Workout Plan")
 
+        st.subheader("Your Personalized Workout Plan")
         st.write(result)
-