Update app.py

app.py

@@ -1,90 +1,83 @@
 import streamlit as st
 from auth import generate_otp, create_jwt, verify_jwt
-from emails_utils import send_otp_email 
+from emails_utils import send_otp_email
 from dotenv import load_dotenv
 from model_api import query_model
 from prompt_builder import build_prompt
 
-st.set_page_config(page_title="FitPlan AI Login")
+load_dotenv()
+
+st.set_page_config(page_title="FitPlan AI", page_icon="🏋️", layout="centered")
+
+# ---------------- SESSION STATE ---------------- #
+
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
+
+if "otp_sent" not in st.session_state:
+    st.session_state.otp_sent = False
 
-# Initialize session state
 if "otp" not in st.session_state:
     st.session_state.otp = None
 
-if "authenticated" not in st.session_state:
-    st.session_state.authenticated = False
-
-if "token" not in st.session_state:
-    st.session_state.token = None
 
+# ---------------- LOGIN PAGE ---------------- #
 
-st.title("🔐 FitPlan AI - OTP Login")
+if not st.session_state.logged_in:
 
-# ------------------------
-# STEP 1: Enter Email
-# ------------------------
-if not st.session_state.authenticated:
+    st.title("🔐 FitPlan AI - OTP Login")
 
     email = st.text_input("Enter your Email")
 
-    if st.button("Send OTP"):
-        if email:
+    if not st.session_state.otp_sent:
+
+        if st.button("Send OTP"):
+
             otp = generate_otp()
+
             st.session_state.otp = otp
             st.session_state.email = email
-            
+            st.session_state.otp_sent = True
+
             send_otp_email(email, otp)
-            
-            st.success("OTP sent to your email 📧")
-        else:
-            st.error("Please enter email")
 
-    # ------------------------
-    # STEP 2: Verify OTP
-    # ------------------------
-    if st.session_state.otp:
-        entered_otp = st.text_input("Enter OTP")
+            st.success("OTP sent to your email")
+
+    else:
+
+        otp_input = st.text_input("Enter OTP")
 
         if st.button("Verify OTP"):
-            if entered_otp == st.session_state.otp:
+
+            if otp_input == st.session_state.otp:
+
                 token = create_jwt(st.session_state.email)
-                st.session_state.token = token
-                st.session_state.authenticated = True
-                st.success("Login Successful!")
-                st.rerun()
-            else:
-                st.error("Invalid OTP")
 
+                st.session_state.jwt = token
+                st.session_state.logged_in = True
 
-# ------------------------
-# Protected Page
-# ------------------------
-else:
-    decoded = verify_jwt(st.session_state.token)
+                st.success("Login Successful")
+                st.rerun()
 
-    if decoded:
-        st.success(f"Welcome {decoded['email']} 🎉")
+            else:
+                st.error("Invalid OTP")
 
-        st.write("You can now access FitPlan AI")
+    st.stop()
 
-        if st.button("Logout"):
-            st.session_state.authenticated = False
-            st.session_state.token = None
-            st.session_state.otp = None
-            st.rerun()
-    else:
-        st.error("Session Expired")
 
+# ---------------- APP CONTENT ---------------- #
 
 st.title("🏋️ AI Personalized 5-Day Workout Planner")
 
 name = st.text_input("Name")
-age = st.number_input("Age", 10, 80)
+
+age = st.number_input("Age", 0, 100)
 
 gender = st.selectbox("Gender", ["Male", "Female", "Other"])
 
-height = st.number_input("Height (cm)", 100, 250)
-weight = st.number_input("Weight (kg)", 30, 200)
+height = st.number_input("Height (cm)", 0, 250)
+
+weight = st.number_input("Weight (kg)", 0, 200)
 
 goal = st.selectbox(
     "Fitness Goal",
@@ -116,7 +109,7 @@ if st.button("Generate 5-Day Plan 💪"):
         goal, fitness_level, equipment
     )
 
-    st.subheader(f"BMI: {bmi} ({bmi_status})")
+    st.subheader(f"BMI: {bmi:.2f} ({bmi_status})")
 
     with st.spinner("Generating workout plan..."):
         result = query_model(prompt)