Update app.py

app.py

@@ -13,7 +13,10 @@ from prompt_builder import build_prompt
 
 create_tables()
 
-st.set_page_config(page_title="FitPlan AI", layout="centered")
+st.set_page_config(
+    page_title="FitPlan AI",
+    layout="centered"
+)
 
 st.title("🏋️ FitPlan AI - Personalized Fitness Planner")
 
@@ -112,7 +115,7 @@ if st.session_state.credentials_verified and not st.session_state.otp_sent:
 
 
 # -------------------------
-# VERIFY + RESEND OTP
+# VERIFY OTP
 # -------------------------
 
 if st.session_state.otp_sent:
@@ -123,7 +126,6 @@ if st.session_state.otp_sent:
 
     col1, col2 = st.columns(2)
 
-    # VERIFY OTP
     with col1:
 
         if st.button("Verify OTP"):
@@ -141,10 +143,12 @@ if st.session_state.otp_sent:
 
                 st.success("Login successful")
 
+                st.rerun()
+
             else:
+
                 st.error("Invalid OTP")
 
-    # RESEND OTP
     with col2:
 
         if st.button("Resend OTP"):
@@ -155,80 +159,50 @@ if st.session_state.otp_sent:
 
             send_otp_email(st.session_state.email, otp)
 
-            st.success("New OTP sent to your email")
+            st.success("New OTP sent")
 
 
 # -------------------------
-# DASHBOARD
+# MAIN APPLICATION
 # -------------------------
 
 if st.session_state.user:
 
     st.sidebar.success(f"Logged in as {st.session_state.user}")
 
-    page = st.sidebar.selectbox(
-        "Navigation",
-        ["Dashboard", "Weight Tracker", "Generate Fitness Plan"]
+    tabs = st.tabs(
+        [
+            "Dashboard",
+            "Profile",
+            "Workout Plan",
+            "Weight Tracker",
+            "Logout"
+        ]
     )
 
 
-    # -------------------------
-    # DASHBOARD PAGE
-    # -------------------------
+# -------------------------
+# DASHBOARD
+# -------------------------
 
-    if page == "Dashboard":
+    with tabs[0]:
 
-        st.header("Welcome to FitPlan AI")
+        st.header("Dashboard")
 
         st.write(
-            "Track your fitness progress and generate AI workout plans."
-        )
-
-
-    # -------------------------
-    # WEIGHT TRACKER
-    # -------------------------
-
-    if page == "Weight Tracker":
-
-        st.header("Weight Tracker")
-
-        weight = st.number_input(
-            "Enter today's weight (kg)",
-            min_value=30.0
-        )
-
-        if st.button("Save Weight"):
-
-            save_weight(
-                st.session_state.user,
-                weight
-            )
-
-            st.success("Weight saved successfully")
-
-        st.subheader("Weight History")
-
-        history = get_weight_history(
-            st.session_state.user
+            "Welcome to FitPlan AI. Generate personalized workout plans and track your fitness progress."
         )
 
-        if history:
-
-            st.table(history)
-
-        else:
-
-            st.info("No weight data available")
 
+# -------------------------
+# PROFILE
+# -------------------------
 
-    # -------------------------
-    # AI FITNESS PLAN
-    # -------------------------
+    with tabs[1]:
 
-    if page == "Generate Fitness Plan":
+        st.header("User Profile")
 
-        st.header("Generate Personalized Fitness Plan")
+        name = st.text_input("Name")
 
         age = st.number_input(
             "Age",
@@ -236,12 +210,26 @@ if st.session_state.user:
             80
         )
 
+        gender = st.selectbox(
+            "Gender",
+            ["Male", "Female", "Other"]
+        )
+
         height = st.number_input(
             "Height (cm)",
             100,
             220
         )
 
+
+# -------------------------
+# WORKOUT PLAN GENERATOR
+# -------------------------
+
+    with tabs[2]:
+
+        st.header("Generate Personalized Workout Plan")
+
         weight = st.number_input(
             "Weight (kg)",
             30,
@@ -251,14 +239,14 @@ if st.session_state.user:
         goal = st.selectbox(
             "Fitness Goal",
             [
-                "Weight Loss",
-                "Muscle Gain",
+                "Lose Weight",
+                "Build Muscle",
                 "Maintain Fitness"
             ]
         )
 
-        activity = st.selectbox(
-            "Activity Level",
+        fitness_level = st.selectbox(
+            "Fitness Level",
             [
                 "Beginner",
                 "Intermediate",
@@ -266,34 +254,91 @@ if st.session_state.user:
             ]
         )
 
-        if st.button("Generate Plan"):
+        equipment = st.multiselect(
+            "Available Equipment",
+            [
+                "Dumbbells",
+                "Barbell",
+                "Resistance Bands",
+                "Bodyweight",
+                "None"
+            ]
+        )
 
-            prompt = build_prompt(
+        if st.button("Generate AI Plan"):
+
+            prompt, bmi, bmi_status = build_prompt(
+                name,
                 age,
+                gender,
                 height,
                 weight,
                 goal,
-                activity
+                fitness_level,
+                equipment
             )
 
-            with st.spinner("Generating AI plan..."):
+            st.write(f"BMI: {bmi:.2f} ({bmi_status})")
+
+            with st.spinner("Generating workout plan..."):
 
                 result = query_model(prompt)
 
-            st.subheader("Your AI Fitness Plan")
+            st.subheader("Your AI Workout Plan")
 
             st.write(result)
 
 
-    # -------------------------
-    # LOGOUT
-    # -------------------------
+# -------------------------
+# WEIGHT TRACKER
+# -------------------------
+
+    with tabs[3]:
+
+        st.header("Weight Tracker")
+
+        today_weight = st.number_input(
+            "Enter today's weight",
+            min_value=30.0
+        )
+
+        if st.button("Save Weight"):
+
+            save_weight(
+                st.session_state.user,
+                today_weight
+            )
+
+            st.success("Weight saved")
+
+        st.subheader("Weight History")
+
+        history = get_weight_history(
+            st.session_state.user
+        )
+
+        if history:
+
+            st.table(history)
+
+        else:
+
+            st.info("No weight records found")
+
+
+# -------------------------
+# LOGOUT
+# -------------------------
+
+    with tabs[4]:
+
+        st.header("Logout")
 
-    if st.sidebar.button("Logout"):
+        if st.button("Logout"):
 
-        st.session_state.user = None
-        st.session_state.email = None
-        st.session_state.credentials_verified = False
-        st.session_state.otp_sent = False
+            st.session_state.user = None
+            st.session_state.email = None
+            st.session_state.credentials_verified = False
+            st.session_state.otp_sent = False
 
-        st.rerun()
+            st.rerun()