Update app.py

app.py

@@ -1,3 +1,4 @@
+```python
 import streamlit as st
 
 from database import create_tables, save_weight, get_weight_history, save_otp
@@ -7,7 +8,10 @@ from model_api import query_model
 from prompt_builder import build_prompt
 
 
-# Create database tables automatically
+# -------------------------
+# DATABASE INIT
+# -------------------------
+
 create_tables()
 
 st.set_page_config(page_title="FitPlan AI", layout="centered")
@@ -43,7 +47,7 @@ menu = st.sidebar.selectbox(
 
 
 # -------------------------
-# SIGNUP
+# SIGNUP PAGE
 # -------------------------
 
 if menu == "Signup":
@@ -64,7 +68,7 @@ if menu == "Signup":
 
 
 # -------------------------
-# LOGIN
+# LOGIN PAGE
 # -------------------------
 
 if menu == "Login" and st.session_state.user is None:
@@ -74,7 +78,6 @@ if menu == "Login" and st.session_state.user is None:
     email = st.text_input("Email")
     password = st.text_input("Password", type="password")
 
-    # Step 1: Verify credentials
     if st.button("Login"):
 
         user = login_user(email, password)
@@ -110,7 +113,7 @@ if st.session_state.credentials_verified and not st.session_state.otp_sent:
 
 
 # -------------------------
-# VERIFY OTP
+# VERIFY + RESEND OTP
 # -------------------------
 
 if st.session_state.otp_sent:
@@ -119,20 +122,41 @@ if st.session_state.otp_sent:
 
     otp_input = st.text_input("OTP")
 
-    if st.button("Verify OTP"):
+    col1, col2 = st.columns(2)
 
-        verified = verify_user_otp(st.session_state.email, otp_input)
+    # VERIFY OTP
+    with col1:
 
-        if verified:
+        if st.button("Verify OTP"):
 
-            st.session_state.user = st.session_state.email
-            st.session_state.credentials_verified = False
-            st.session_state.otp_sent = False
+            verified = verify_user_otp(
+                st.session_state.email,
+                otp_input
+            )
 
-            st.success("Login successful")
+            if verified:
 
-        else:
-            st.error("Invalid OTP")
+                st.session_state.user = st.session_state.email
+                st.session_state.credentials_verified = False
+                st.session_state.otp_sent = False
+
+                st.success("Login successful")
+
+            else:
+                st.error("Invalid OTP")
+
+    # RESEND OTP
+    with col2:
+
+        if st.button("Resend OTP"):
+
+            otp = generate_otp()
+
+            save_otp(st.session_state.email, otp)
+
+            send_otp_email(st.session_state.email, otp)
+
+            st.success("New OTP sent to your email")
 
 
 # -------------------------
@@ -148,12 +172,18 @@ if st.session_state.user:
         ["Dashboard", "Weight Tracker", "Generate Fitness Plan"]
     )
 
-    # Dashboard
+
+    # -------------------------
+    # DASHBOARD PAGE
+    # -------------------------
+
     if page == "Dashboard":
 
         st.header("Welcome to FitPlan AI")
 
-        st.write("Track your fitness progress and generate AI fitness plans.")
+        st.write(
+            "Track your fitness progress and generate AI workout plans."
+        )
 
 
     # -------------------------
@@ -164,21 +194,32 @@ if st.session_state.user:
 
         st.header("Weight Tracker")
 
-        weight = st.number_input("Enter today's weight (kg)", min_value=30.0)
+        weight = st.number_input(
+            "Enter today's weight (kg)",
+            min_value=30.0
+        )
 
         if st.button("Save Weight"):
 
-            save_weight(st.session_state.user, weight)
+            save_weight(
+                st.session_state.user,
+                weight
+            )
 
             st.success("Weight saved successfully")
 
         st.subheader("Weight History")
 
-        history = get_weight_history(st.session_state.user)
+        history = get_weight_history(
+            st.session_state.user
+        )
 
         if history:
+
             st.table(history)
+
         else:
+
             st.info("No weight data available")
 
 
@@ -190,23 +231,51 @@ if st.session_state.user:
 
         st.header("Generate Personalized Fitness Plan")
 
-        age = st.number_input("Age", 10, 80)
-        height = st.number_input("Height (cm)", 100, 220)
-        weight = st.number_input("Weight (kg)", 30, 150)
+        age = st.number_input(
+            "Age",
+            10,
+            80
+        )
+
+        height = st.number_input(
+            "Height (cm)",
+            100,
+            220
+        )
+
+        weight = st.number_input(
+            "Weight (kg)",
+            30,
+            150
+        )
 
         goal = st.selectbox(
             "Fitness Goal",
-            ["Weight Loss", "Muscle Gain", "Maintain Fitness"]
+            [
+                "Weight Loss",
+                "Muscle Gain",
+                "Maintain Fitness"
+            ]
         )
 
         activity = st.selectbox(
             "Activity Level",
-            ["Beginner", "Intermediate", "Advanced"]
+            [
+                "Beginner",
+                "Intermediate",
+                "Advanced"
+            ]
         )
 
         if st.button("Generate Plan"):
 
-            prompt = build_prompt(age, height, weight, goal, activity)
+            prompt = build_prompt(
+                age,
+                height,
+                weight,
+                goal,
+                activity
+            )
 
             with st.spinner("Generating AI plan..."):
 
@@ -228,4 +297,5 @@ if st.session_state.user:
         st.session_state.credentials_verified = False
         st.session_state.otp_sent = False
 
-        st.rerun()
+        st.rerun()
+```