Update app.py

app.py

@@ -1,222 +1,230 @@
 import streamlit as st
-import random
 import smtplib
-import datetime
-from email.mime.text import MIMEText
+import random
+import jwt
+from datetime import datetime, timedelta
+
 from model_api import query_model
 from prompt_builder import build_prompt
 
-# ---------------- EMAIL CONFIG ----------------
-# CHANGE THESE 3 VALUES
+# ==============================
+# CONFIG
+# ==============================
 
-SENDER_EMAIL = "arumugaselvis61@gmail.com"
-SENDER_PASSWORD = "luassmflirakjctw"
 SECRET_KEY = "646d316733839e58469287c95a003062d6d01ca042d457b59517400d26068549"
 
-# ---------------- PAGE CONFIG ----------------
-
-st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
-
-# ---------------- SESSION STATE ----------------
-
-if "users" not in st.session_state:
-    st.session_state.users = {}
-
-if "otp_store" not in st.session_state:
-    st.session_state.otp_store = {}
+SENDER_EMAIL = "arumugaselvis61@gmail.com"
+SENDER_PASSWORD = "luas smfl irak jctw"
 
-if "authenticated" not in st.session_state:
-    st.session_state.authenticated = False
+# ==============================
+# SESSION STATE
+# ==============================
 
 if "page" not in st.session_state:
-    st.session_state.page = "login"
+    st.session_state.page = "home"
 
-# ---------------- OTP GENERATION ----------------
+if "otp" not in st.session_state:
+    st.session_state.otp = None
 
-def generate_otp():
-    return str(random.randint(100000, 999999))
+if "email" not in st.session_state:
+    st.session_state.email = None
 
+if "verified" not in st.session_state:
+    st.session_state.verified = False
 
-# ---------------- SEND EMAIL ----------------
+if "token" not in st.session_state:
+    st.session_state.token = None
+
+# STORE USERS HERE (instead of dictionary)
+if "users" not in st.session_state:
+    st.session_state.users = {}
 
-def send_otp_email(receiver_email, otp):
 
-    try:
+# ==============================
+# SEND OTP
+# ==============================
 
-        msg = MIMEText(f"""
-Hello,
+def send_otp(email):
 
-Your FitPlan AI verification OTP is: {otp}
+    otp = str(random.randint(100000, 999999))
 
-This OTP will expire in 10 minutes.
-""")
+    st.session_state.otp = otp
+    st.session_state.email = email
 
-        msg["Subject"] = "FitPlan AI OTP"
-        msg["From"] = SENDER_EMAIL
-        msg["To"] = receiver_email
+    subject = "FitPlan AI OTP"
+    message = f"Subject:{subject}\n\nYour OTP is {otp}"
 
+    try:
         server = smtplib.SMTP("smtp.gmail.com", 587)
         server.starttls()
         server.login(SENDER_EMAIL, SENDER_PASSWORD)
-
-        server.sendmail(SENDER_EMAIL, receiver_email, msg.as_string())
+        server.sendmail(SENDER_EMAIL, email, message)
         server.quit()
 
         return True
 
     except Exception as e:
-        st.error(f"Email sending failed: {e}")
+        st.error(f"Email error: {e}")
         return False
 
 
-# =========================================================
-# LOGIN PAGE
-# =========================================================
+# ==============================
+# GENERATE JWT
+# ==============================
 
-if st.session_state.page == "login":
+def generate_token(email):
 
-    st.title("🔐 FitPlan AI Login")
+    payload = {
+        "email": email,
+        "exp": datetime.utcnow() + timedelta(hours=2)
+    }
 
-    email = st.text_input("Email")
+    token = jwt.encode(payload, SECRET_KEY, algorithm="HS256")
 
-    if st.button("Login"):
+    return token
 
-        user = st.session_state.users.get(email)
 
-        if not user:
-            st.error("User not found. Please Sign Up.")
-        else:
-            st.session_state.authenticated = True
-            st.session_state.page = "landing"
-            st.rerun()
+# ==============================
+# VERIFY JWT
+# ==============================
+
+def verify_token(token):
+
+    try:
+        data = jwt.decode(token, SECRET_KEY, algorithms=["HS256"])
+        return data["email"]
 
-    if st.button("New User? Sign Up"):
+    except:
+        return None
+
+
+# ==============================
+# HOME PAGE
+# ==============================
+
+if st.session_state.page == "home":
+
+    st.title("💪 FitPlan AI")
+
+    col1, col2 = st.columns(2)
+
+    if col1.button("New User Sign Up"):
         st.session_state.page = "signup"
         st.rerun()
 
+    if col2.button("Login"):
+        st.session_state.page = "login"
+        st.rerun()
+
 
-# =========================================================
+# ==============================
 # SIGNUP PAGE
-# =========================================================
+# ==============================
 
 elif st.session_state.page == "signup":
 
-    st.title("📝 Sign Up")
+    st.header("Create Account")
 
     email = st.text_input("Enter Email")
 
     if st.button("Send OTP"):
 
-        if not email:
-            st.error("Enter email first")
+        if send_otp(email):
+            st.success("OTP sent to your email")
 
-        else:
+    otp_input = st.text_input("Enter OTP")
 
-            otp = generate_otp()
+    if st.session_state.otp:
 
-            st.session_state.otp_store[email] = {
-                "otp": otp,
-                "expiry": datetime.datetime.utcnow() + datetime.timedelta(minutes=10)
-            }
+        if st.button("Verify OTP"):
 
-            if send_otp_email(email, otp):
+            if otp_input == st.session_state.otp:
 
-                st.success("OTP sent to your email")
+                st.session_state.verified = True
+                st.success("Email Verified")
 
