Update app.py

app.py

@@ -1,18 +1,25 @@
 import streamlit as st
-from database import create_tables, register_user, login_user, save_weight, get_weight_history
+from auth import signup_user, login_user, verify_user_otp
+from email_utils import generate_otp, send_otp_email
+from database import save_weight, get_weight_history
 from model_api import query_model
 from prompt_builder import build_prompt
 
-# ---------------- INIT DB ----------------
-create_tables()
-
 # ---------------- PAGE CONFIG ----------------
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-# ---------------- SESSION ----------------
+st.title("🏋️ FitPlan AI")
+
+# ---------------- SESSION STATE ----------------
 if "user" not in st.session_state:
     st.session_state.user = None
 
+if "otp_sent" not in st.session_state:
+    st.session_state.otp_sent = False
+
+if "otp_email" not in st.session_state:
+    st.session_state.otp_email = None
+
 
 # =========================================================
 # LOGIN / SIGNUP PAGE
@@ -20,54 +27,92 @@ if "user" not in st.session_state:
 
 if st.session_state.user is None:
 
-    st.title("🏋️ FitPlan AI Login")
-
     menu = ["Login", "Signup"]
     choice = st.sidebar.selectbox("Menu", menu)
 
+    st.subheader(choice)
+
     email = st.text_input("Email")
     password = st.text_input("Password", type="password")
 
     # ---------------- SIGNUP ----------------
     if choice == "Signup":
 
-        if st.button("Create Account"):
+        if st.button("Register"):
+
+            success = signup_user(email, password)
+
+            if success:
+
+                otp = generate_otp()
+
+                send_otp_email(email, otp)
+
+                st.session_state.otp = otp
+                st.session_state.otp_email = email
+                st.session_state.otp_sent = True
+
+                st.success("OTP sent to email")
 
-            if register_user(email, password):
-                st.success("Account created successfully")
             else:
                 st.error("User already exists")
 
     # ---------------- LOGIN ----------------
     if choice == "Login":
 
-        if st.button("Login"):
+        if st.button("Send OTP"):
 
             user = login_user(email, password)
 
             if user:
-                st.session_state.user = email
+
+                otp = generate_otp()
+
+                send_otp_email(email, otp)
+
+                st.session_state.otp = otp
+                st.session_state.otp_email = email
+                st.session_state.otp_sent = True
+
+                st.success("OTP sent to your email")
+
+            else:
+                st.error("Invalid email or password")
+
+
+    # ---------------- OTP VERIFICATION ----------------
+
+    if st.session_state.otp_sent:
+
+        otp_input = st.text_input("Enter OTP")
+
+        if st.button("Verify OTP"):
+
+            if otp_input == st.session_state.otp:
+
+                st.session_state.user = st.session_state.otp_email
+                st.session_state.otp_sent = False
+
                 st.success("Login successful")
                 st.rerun()
+
             else:
-                st.error("Invalid email or password")
+                st.error("Invalid OTP")
 
 
 # =========================================================
-# MAIN APP
+# MAIN APPLICATION
 # =========================================================
 
 else:
 
-    st.title("🏋️ FitPlan AI — Personalized Workout Generator")
-
     st.sidebar.success(f"Logged in as {st.session_state.user}")
 
     if st.sidebar.button("Logout"):
         st.session_state.user = None
         st.rerun()
 
-    st.write("Fill your fitness details to generate a workout plan.")
+    st.header("Generate Your Workout Plan")
 
     # ---------------- USER INPUT ----------------
 
@@ -132,15 +177,15 @@ else:
 
     if st.button("View Weight History"):
 
-        data = get_weight_history(st.session_state.user)
+        history = get_weight_history(st.session_state.user)
 
-        if data:
-            st.write("📊 Weight History")
-            st.table(data)
+        if history:
+            st.table(history)
         else:
-            st.info("No weight history available")
+            st.info("No history available")
 
-    # ---------------- GENERATE WORKOUT ----------------
+
+    # ---------------- GENERATE WORKOUT PLAN ----------------
 
     if st.button("Generate Workout Plan"):
 
@@ -148,7 +193,7 @@ else:
             st.warning("Please enter your name")
 
         elif not equipment:
-            st.warning("Select equipment")
+            st.warning("Select at least one equipment")
 
         else:
 
@@ -163,20 +208,23 @@ else:
                 equipment
             )
 
-            with st.spinner("Generating AI workout plan..."):
+            with st.spinner("Generating workout plan..."):
                 response = query_model(prompt)
 
-            st.subheader("📋 Your Workout Plan")
+            st.subheader("📋 Your Personalized Workout Plan")
+
             st.write(response)
 
-            st.info(f"""
-            **Profile Summary**
+            st.info(
+                f"""
+                **Profile Summary**
 
-            BMI: {bmi:.2f} ({bmi_status})
+                BMI: {bmi:.2f} ({bmi_status})
 
-            Goal: {goal}
+                Goal: {goal}
 
-            Fitness Level: {fitness_level}
+                Fitness Level: {fitness_level}
 
-            Equipment: {", ".join(equipment)}
-            """)
+                Equipment: {", ".join(equipment)}
+                """
+            )