Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,23 +1,92 @@
 import streamlit as st
 
-st.title("FIT Plan AI – Personalized Fitness Plan Generator")
+st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-goal = st.selectbox("Goal",["Flexibility","Weight Loss", "Build Muscle","Strength Gaining","ABs Building" ])
+st.title("🏋️ Your Fitness Profile")
 
+# --------------------------
+# GOALS
+# --------------------------
+goal = st.selectbox(
+    "Fitness Goal",
+    [
+        "Flexible",
+        "Weight Loss",
+        "Build Muscle",
+        "Strength Gaining",
+        "Abs Building"
+    ]
+)
+
+# --------------------------
+# EQUIPMENT
+# --------------------------
 st.subheader("Available Equipment")
 
-dumbbells = st.checkbox("Dumbbells")
-bands = st.checkbox("Resistance Bands")
-Yoga = st.checkbox("Yoga mat")
-Inclined bench = st.checkbox("Inclined bench")
-Thread mill = st.checkbox("Thread mill")
-Cycle = st.checkbox("Cycle")
-Skipping rope = st.checkbox("Skipping rope")
-Hand gripper = st.checkbox("Hand gripper")
-PullUps = st.checkbox("Pullups bar")
-Weight = st.checkbox("Weight Plates")
-Hula = st.checkbox("Hula hoop ring")
-bosu = st.checkbox("Bosu ball")
-no_equipment = st.checkbox("No Equipment")
-
-st.write("You selected:", goal)
+col1, col2 = st.columns(2)
+
+with col1:
+    dumbbells = st.checkbox("Dumbbells")
+    resistance_band = st.checkbox("Resistance Band")
+    yoga_mat = st.checkbox("Yoga Mat")
+    no_equipment = st.checkbox("No Equipment")
+    inclined_bench = st.checkbox("Inclined Bench")
+    treadmill = st.checkbox("Treadmill")
+    cycle = st.checkbox("Cycle")
+
+with col2:
+    skipping_rope = st.checkbox("Skipping Rope")
+    hand_gripper = st.checkbox("Hand Gripper")
+    pullups_bar = st.checkbox("Pullups Bar")
+    weight_plates = st.checkbox("Weight Plates")
+    hula_hoop = st.checkbox("Hula Hoop Ring")
+    bosu_ball = st.checkbox("Bosu Ball")
+
+# Collect equipment into list
+equipment = []
+
+equipment_map = {
+    "Dumbbells": dumbbells,
+    "Resistance Band": resistance_band,
+    "Yoga Mat": yoga_mat,
+    "No Equipment": no_equipment,
+    "Inclined Bench": inclined_bench,
+    "Treadmill": treadmill,
+    "Cycle": cycle,
+    "Skipping Rope": skipping_rope,
+    "Hand Gripper": hand_gripper,
+    "Pullups Bar": pullups_bar,
+    "Weight Plates": weight_plates,
+    "Hula Hoop Ring": hula_hoop,
+    "Bosu Ball": bosu_ball,
+}
+
+for item, selected in equipment_map.items():
+    if selected:
+        equipment.append(item)
+
+# --------------------------
+# FITNESS LEVEL
+# --------------------------
+st.subheader("Fitness Level")
+
+fitness_level = st.radio(
+    "",
+    ["Beginner", "Intermediate", "Advanced"],
+    horizontal=True
+)
+
+# --------------------------
+# SUBMIT BUTTON
+# --------------------------
+if st.button("Generate Personalised Plan"):
+    
+    if not equipment:
+        st.error("Please select at least one equipment option.")
+    else:
+        st.success("Profile Submitted ✅")
+        st.json({
+            "goal": goal,
+            "fitness_level": fitness_level,
+            "equipment": equipment
+        })