-                st.session_state.current_email = email
-                st.session_state.page = "verify"
-                st.rerun()
+            else:
+                st.error("Invalid OTP")
 
-    if st.button("Back to Login"):
-        st.session_state.page = "login"
-        st.rerun()
+    if st.session_state.verified:
 
+        password = st.text_input("Create Password", type="password")
 
-# =========================================================
-# OTP VERIFY PAGE
-# =========================================================
+        if st.button("Register"):
 
-elif st.session_state.page == "verify":
+            # SAVE USER
+            st.session_state.users[st.session_state.email] = password
 
-    st.title("🔢 Verify OTP")
+            st.success("Account created successfully!")
 
-    otp_input = st.text_input("Enter OTP")
+            st.session_state.page = "login"
+            st.rerun()
 
-    if st.button("Verify OTP"):
+    if st.button("Back"):
+        st.session_state.page = "home"
+        st.rerun()
+
+
+# ==============================
+# LOGIN PAGE
+# ==============================
 
-        email = st.session_state.current_email
-        data = st.session_state.otp_store.get(email)
+elif st.session_state.page == "login":
 
-        if not data:
-            st.error("Session expired")
+    st.header("Login")
 
-        elif datetime.datetime.utcnow() > data["expiry"]:
-            st.error("OTP expired")
+    email = st.text_input("Email")
+    password = st.text_input("Password", type="password")
 
-        elif otp_input == data["otp"]:
+    if st.button("Login"):
 
-            st.session_state.users[email] = True
+        if email in st.session_state.users and st.session_state.users[email] == password:
 
-            st.success("Signup successful. Please login.")
+            token = generate_token(email)
 
-            st.session_state.page = "login"
+            st.session_state.token = token
+            st.session_state.page = "dashboard"
+
+            st.success("Login successful")
             st.rerun()
 
         else:
-            st.error("Invalid OTP")
+            st.error("Invalid email or password")
 
+    if st.button("Back"):
+        st.session_state.page = "home"
+        st.rerun()
 
-# =========================================================
-# LANDING PAGE
-# =========================================================
 
-elif st.session_state.page == "landing" and st.session_state.authenticated:
+# ==============================
+# DASHBOARD
+# ==============================
 
-    st.markdown("""
-    <style>
-    [data-testid="stAppViewContainer"] {
-        background-image: url("https://images.unsplash.com/photo-1483721310020-03333e577078");
-        background-size: cover;
-        background-position: center;
-    }
-    .center-box {
-        text-align:center;
-        margin-top:200px;
-        font-size:48px;
-        font-weight:bold;
-        color:black;
-    }
-    </style>
-    """, unsafe_allow_html=True)
+elif st.session_state.page == "dashboard":
 
-    st.markdown('<div class="center-box">💪 Welcome to FitPlan AI</div>', unsafe_allow_html=True)
+    email = verify_token(st.session_state.token)
 
-    if st.button("🚀 Get Started"):
-        st.session_state.page = "main"
-        st.rerun()
-
-    if st.button("Logout"):
-        st.session_state.authenticated = False
+    if not email:
+        st.error("Session expired. Login again.")
         st.session_state.page = "login"
         st.rerun()
 
+    st.title("🏋 Welcome to FitPlan AI")
 
-# =========================================================
-# PROFILE PAGE
-# =========================================================
+    st.write(f"Logged in as: {email}")
 
-elif st.session_state.page == "main":
-
-    st.title("💪 FitPlan AI")
+    # ---------------- USER PROFILE ----------------
 
     name = st.text_input("Name")
-    age = st.number_input("Age", min_value=10, max_value=100)
+    age = st.number_input("Age", min_value=10, max_value=80)
+
     gender = st.selectbox("Gender", ["Male", "Female", "Other"])
-    height_cm = st.number_input("Height (cm)")
-    weight_kg = st.number_input("Weight (kg)")
+
+    height = st.number_input("Height (cm)")
+    weight = st.number_input("Weight (kg)")
 
     goal = st.selectbox(
-        "Fitness Goal",
+        "Goal",
         ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building"]
     )
 
@@ -230,34 +238,33 @@ elif st.session_state.page == "main":
         ["Beginner", "Intermediate", "Advanced"]
     )
 
-    if st.button("Generate Plan"):
+    # ---------------- GENERATE PLAN ----------------
+
+    if st.button("Generate FitPlan"):
 
         prompt, bmi, bmi_status = build_prompt(
-            name, age, gender, height_cm, weight_kg,
-            goal, fitness_level, equipment
+            name,
+            age,
+            gender,
+            height,
+            weight,
+            goal,
+            fitness_level,
+            equipment
         )
 
-        with st.spinner("Generating Workout Plan..."):
+        with st.spinner("Generating AI Workout Plan..."):
             result = query_model(prompt)
 
-        st.session_state.workout_plan = result
-        st.session_state.bmi = bmi
-        st.session_state.bmi_status = bmi_status
-        st.session_state.page = "result"
-        st.rerun()
-
+        st.subheader("Your 5-Day Workout Plan")
 
-# =========================================================
-# RESULT PAGE
-# =========================================================
+        st.write(result)
 
-elif st.session_state.page == "result":
+        st.info(f"BMI: {bmi:.2f} ({bmi_status})")
 
-    st.title("🏋️ Your Workout Plan")
+    if st.button("Logout"):
 
-    st.markdown(st.session_state.workout_plan)
-    st.info(f"BMI: {st.session_state.bmi:.2f} ({st.session_state.bmi_status})")
+        st.session_state.token = None
+        st.session_state.page = "home"
 
-    if st.button("Back"):
-        st.session_state.page = "main"
         st.rerun()