Update app.py

app.py

@@ -1,118 +1,183 @@
 import streamlit as st
+from dotenv import load_dotenv
 from auth import generate_otp, create_jwt, verify_jwt
 from emails_utils import send_otp_email
-from dotenv import load_dotenv
 from model_api import query_model
 from prompt_builder import build_prompt
 
 load_dotenv()
 
-st.set_page_config(page_title="FitPlan AI", page_icon="🏋️", layout="centered")
-
-# ---------------- SESSION STATE ---------------- #
+# ---------- PAGE CONFIG ----------
+st.set_page_config(
+    page_title="AI Fitness Planner",
+    page_icon="💪",
+    layout="centered"
+)
 
+# ---------- CUSTOM CSS ----------
+st.markdown("""
+<style>
+
+.stApp{
+background-image: url("https://images.unsplash.com/photo-1571019613914-85f342c55f2b");
+background-size: cover;
+}
+
+.title{
+text-align:center;
+font-size:40px;
+font-weight:bold;
+color:white;
+}
+
+.card{
+background: rgba(0,0,0,0.6);
+padding:30px;
+border-radius:15px;
+}
+
+label{
+color:white !important;
+font-weight:bold;
+}
+
+</style>
+""", unsafe_allow_html=True)
+
+# ---------- SESSION ----------
 if "logged_in" not in st.session_state:
     st.session_state.logged_in = False
 
-if "otp_sent" not in st.session_state:
-    st.session_state.otp_sent = False
-
 if "otp" not in st.session_state:
     st.session_state.otp = None
 
 
-# ---------------- LOGIN PAGE ---------------- #
+# =====================================================
+# 🔐 LOGIN PAGE
+# =====================================================
 
 if not st.session_state.logged_in:
 
-    st.title("🔐 FitPlan AI - OTP Login")
+    st.markdown('<div class="title">🏋️ FitPlan AI Login</div>', unsafe_allow_html=True)
 
-    email = st.text_input("Enter your Email")
+    with st.container():
+        st.markdown('<div class="card">', unsafe_allow_html=True)
 
-    if not st.session_state.otp_sent:
+        email = st.text_input("📧 Enter Email")
 
         if st.button("Send OTP"):
 
-            otp = generate_otp()
+            if email:
+                otp = generate_otp()
+                st.session_state.otp = otp
 
-            st.session_state.otp = otp
-            st.session_state.email = email
-            st.session_state.otp_sent = True
-            
-            send_otp_email(email, otp)
+                send_otp_email(email, otp)
 
-            st.success("OTP sent to your email")
+                st.success("OTP sent to your email")
 
-    else:
+            else:
+                st.warning("Please enter email")
 
-        otp_input = st.text_input("Enter OTP")
+        user_otp = st.text_input("🔑 Enter OTP")
 
         if st.button("Verify OTP"):
 
-            if otp_input == st.session_state.otp:
+            if user_otp == st.session_state.otp:
 
-                token = create_jwt(st.session_state.email)
+                token = create_jwt(email)
 
-                st.session_state.jwt = token
+                st.session_state.token = token
                 st.session_state.logged_in = True
 
-                st.success("Login Successful")
+                st.success("Login Successful 🎉")
                 st.rerun()
 
             else:
                 st.error("Invalid OTP")
 
+        st.markdown("</div>", unsafe_allow_html=True)
+
     st.stop()
 
 
-# ---------------- APP CONTENT ---------------- #
+# =====================================================
+# 💪 WORKOUT GENERATOR PAGE
+# =====================================================
 
-st.title("🏋️ AI Personalized 5-Day Workout Planner")
+st.markdown('<div class="title">💪 AI Personalized 5-Day Workout Planner</div>', unsafe_allow_html=True)
 
-name = st.text_input("Name")
+with st.container():
 
-age = st.number_input("Age", 0, 100)
+    st.markdown('<div class="card">', unsafe_allow_html=True)
 
-gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+    name = st.text_input("Name")
 
-height = st.number_input("Height (cm)", 0, 250)
+    age = st.number_input("Age", 10, 80)
 
-weight = st.number_input("Weight (kg)", 0, 200)
+    gender = st.selectbox(
+        "Gender",
+        ["Male", "Female", "Other"]
+    )
 
-goal = st.selectbox(
-    "Fitness Goal",
-    ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexibility"]
-)
+    height = st.number_input("Height (cm)", 100, 250)
 
-fitness_level = st.selectbox(
-    "Fitness Level",
-    ["Beginner", "Intermediate", "Advanced"]
-)
+    weight = st.number_input("Weight (kg)", 30, 200)
 
-equipment = st.multiselect(
-    "Equipment",
-    [
-        "Dumbbells",
-        "Resistance Band",
-        "Yoga Mat",
-        "Skipping Rope",
-        "Pullups Bar",
-        "Inclined Bench",
-        "Weight Plates"
-    ]
-)
+    goal = st.selectbox(
+        "Fitness Goal",
+        [
+            "Build Muscle",
+            "Weight Loss",
+            "Strength Gain",
+            "Abs Building",
+            "Flexibility"
+        ]
+    )
 
-if st.button("Generate 5-Day Plan 💪"):
+    fitness_level = st.selectbox(
+        "Fitness Level",
+        [
+            "Beginner",
+            "Intermediate",
+            "Advanced"
+        ]
+    )
 
-    prompt, bmi, bmi_status = build_prompt(
-        name, age, gender, height, weight,
-        goal, fitness_level, equipment
+    equipment = st.multiselect(
+        "Available Equipment",
+        [
+            "Dumbbells",
+            "Resistance Band",
+            "Yoga Mat",
+            "Skipping Rope",
+            "Pullups Bar",
+            "Inclined Bench",
+            "Weight Plates"
+        ]
     )
 
-    st.subheader(f"BMI: {bmi:.2f} ({bmi_status})")
+    if st.button("Generate 5-Day Plan 💪"):
+
+        prompt, bmi, bmi_status = build_prompt(
+            name, age, gender,
+            height, weight,
+            goal, fitness_level,
+            equipment
+        )
+
+        st.subheader(f"📊 BMI: {bmi:.2f} ({bmi_status})")
 
-    with st.spinner("Generating workout plan..."):
-        result = query_model(prompt)
+        with st.spinner("Generating your AI workout plan..."):
 
-    st.markdown("## 🗓️ Your 5-Day Workout Plan")
-    st.markdown(result)
+            result = query_model(prompt)
+
+        st.markdown("## 🗓️ Your Workout Plan")
+        st.markdown(result)
+
+    st.markdown("</div>", unsafe_allow_html=True)
+
+
+# ---------- LOGOUT ----------
+if st.button("Logout"):
+    st.session_state.logged_in = False
+    st.rerun()