Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,169 +1,203 @@
 import streamlit as st
 import time
-
-st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
-
-# ------------------ CUSTOM CSS ------------------
+import json
+
+# -----------------------
+# PAGE CONFIG
+# -----------------------
+st.set_page_config(
+    page_title="FitPlan AI",
+    page_icon="💪",
+    layout="wide",
+)
+
+# -----------------------
+# CUSTOM CSS (MODERN UI)
+# -----------------------
 st.markdown("""
 <style>
-.main {
-    background-color: #f8fafc;
+
+body {
+    background-color: #0E1117;
+}
+
+.main-title {
+    font-size: 42px;
+    font-weight: 700;
+    color: white;
+}
+
+.subtitle {
+    font-size: 18px;
+    color: #A0AEC0;
 }
-.section-card {
-    background-color: white;
+
+.card {
+    background-color: #1C1F26;
     padding: 25px;
     border-radius: 15px;
-    box-shadow: 0px 6px 18px rgba(0,0,0,0.05);
-    margin-bottom: 25px;
-}
-h3 {
-    margin-top: 0px;
+    box-shadow: 0px 4px 20px rgba(0,0,0,0.4);
 }
+
 .stButton>button {
-    background-color: black;
+    background: linear-gradient(90deg, #4F46E5, #7C3AED);
     color: white;
     border-radius: 10px;
-    padding: 10px 25px;
+    height: 3em;
+    width: 100%;
+    font-size: 16px;
     font-weight: 600;
 }
+
 .stButton>button:hover {
-    background-color: #333;
+    background: linear-gradient(90deg, #6366F1, #8B5CF6);
 }
-</style>
-""", unsafe_allow_html=True)
-
-# ------------------ HEADER ------------------
-st.title("💪 FitPlan AI")
-st.subheader("Complete Fitness Profile Setup")
 
-st.divider()
-
-# ------------------ USER PROFILE FORM ------------------
+.plan-box {
+    background-color: #1F2937;
+    padding: 20px;
+    border-radius: 10px;
+    margin-bottom: 15px;
+    color: white;
+}
 
