Update app.py

app.py

@@ -1,75 +1,182 @@
 import streamlit as st
-from auth import generate_otp, create_jwt, verify_jwt
-from email_utils import send_otp_email  
-from dotenv import load_dotenv
-load_dotenv()
+from database import create_tables, register_user, login_user, save_weight, get_weight_history
+from model_api import query_model
+from prompt_builder import build_prompt
 
-st.set_page_config(page_title="FitPlan AI Login")
+# ---------------- INIT DB ----------------
+create_tables()
 
-# Initialize session state
-if "otp" not in st.session_state:
-    st.session_state.otp = None
+# ---------------- PAGE CONFIG ----------------
+st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-if "authenticated" not in st.session_state:
-    st.session_state.authenticated = False
+# ---------------- SESSION ----------------
+if "user" not in st.session_state:
+    st.session_state.user = None
 
-if "token" not in st.session_state:
-    st.session_state.token = None
 
+# =========================================================
+# LOGIN / SIGNUP PAGE
+# =========================================================
 
-st.title("🔐 FitPlan AI - OTP Login")
+if st.session_state.user is None:
 
-# ------------------------
-# STEP 1: Enter Email
-# ------------------------
-if not st.session_state.authenticated:
+    st.title("🏋️ FitPlan AI Login")
 
-    email = st.text_input("Enter your Email")
+    menu = ["Login", "Signup"]
+    choice = st.sidebar.selectbox("Menu", menu)
 
-    if st.button("Send OTP"):
-        if email:
-            otp = generate_otp()
-            st.session_state.otp = otp
-            st.session_state.email = email
-            
-            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")
-
-        if st.button("Verify OTP"):
-            if entered_otp == 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!")
+    email = st.text_input("Email")
+    password = st.text_input("Password", type="password")
+
+    # ---------------- SIGNUP ----------------
+    if choice == "Signup":
+
+        if st.button("Create Account"):
+
+            if register_user(email, password):
+                st.success("Account created successfully")
+            else:
+                st.error("User already exists")
+
+    # ---------------- LOGIN ----------------
+    if choice == "Login":
+
+        if st.button("Login"):
+
+            user = login_user(email, password)
+
+            if user:
+                st.session_state.user = email
+                st.success("Login successful")
                 st.rerun()
             else:
-                st.error("Invalid OTP")
+                st.error("Invalid email or password")
 
 
-# ------------------------
-# Protected Page
-# ------------------------
+# =========================================================
+# MAIN APP
+# =========================================================
+
 else:
-    decoded = verify_jwt(st.session_state.token)
 
-    if decoded:
-        st.success(f"Welcome {decoded['email']} 🎉")
+    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.")
+
+    # ---------------- USER INPUT ----------------
+
+    name = st.text_input("Name")
+
+    age = st.number_input(
+        "Age",
+        min_value=10,
+        max_value=100
+    )
+
+    gender = st.radio("Gender", ["Male", "Female"])
+
+    height = st.number_input(
+        "Height (cm)",
+        min_value=100.0,
+        max_value=250.0
+    )
+
+    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"]
+    )
+
+    equipment = st.multiselect(
+        "Available Equipment",
+        [
+            "No Equipment",
+            "Dumbbells",
+            "Barbell",
+            "Pull-up Bar",
+            "Resistance Bands",
+            "Treadmill",
+            "Kettlebells",
+            "Full Gym"
+        ]
+    )
+
+    # ---------------- SAVE WEIGHT ----------------
+
+    if st.button("Save Today's Weight"):
+        save_weight(st.session_state.user, weight)
+        st.success("Weight saved")
+
+    # ---------------- WEIGHT HISTORY ----------------
+
+    if st.button("View Weight History"):
+
+        data = get_weight_history(st.session_state.user)
+
+        if data:
+            st.write("📊 Weight History")
+            st.table(data)
+        else:
+            st.info("No weight history available")
+
+    # ---------------- GENERATE WORKOUT ----------------
+
+    if st.button("Generate Workout Plan"):
+
+        if not name:
+            st.warning("Please enter your name")
+
+        elif not equipment:
+            st.warning("Select equipment")
+
+        else:
+
+            prompt, bmi, bmi_status = build_prompt(
+                name,
+                age,
+                gender,
+                height,
+                weight,
+                goal,
+                fitness_level,
+                equipment
+            )
+
+            with st.spinner("Generating AI workout plan..."):
+                response = query_model(prompt)
+
+            st.subheader("📋 Your Workout Plan")
+            st.write(response)
+
+            st.info(f"""
+            **Profile Summary**
+
+            BMI: {bmi:.2f} ({bmi_status})
+
+            Goal: {goal}
 
-        st.write("You can now access FitPlan AI")
+            Fitness Level: {fitness_level}
 
-        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")
+            Equipment: {", ".join(equipment)}
+            """)