Update app.py

app.py

@@ -1,229 +1,216 @@
 import streamlit as st
+
+from database import create_tables, save_weight, get_weight_history
 from auth import signup_user, login_user, verify_user_otp, generate_otp
 from email_utils import send_otp_email
-from database import save_weight, get_weight_history
 from model_api import query_model
 from prompt_builder import build_prompt
-# ---------------- PAGE CONFIG ----------------
+
+
+# create tables automatically
+create_tables()
+
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-st.title("🏋️ FitPlan AI")
+st.title("🏋️ FitPlan AI - Personalized Fitness Planner")
 
-# ---------------- SESSION STATE ----------------
+# 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
+if "email" not in st.session_state:
+    st.session_state.email = None
 
 
-# =========================================================
+# -------------------------
 # LOGIN / SIGNUP PAGE
-# =========================================================
+# -------------------------
+
+menu = st.sidebar.selectbox("Menu", ["Login", "Signup"])
 
-if st.session_state.user is None:
 
-    menu = ["Login", "Signup"]
-    choice = st.sidebar.selectbox("Menu", menu)
+# -------------------------
+# SIGNUP
+# -------------------------
 
-    st.subheader(choice)
+if menu == "Signup":
+
+    st.subheader("Create Account")
 
     email = st.text_input("Email")
     password = st.text_input("Password", type="password")
 
-    # ---------------- SIGNUP ----------------
-    if choice == "Signup":
+    if st.button("Signup"):
 
-        if st.button("Register"):
+        success = signup_user(email, password)
 
-            success = signup_user(email, password)
+        if success:
+            st.success("Account created successfully. Please login.")
+        else:
+            st.error("User already exists")
 
-            if success:
 
-                otp = generate_otp()
+# -------------------------
+# LOGIN
+# -------------------------
 
-                send_otp_email(email, otp)
+if menu == "Login" and st.session_state.user is None:
 
-                st.session_state.otp = otp
-                st.session_state.otp_email = email
-                st.session_state.otp_sent = True
+    st.subheader("Login")
 
-                st.success("OTP sent to email")
+    email = st.text_input("Email")
+    password = st.text_input("Password", type="password")
 
-            else:
-                st.error("User already exists")
+    if st.button("Send OTP"):
 
-    # ---------------- LOGIN ----------------
-    if choice == "Login":
+        user = login_user(email, password)
 
-        if st.button("Send OTP"):
+        if user:
 
-            user = login_user(email, password)
+            otp = generate_otp()
 
-            if user:
+            send_otp_email(email, otp)
 
-                otp = generate_otp()
+            from database import save_otp
+            save_otp(email, otp)
 
-                send_otp_email(email, otp)
+            st.session_state.otp_sent = True
+            st.session_state.email = email
 
-                st.session_state.otp = otp
-                st.session_state.otp_email = email
-                st.session_state.otp_sent = True
+            st.success("OTP sent to your email")
 
-                st.success("OTP sent to your email")
+        else:
+            st.error("Invalid email or password")
+
+
+
+# -------------------------
+# OTP VERIFICATION
+# -------------------------
 
-            else:
-                st.error("Invalid email or password")
+if st.session_state.otp_sent:
 
+    st.subheader("Verify OTP")
 
-    # ---------------- OTP VERIFICATION ----------------
+    otp_input = st.text_input("Enter OTP")
 
-    if st.session_state.otp_sent:
+    if st.button("Verify OTP"):
 
-        otp_input = st.text_input("Enter OTP")
+        verified = verify_user_otp(st.session_state.email, otp_input)
 
-        if st.button("Verify OTP"):
+        if verified:
 
-            if otp_input == st.session_state.otp:
+            st.session_state.user = st.session_state.email
+            st.session_state.otp_sent = False
 
-                st.session_state.user = st.session_state.otp_email
-                st.session_state.otp_sent = False
+            st.success("Login successful")
 
-                st.success("Login successful")
-                st.rerun()
+        else:
+            st.error("Invalid OTP")
 
-            else:
-                st.error("Invalid OTP")
 
 
-# =========================================================
-# MAIN APPLICATION
-# =========================================================
+# -------------------------
+# MAIN DASHBOARD
+# -------------------------
 
-else:
+if st.session_state.user:
 
     st.sidebar.success(f"Logged in as {st.session_state.user}")
 