-with st.form("profile_form"):
+</style>
+""", unsafe_allow_html=True)
 
-    st.markdown("### 👤 Personal Information")
 
-    col1, col2 = st.columns(2)
+# -----------------------
+# SIDEBAR NAVIGATION
+# -----------------------
+st.sidebar.title("💪 FitPlan AI")
+page = st.sidebar.radio(
+    "Navigate",
+    ["🏠 Home", "👤 User Profile", "📝 Fitness Details", "📅 Generated Plan"]
+)
+
+
+# -----------------------
+# HOME PAGE
+# -----------------------
+if page == "🏠 Home":
+    st.markdown("<div class='main-title'>FitPlan AI</div>", unsafe_allow_html=True)
+    st.markdown("<div class='subtitle'>Your Personalized AI Fitness Planner</div>", unsafe_allow_html=True)
+    st.write("")
+
+    st.markdown("""
+    <div class='card'>
+    <h3 style='color:white;'>Why FitPlan AI?</h3>
+    <ul style='color:#A0AEC0;'>
+        <li>Personalized workout plans</li>
+        <li>AI-powered fitness recommendations</li>
+        <li>Structured sets, reps & rest periods</li>
+        <li>No expensive personal trainer needed</li>
+    </ul>
+    </div>
+    """, unsafe_allow_html=True)
+
+
+# -----------------------
+# USER PROFILE PAGE
+# -----------------------
+elif page == "👤 User Profile":
+
+    st.markdown("<div class='main-title'>User Profile</div>", unsafe_allow_html=True)
+    st.write("")
+
+    with st.container():
+        st.markdown("<div class='card'>", unsafe_allow_html=True)
 
-    with col1:
         name = st.text_input("Full Name")
-        age = st.number_input("Age", 15, 70, 21)
-        gender = st.selectbox("Gender", ["Female", "Male", "Other"])
-
-    with col2:
-        height = st.number_input("Height (cm)", 120, 220, 160)
-        weight = st.number_input("Weight (kg)", 30, 150, 55)
-        fitness_level = st.selectbox("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
-
-    st.divider()
-
-    st.markdown("### 🎯 Fitness Goals")
+        age = st.number_input("Age", min_value=10, max_value=80)
+        gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+        height = st.number_input("Height (cm)")
+        weight = st.number_input("Weight (kg)")
 
-    primary_goal = st.selectbox(
-        "Primary Goal",
-        ["Build Muscle", "Lose Weight", "Fat Loss", "Endurance", "Flexibility", "General Fitness"]
-    )
+        if st.button("Save Profile"):
+            st.success("Profile saved successfully!")
 
-    target_weight = st.number_input("Target Weight (kg)", 30, 150, 50)
-    workout_days = st.slider("Workout Days per Week", 2, 7, 4)
+        st.markdown("</div>", unsafe_allow_html=True)
 
-    st.divider()
 
-    st.markdown("### 🏋️ Workout Preferences")
+# -----------------------
+# FITNESS DETAILS PAGE
+# -----------------------
+elif page == "📝 Fitness Details":
 
-    equipment = st.multiselect(
-        "Available Equipment",
-        ["No Equipment", "Dumbbells", "Resistance Bands", "Barbell", "Full Gym"]
-    )
+    st.markdown("<div class='main-title'>Enter Your Fitness Details</div>", unsafe_allow_html=True)
+    st.write("")
 
-    workout_duration = st.selectbox(
-        "Preferred Workout Duration",
-        ["30 mins", "45 mins", "60 mins", "90 mins"]
-    )
+    with st.container():
+        st.markdown("<div class='card'>", unsafe_allow_html=True)
 
-    preferred_time = st.selectbox(
-        "Preferred Workout Time",
-        ["Morning", "Afternoon", "Evening"]
-    )
+        goal = st.selectbox(
+            "Fitness Goal",
+            ["Build Muscle", "Lose Weight", "Improve Cardio", "Increase Flexibility"]
+        )
 
-    st.divider()
+        fitness_level = st.selectbox(
+            "Fitness Level",
+            ["Beginner", "Intermediate", "Advanced"]
+        )
 
-    st.markdown("### 🥗 Lifestyle & Diet")
+        equipment = st.multiselect(
+            "Available Equipment",
+            ["No Equipment", "Dumbbells", "Barbell", "Resistance Bands", "Gym Access"]
+        )
 
-    diet_type = st.selectbox(
-        "Diet Preference",
-        ["Vegetarian", "Non-Vegetarian", "Vegan", "Keto", "No Preference"]
-    )
+        days = st.slider("Workout Days per Week", 1, 7, 3)
 
-    sleep_hours = st.slider("Average Sleep (hours)", 4, 10, 7)
+        if st.button("Generate Plan"):
+            if not equipment:
+                st.error("Please select at least one equipment option.")
+            else:
+                st.session_state["goal"] = goal
+                st.session_state["fitness_level"] = fitness_level
+                st.session_state["equipment"] = equipment
+                st.session_state["days"] = days
 
-    water_intake = st.slider("Water Intake (liters/day)", 1, 5, 2)
+                with st.spinner("AI is generating your personalized plan..."):
+                    time.sleep(2)
 
-    stress_level = st.selectbox(
-        "Stress Level",
-        ["Low", "Moderate", "High"]
-    )
+                st.success("Plan Generated! Go to 'Generated Plan' page.")
 
-    st.divider()
+        st.markdown("</div>", unsafe_allow_html=True)
 
-    st.markdown("### ⚕ Health Conditions")
 
-    injuries = st.text_area("Any injuries or medical conditions?")
-    medications = st.text_area("Are you on any medications?")
+# -----------------------
+# GENERATED PLAN PAGE
+# -----------------------
+elif page == "📅 Generated Plan":
 
-    st.divider()
+    st.markdown("<div class='main-title'>Your Weekly Workout Plan</div>", unsafe_allow_html=True)
+    st.write("")
 
-    submitted = st.form_submit_button("Create My Fitness Profile")
-
-# ------------------ AFTER SUBMISSION ------------------
-
-if submitted:
-
-    if name == "":
-        st.error("Please enter your name.")
+    if "goal" not in st.session_state:
+        st.warning("Please fill Fitness Details first.")
     else:
 
-        with st.spinner("Creating your fitness profile..."):
-            time.sleep(2)
-
-        st.success(f"Profile Created Successfully, {name}! 🔥")
-
-        st.divider()
-
-        # ------------------ PROFILE SUMMARY ------------------
-
-        st.markdown("## 📊 Your Fitness Summary")
-
-        bmi = weight / ((height/100) ** 2)
-
-        col1, col2 = st.columns(2)
-
-        with col1:
-            st.metric("BMI", round(bmi, 2))
-            st.metric("Current Weight", f"{weight} kg")
-            st.metric("Target Weight", f"{target_weight} kg")
-
-        with col2:
-            st.metric("Workout Days", workout_days)
-            st.metric("Sleep Hours", sleep_hours)
-            st.metric("Water Intake", f"{water_intake} L")
-
-        st.divider()
-
-        st.markdown("### 📝 Complete Profile Details")
+        # MOCK PLAN (Replace with LLM output later)
+        plan = {
+            "Day 1": [
+                "Push-ups – 3 sets × 12 reps",
+                "Dumbbell Squats – 3 sets × 15 reps",
+                "Plank – 3 sets × 30 sec"
+            ],
+            "Day 2": [
+                "Lunges – 3 sets × 12 reps",
+                "Shoulder Press – 3 sets × 10 reps",
+                "Jumping Jacks – 3 sets × 20 reps"
+            ],
+            "Day 3": [
+                "Deadlifts – 3 sets × 10 reps",
+                "Bicep Curls – 3 sets × 12 reps",
+                "Mountain Climbers – 3 sets × 30 sec"
+            ]
+        }
+
+        for day, exercises in plan.items():
+            st.markdown(f"<div class='plan-box'><h4>{day}</h4>", unsafe_allow_html=True)
+            for exercise in exercises:
+                st.write(f"• {exercise}")
+            st.markdown("</div>", unsafe_allow_html=True)
 
-        st.write({
-            "Name": name,
-            "Age": age,
-            "Gender": gender,
-            "Fitness Level": fitness_level,
-            "Goal": primary_goal,
-            "Equipment": equipment,
-            "Workout Duration": workout_duration,
-            "Preferred Time": preferred_time,
-            "Diet": diet_type,
-            "Stress Level": stress_level,
-            "Injuries": injuries,
-            "Medications": medications
-        })