Update src/streamlit_app.py

src/streamlit_app.py

@@ -57,67 +57,40 @@ goal = st.selectbox(
     ]
 )
 
-# ---------------------------------------------------
-# EQUIPMENT (FULL IMPLEMENTATION)
-# ---------------------------------------------------
 st.subheader("🏋️ Available Equipment")
 
 equipment_map = {}
 
-# 🏠 Bodyweight
-
-equipment_map["No Equipment"] = st.checkbox("No Equipment")
-equipment_map["Pull-up Bar"] = st.checkbox("Pull-up Bar")
-equipment_map["Dip Bars"] = st.checkbox("Dip Bars")
-equipment_map["Push-up Handles"] = st.checkbox("Push-up Handles")
-
-# 🏋️ Free Weights
-
-equipment_map["Dumbbells"] = st.checkbox("Dumbbells")
-equipment_map["Adjustable Dumbbells"] = st.checkbox("Adjustable Dumbbells")
-equipment_map["Barbell"] = st.checkbox("Barbell")
-equipment_map["Weight Plates"] = st.checkbox("Weight Plates")
-equipment_map["Kettlebells"] = st.checkbox("Kettlebells")
-equipment_map["Medicine Ball"] = st.checkbox("Medicine Ball")
-
-# 🧘 Flexibility & Core
-
-equipment_map["Yoga Mat"] = st.checkbox("Yoga Mat")
-equipment_map["Resistance Band"] = st.checkbox("Resistance Band")
-equipment_map["Bosu Ball"] = st.checkbox("Bosu Ball")
-equipment_map["Stability Ball"] = st.checkbox("Stability Ball")
-equipment_map["Foam Roller"] = st.checkbox("Foam Roller")
-equipment_map["Hula Hoop Ring"] = st.checkbox("Hula Hoop Ring")
-
-# 🏃 Cardio Equipment
+col1, col2, col3 = st.columns(3)
 
-equipment_map["Treadmill"] = st.checkbox("Treadmill")
-equipment_map["Exercise Bike"] = st.checkbox("Exercise Bike")
-equipment_map["Skipping Rope"] = st.checkbox("Skipping Rope")
-equipment_map["Rowing Machine"] = st.checkbox("Rowing Machine")
-equipment_map["Elliptical Trainer"] = st.checkbox("Elliptical Trainer")
-equipment_map["Stair Climber"] = st.checkbox("Stair Climber")
-
-# 💪 Strength Machines
-
-equipment_map["Incline Bench"] = st.checkbox("Incline Bench")
-equipment_map["Flat Bench"] = st.checkbox("Flat Bench")
-equipment_map["Smith Machine"] = st.checkbox("Smith Machine")
-equipment_map["Leg Press Machine"] = st.checkbox("Leg Press Machine")
-equipment_map["Cable Machine"] = st.checkbox("Cable Machine")
-equipment_map["Lat Pulldown Machine"] = st.checkbox("Lat Pulldown Machine")
-equipment_map["Chest Press Machine"] = st.checkbox("Chest Press Machine")
-
-# ✋ Small Tools
-
-equipment_map["Hand Gripper"] = st.checkbox("Hand Gripper")
-equipment_map["Wrist Roller"] = st.checkbox("Wrist Roller")
-equipment_map["Ankle Weights"] = st.checkbox("Ankle Weights")
-equipment_map["Resistance Tubes"] = st.checkbox("Resistance Tubes")
+with col1:
+    equipment_map["No Equipment"] = st.checkbox("No Equipment")
+    equipment_map["Pull-up Bar"] = st.checkbox("Pull-up Bar")
+    equipment_map["Dip Bars"] = st.checkbox("Dip Bars")
+    equipment_map["Push-up Handles"] = st.checkbox("Push-up Handles")
+    equipment_map["Dumbbells"] = st.checkbox("Dumbbells")
+    equipment_map["Adjustable Dumbbells"] = st.checkbox("Adjustable Dumbbells")
 
-# Convert selected equipment to list
+with col2:
+    equipment_map["Barbell"] = st.checkbox("Barbell")
+    equipment_map["Weight Plates"] = st.checkbox("Weight Plates")
+    equipment_map["Kettlebells"] = st.checkbox("Kettlebells")
+    equipment_map["Medicine Ball"] = st.checkbox("Medicine Ball")
+    equipment_map["Yoga Mat"] = st.checkbox("Yoga Mat")
+    equipment_map["Resistance Band"] = st.checkbox("Resistance Band")
+
+with col3:
+    equipment_map["Bosu Ball"] = st.checkbox("Bosu Ball")
+    equipment_map["Stability Ball"] = st.checkbox("Stability Ball")
+    equipment_map["Foam Roller"] = st.checkbox("Foam Roller")
+    equipment_map["Treadmill"] = st.checkbox("Treadmill")
+    equipment_map["Exercise Bike"] = st.checkbox("Exercise Bike")
+    equipment_map["Skipping Rope"] = st.checkbox("Skipping Rope")
+
+# Convert selected to list
 equipment = [item for item, selected in equipment_map.items() if selected]
 
+
 # ---------------------------------------------------
 # FITNESS LEVEL
 # ---------------------------------------------------