-    if st.sidebar.button("Logout"):
-        st.session_state.user = None
-        st.rerun()
+    page = st.sidebar.selectbox(
+        "Navigation",
+        ["Dashboard", "Weight Tracker", "Generate Fitness Plan"]
+    )
 
-    st.header("Generate Your Workout Plan")
 
-    # ---------------- USER INPUT ----------------
 
-    name = st.text_input("Name")
+    # -------------------------
+    # DASHBOARD
+    # -------------------------
 
-    age = st.number_input(
-        "Age",
-        min_value=10,
-        max_value=100
-    )
+    if page == "Dashboard":
 
-    gender = st.radio("Gender", ["Male", "Female"])
+        st.header("Welcome to FitPlan AI")
 
-    height = st.number_input(
-        "Height (cm)",
-        min_value=100.0,
-        max_value=250.0
-    )
+        st.write("Track your fitness journey and generate personalized plans.")
 
-    weight = st.number_input(
-        "Weight (kg)",
-        min_value=30.0,
-        max_value=200.0
-    )
 
-    goal = st.selectbox(
-        "Fitness Goal",
-        [
-            "Build Muscle",
-            "Lose Weight",
-            "Improve Endurance",
-            "General Fitness"
-        ]
-    )
 
-    fitness_level = st.radio(
-        "Fitness Level",
-        ["Beginner", "Intermediate", "Advanced"]
-    )
+    # -------------------------
+    # WEIGHT TRACKER
+    # -------------------------
 
-    equipment = st.multiselect(
-        "Available Equipment",
-        [
-            "No Equipment",
-            "Dumbbells",
-            "Barbell",
-            "Pull-up Bar",
-            "Resistance Bands",
-            "Treadmill",
-            "Kettlebells",
-            "Full Gym"
-        ]
-    )
+    if page == "Weight Tracker":
+
+        st.header("Weight Tracker")
 
-    # ---------------- SAVE WEIGHT ----------------
+        weight = st.number_input("Enter today's weight (kg)", min_value=30.0)
 
-    if st.button("Save Today's Weight"):
-        save_weight(st.session_state.user, weight)
-        st.success("Weight saved")
+        if st.button("Save Weight"):
 
-    # ---------------- WEIGHT HISTORY ----------------
+            save_weight(st.session_state.user, weight)
 
-    if st.button("View Weight History"):
+            st.success("Weight saved successfully")
+
+        st.subheader("Weight History")
 
         history = get_weight_history(st.session_state.user)
 
         if history:
             st.table(history)
         else:
-            st.info("No history available")
+            st.info("No weight history found")
 
 
-    # ---------------- GENERATE WORKOUT PLAN ----------------
 
-    if st.button("Generate Workout Plan"):
+    # -------------------------
+    # AI FITNESS PLAN
+    # -------------------------
 
-        if not name:
-            st.warning("Please enter your name")
+    if page == "Generate Fitness Plan":
 
-        elif not equipment:
-            st.warning("Select at least one equipment")
+        st.header("Generate Personalized Fitness Plan")
 
-        else:
+        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"]
+        )
+
+        activity = st.selectbox(
+            "Activity Level",
+            ["Beginner", "Intermediate", "Advanced"]
+        )
+
+        if st.button("Generate Plan"):
+
+            prompt = build_prompt(age, height, weight, goal, activity)
+
+            with st.spinner("Generating plan..."):
 
-            prompt, bmi, bmi_status = build_prompt(
-                name,
-                age,
-                gender,
-                height,
-                weight,
-                goal,
-                fitness_level,
-                equipment
-            )
-
-            with st.spinner("Generating workout plan..."):
                 response = query_model(prompt)
 
-            st.subheader("📋 Your Personalized Workout Plan")
+            st.subheader("Your AI Fitness Plan")
 
             st.write(response)
 
-            st.info(
-                f"""
-                **Profile Summary**
 
-                BMI: {bmi:.2f} ({bmi_status})
 
-                Goal: {goal}
+    # -------------------------
+    # LOGOUT
+    # -------------------------
+
+    if st.sidebar.button("Logout"):
 
-                Fitness Level: {fitness_level}
+        st.session_state.user = None
+        st.session_state.email = None
+        st.session_state.otp_sent = False
 
-                Equipment: {", ".join(equipment)}
-                """
-            )
+        st.experimental_rerun()