Update app.py

app.py

@@ -9,10 +9,27 @@ st.set_page_config(
     layout="centered"
 )
 
+# Helper function to verify workout plan - MUST be defined before it's used
+def verify_workout_plan(plan_text):
+    """Check if the workout plan contains all 5 days"""
+    if not plan_text or plan_text.startswith("Error"):
+        return False, "No valid plan generated"
+    
+    days_present = []
+    for i in range(1, 6):
+        if f"Day {i}" in plan_text:
+            days_present.append(i)
+    
+    if len(days_present) == 5:
+        return True, "Complete 5-day plan"
+    else:
+        missing_days = set(range(1,6)) - set(days_present)
+        return False, f"Missing day(s): {', '.join(map(str, missing_days))}"
+
 st.title("💪 FIT Plan AI – Personalized Fitness Profile")
 st.markdown("---")
 
-# Test API connection on startup (optional)
+# Test API connection on startup
 if "api_status" not in st.session_state:
     with st.spinner("Checking API connection..."):
         api_ok, api_message = test_api_connection()
@@ -210,13 +227,14 @@ with tab2:
                     # Call the model API from model_api.py
                     workout_plan = query_model(st.session_state.prompt)
                     st.session_state.workout_plan = workout_plan
+                    st.rerun()
                 
         # Display workout plan if generated
         if st.session_state.workout_plan:
             st.markdown("---")
             st.subheader("📋 Your 5-Day Workout Plan")
             
-            # Check if plan is complete
+            # Check if plan is complete - Now verify_workout_plan is defined
             is_complete, message = verify_workout_plan(st.session_state.workout_plan)
             if not is_complete:
                 st.warning(f"⚠️ {message}. You can regenerate the plan if needed.")
@@ -291,21 +309,4 @@ Generated by FIT Plan AI
 
 # Footer
 st.markdown("---")
-st.caption("FIT Plan AI - Milestone 1 | Powered by Mistral-7B")
-
-# Helper function to verify workout plan
-def verify_workout_plan(plan_text):
-    """Check if the workout plan contains all 5 days"""
-    if not plan_text or plan_text.startswith("Error"):
-        return False, "No valid plan generated"
-    
-    days_present = []
-    for i in range(1, 6):
-        if f"Day {i}" in plan_text:
-            days_present.append(i)
-    
-    if len(days_present) == 5:
-        return True, "Complete 5-day plan"
-    else:
-        missing_days = set(range(1,6)) - set(days_present)
-        return False, f"Missing day(s): {', '.join(map(str, missing_days))}"
+st.caption("FIT Plan AI - Milestone 1 | Powered by Mistral-7B")