Update src/streamlit_app.py

src/streamlit_app.py

@@ -29,7 +29,7 @@ def get_category(bmi):
 @st.cache_resource
 def load_model():
     return pipeline(
-        "text-generation",
+        "text2text-generation",  # Correct for FLAN-T5
         model="google/flan-t5-base"
     )
 
@@ -40,6 +40,7 @@ generator = load_model()
 # -------------------------
 st.title("💪 FitPlan-AI: Personalized Fitness Profile")
 
+# Form to collect all inputs
 with st.form("fitness_form"):
 
     st.subheader("Personal Information")
@@ -62,9 +63,14 @@ with st.form("fitness_form"):
         ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", "Kettlebell", "Pull-up Bar"]
     )
 
-    submit = st.form_submit_button("Generate Workout Plan")
+    # -------------------------
+    # SUBMIT BUTTON AT BOTTOM
+    # -------------------------
+    submit = st.form_submit_button("Submit Profile")
 
-    # ✅ ALL LOGIC INSIDE FORM
+    # -------------------------
+    # HANDLE SUBMISSION
+    # -------------------------
     if submit:
         # Validation
         if not name.strip():
@@ -84,6 +90,7 @@ with st.form("fitness_form"):
 
             equipment_list = ", ".join(equipment)
 
+            # Prepare AI prompt
             prompt = f"""
             Create a professional 5-day structured workout plan.
             User Details: