Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,39 +19,79 @@ st.write("Get a personalized fitness plan based on your details")
 # User Inputs
 age = st.number_input("Age", min_value=10, max_value=100, step=1)
 gender = st.selectbox("Gender", ["Male", "Female", "Other"])
-goal = st.selectbox("Fitness Goal", ["Weight Loss", "Muscle Gain", "General Fitness"])
-activity = st.selectbox("Activity Level", ["Low", "Medium", "High"])
+goal = st.selectbox(
+    "Fitness Goal",
+    [
+        "Flexibility",
+        "Weight Loss",
+        "Build Muscle",
+        "Strength Gaining",
+        "Abs Building"
+    ]
+)
+fitness_level = st.selectbox(
+    "Fitness Level",
+    ["Beginner", "Intermediate", "Advanced"]
+)
+
+equipment = st.multiselect(
+    "Available Equipment",
+    [
+        "Dumbbells",
+        "Resistance Band",
+        "Yoga Mat",
+        "Inclined Bench",
+        "Treadmill",
+        "Cycle",
+        "Skipping Rope",
+        "Hand Gripper",
+        "Pull-ups Bar",
+        "Weight Plates",
+        "Hula Hoop Ring",
+        "Bosu Ball"
+    ]
+)
 
 # Plan Logic
 
-def generate_plan(age, goal, activity):
+def generate_plan(goal, fitness_level, equipment):
     plan = ""
 
     if goal == "Weight Loss":
-        plan = "🏃 Cardio 4–5 days/week, strength training 2–3 days/week, calorie-controlled diet."
-    elif goal == "Muscle Gain":
-        plan = "🏋️ Strength training 4–5 days/week, high-protein diet, proper sleep and recovery."
+        plan = "🏃 Focus on calorie-burning workouts with cardio and light strength training."
+    elif goal == "Build Muscle":
+        plan = "🏋️ Progressive strength training with adequate protein intake."
+    elif goal == "Strength Gaining":
+        plan = "💪 Heavy compound lifts with longer rest periods."
+    elif goal == "Abs Building":
+        plan = "🔥 Core-focused workouts combined with fat-loss strategies."
+    else:
+        plan = "🧘 Flexibility and mobility routines with stretching and yoga."
+
+    if fitness_level == "Beginner":
+        plan += " Start slow, focus on form, and train 3–4 days/week."
+    elif fitness_level == "Intermediate":
+        plan += " Increase intensity and volume, train 4–5 days/week."
     else:
-        plan = "🤸 Combination of cardio & strength training 3–4 days/week, balanced nutrition."
+        plan += " High-intensity training with structured programming, 5–6 days/week."
 
-    if activity == "Low":
-        plan += " Start with light intensity and gradually increase."
-    elif activity == "Medium":
-        plan += " Maintain moderate intensity workouts."
+    if equipment:
+        plan += f" Use available equipment: {', '.join(equipment)}."
     else:
-        plan += " Include high-intensity workouts with proper rest days."
+        plan += " Bodyweight exercises will be emphasized."
 
     return plan
 
 # Button
 if st.button("Generate Fitness Plan"):
-    plan = generate_plan(age, goal, activity)
+    plan = generate_plan(goal, fitness_level, equipment)
     st.success("Your Personalized Fitness Plan")
     st.markdown(f"""
     **Age:** {age}  
     **Gender:** {gender}  
     **Goal:** {goal}  
-    **Activity Level:** {activity}
+    **Fitness Level:** {fitness_level}  
+**Equipment:** {', '.join(equipment) if equipment else 'None'}
 
     ---
     {plan}