Update app.py

app.py

@@ -107,7 +107,7 @@ if not st.session_state.logged_in:
 
     if st.button("Verify OTP"):
 
-        if user_otp == st.session_state.otp:
+       if user_otp and user_otp == st.session_state.otp:
 
             token = create_jwt(email)
 
@@ -165,10 +165,25 @@ if st.button("Generate 5-Day Plan 💪"):
         result = query_model(prompt)
 
     st.markdown("## 🗓️ Your 5-Day Workout Plan")
+
+    if isinstance(result, dict):
+
+        for day, exercises in result.items():
+
+          st.subheader(day)
+
+          for ex in exercises:
+            st.write(
+                f"🏋️ **{ex['exercise']}**  |  {ex['sets']} sets  |  {ex['reps']} reps  |  {ex['rest']} rest"
+            )
+
+    else:
+    st.error("Model returned invalid workout format")
     st.write(result)
 
 
 # ---------- LOGOUT ----------
 if st.button("Logout"):
     st.session_state.logged_in = False
+    st.session_state.token = None
     st.rerun()