Update app.py

app.py

@@ -1,183 +1,114 @@
 import streamlit as st
-from dotenv import load_dotenv
-from auth import generate_otp, create_jwt, verify_jwt
+from auth import generate_otp, create_jwt
 from emails_utils import send_otp_email
-from model_api import query_model
-from prompt_builder import build_prompt
 
-load_dotenv()
+st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-# ---------- PAGE CONFIG ----------
-st.set_page_config(
-    page_title="AI Fitness Planner",
-    page_icon="💪",
-    layout="centered"
-)
+# ------------------ CUSTOM CSS ------------------
 
-# ---------- CUSTOM CSS ----------
 st.markdown("""
 <style>
 
 .stApp{
-background-image: url("https://images.unsplash.com/photo-1571019613914-85f342c55f2b");
-background-size: cover;
+background: linear-gradient(135deg,#0f172a,#020617);
+color:white;
+}
+
+.login-card{
+background: rgba(0,0,0,0.65);
+padding:40px;
+border-radius:20px;
+width:400px;
+margin:auto;
+box-shadow:0 10px 30px rgba(0,0,0,0.6);
 }
 
 .title{
 text-align:center;
-font-size:40px;
-font-weight:bold;
-color:white;
+font-size:34px;
+font-weight:700;
+margin-bottom:25px;
 }
 
-.card{
-background: rgba(0,0,0,0.6);
-padding:30px;
-border-radius:15px;
+.stTextInput input{
+background:#111827;
+color:white;
+border-radius:10px;
 }
 
-label{
-color:white !important;
+.stButton>button{
+width:100%;
+background:linear-gradient(90deg,#2563eb,#6366f1);
+color:white;
+border-radius:12px;
 font-weight:bold;
+height:45px;
+}
+
+.otp-box{
+text-align:center;
+letter-spacing:10px;
+font-size:20px;
 }
 
 </style>
 """, unsafe_allow_html=True)
 
-# ---------- SESSION ----------
+# ------------------ SESSION ------------------
+
 if "logged_in" not in st.session_state:
     st.session_state.logged_in = False
 
 if "otp" not in st.session_state:
     st.session_state.otp = None
 
-
-# =====================================================
-# 🔐 LOGIN PAGE
-# =====================================================
+# ------------------ LOGIN SCREEN ------------------
 
 if not st.session_state.logged_in:
 
-    st.markdown('<div class="title">🏋️ FitPlan AI Login</div>', unsafe_allow_html=True)
-
-    with st.container():
-        st.markdown('<div class="card">', unsafe_allow_html=True)
+    st.markdown('<div class="login-card">', unsafe_allow_html=True)
 
-        email = st.text_input("📧 Enter Email")
+    st.markdown('<div class="title">🔐 Login To Your Account</div>', unsafe_allow_html=True)
 
-        if st.button("Send OTP"):
+    email = st.text_input("Email")
 
-            if email:
-                otp = generate_otp()
-                st.session_state.otp = otp
+    if st.button("Send OTP"):
 
-                send_otp_email(email, otp)
+        if email:
 
-                st.success("OTP sent to your email")
+            otp = generate_otp()
+            st.session_state.otp = otp
 
-            else:
-                st.warning("Please enter email")
+            send_otp_email(email, otp)
 
-        user_otp = st.text_input("🔑 Enter OTP")
+            st.success("OTP sent to your email")
 
-        if st.button("Verify OTP"):
+        else:
+            st.warning("Enter email first")
 
-            if user_otp == st.session_state.otp:
+    user_otp = st.text_input("Enter OTP", max_chars=6)
 
-                token = create_jwt(email)
+    if st.button("Verify OTP"):
 
-                st.session_state.token = token
-                st.session_state.logged_in = True
+        if user_otp == st.session_state.otp:
 
-                st.success("Login Successful 🎉")
-                st.rerun()
-
-            else:
-                st.error("Invalid OTP")
-
-        st.markdown("</div>", unsafe_allow_html=True)
-
-    st.stop()
+            token = create_jwt(email)
 
+            st.session_state.logged_in = True
+            st.session_state.token = token
 
-# =====================================================
-# 💪 WORKOUT GENERATOR PAGE
-# =====================================================
+            st.success("Login Successful")
+            st.rerun()
 
-st.markdown('<div class="title">💪 AI Personalized 5-Day Workout Planner</div>', unsafe_allow_html=True)
+        else:
+            st.error("Invalid OTP")
 
-with st.container():
+    st.markdown('</div>', unsafe_allow_html=True)
 
-    st.markdown('<div class="card">', unsafe_allow_html=True)
-
-    name = st.text_input("Name")
-
-    age = st.number_input("Age", 10, 80)
-
-    gender = st.selectbox(
-        "Gender",
-        ["Male", "Female", "Other"]
-    )
-
-    height = st.number_input("Height (cm)", 100, 250)
-
-    weight = st.number_input("Weight (kg)", 30, 200)
-
-    goal = st.selectbox(
-        "Fitness Goal",
-        [
-            "Build Muscle",
-            "Weight Loss",
-            "Strength Gain",
-            "Abs Building",
-            "Flexibility"
-        ]
-    )
-
-    fitness_level = st.selectbox(
-        "Fitness Level",
-        [
-            "Beginner",
-            "Intermediate",
-            "Advanced"
-        ]
-    )
-
-    equipment = st.multiselect(
-        "Available Equipment",
-        [
-            "Dumbbells",
-            "Resistance Band",
-            "Yoga Mat",
-            "Skipping Rope",
-            "Pullups Bar",
-            "Inclined Bench",
-            "Weight Plates"
-        ]
-    )
-
-    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 your AI workout plan..."):
-
-            result = query_model(prompt)
-
-        st.markdown("## 🗓️ Your Workout Plan")
-        st.markdown(result)
+    st.stop()
 
-    st.markdown("</div>", unsafe_allow_html=True)
+# ------------------ AFTER LOGIN ------------------
 
+st.title("🏋️ AI Personalized 5-Day Workout Planner")
 
-# ---------- LOGOUT ----------
-if st.button("Logout"):
-    st.session_state.logged_in = False
-    st.rerun()
+st.success("Login successful! Now generate your workout plan.")