Update app.py

app.py

@@ -3,195 +3,233 @@ import joblib
 import pandas as pd
 import random
 
-# 1. Load Model
+# 1. Load Model (The "Brain" trained on Synthetic Data)
 model = joblib.load('fitness_model.joblib')
 
-# 2. Smart Logic Generator
-def generate_dynamic_workout(plan_category, age, gender, weight, height, goal, equipment, injury, experience):
+# --- 2. MASSIVE EXERCISE LIBRARY ---
+# Format: "Name": (Target, [Bad For], Min_Age, Max_BMI, Equipment_Needed)
+# BMI > 30 means obese, Age > 60 means senior.
+
+exercise_db = {
+    "Legs": [
+        ("Barbell Squat", ["Quads"], ["Back", "Knee"], 0, 30, "Rack"),
+        ("Leg Press", ["Quads"], [], 0, 40, "Machine"),
+        ("Goblet Squat", ["Quads"], ["Knee"], 0, 35, "Dumbbells"),
+        ("Bodyweight Squat", ["Quads"], [], 0, 50, "Bodyweight"),
+        ("Box Squat", ["Quads"], [], 50, 40, "Bodyweight"), # Good for seniors
+        ("Romanian Deadlift", ["Hamstrings"], ["Back"], 0, 30, "Barbell"),
+        ("Dumbbell RDL", ["Hamstrings"], ["Back"], 0, 35, "Dumbbells"),
+        ("Lying Leg Curl", ["Hamstrings"], [], 0, 45, "Machine"),
+        ("Glute Bridge", ["Glutes"], [], 0, 50, "Bodyweight"), # Safe for everyone
+        ("Walking Lunges", ["Glutes"], ["Knee", "Ankle"], 0, 30, "Bodyweight"),
+        ("Bulgarian Split Squat", ["Quads"], ["Knee", "Ankle"], 0, 28, "Dumbbells") # Hard!
+    ],
+    "Push": [
+        ("Barbell Bench Press", ["Chest"], ["Shoulder"], 0, 35, "Rack"),
+        ("Dumbbell Floor Press", ["Chest"], [], 0, 40, "Dumbbells"), # Shoulder safe
+        ("Push-ups", ["Chest"], ["Shoulder"], 0, 32, "Bodyweight"),
+        ("Knee Push-ups", ["Chest"], [], 0, 50, "Bodyweight"),
+        ("Overhead Press (Barbell)", ["Shoulders"], ["Shoulder", "Back"], 0, 30, "Barbell"),
+        ("Seated Dumbbell Press", ["Shoulders"], [], 0, 40, "Dumbbells"),
+        ("Lateral Raises", ["Shoulders"], [], 0, 45, "Dumbbells"),
+        ("Tricep Dips", ["Triceps"], ["Shoulder"], 0, 28, "Bodyweight"),
+        ("Tricep Cable Pushdown", ["Triceps"], [], 0, 45, "Machine")
+    ],
+    "Pull": [
+        ("Pull-ups", ["Back"], ["Shoulder"], 0, 28, "Bodyweight"), # Hard for heavy users
+        ("Lat Pulldowns", ["Back"], [], 0, 50, "Machine"),
+        ("Barbell Row", ["Back"], ["Back"], 0, 30, "Barbell"),
+        ("Chest Supported Row", ["Back"], [], 0, 45, "Machine"), # Back safe
+        ("Single Arm Dumbbell Row", ["Back"], [], 0, 40, "Dumbbells"),
+        ("Face Pulls", ["Rear Delts"], [], 0, 50, "Machine"),
+        ("Bicep Curls", ["Biceps"], [], 0, 50, "Dumbbells")
+    ],
+    "Core": [
+        ("Plank", ["Abs"], [], 0, 40, "Bodyweight"),
+        ("Hanging Leg Raises", ["Abs"], ["Shoulder"], 0, 28, "Barbell"),
+        ("Dead Bug", ["Abs"], [], 0, 50, "Bodyweight"), # Back safe
+        ("Russian Twists", ["Abs"], ["Back"], 0, 35, "Bodyweight"),
+        ("Bird-Dog", ["Abs"], [], 0, 60, "Bodyweight")
+    ],
+    "Cardio": [
+        ("Sprints", ["Cardio"], ["Knee", "Back", "Ankle"], 0, 28, "Bodyweight"),
+        ("Treadmill Jog", ["Cardio"], ["Knee", "Ankle"], 0, 32, "Machine"),
+        ("Elliptical", ["Cardio"], [], 0, 50, "Machine"), # Low impact
+        ("Stationary Bike", ["Cardio"], [], 0, 60, "Machine"), # Very safe
+        ("Swimming", ["Cardio"], [], 0, 60, "Pool"),
+        ("Battle Ropes", ["Cardio"], ["Shoulder"], 0, 40, "Gym"),
+        ("Burpees", ["Cardio"], ["Back", "Knee", "Shoulder"], 0, 28, "Bodyweight")
+    ]
+}
+
+# --- 3. SMART LOGIC FUNCTIONS ---
+
+def calculate_bmi(weight, height_cm):
+    height_m = height_cm / 100
+    return weight / (height_m ** 2)
+
+def filter_exercises(category, injury, age, bmi, equipment):
+    """
+    Selects valid exercises based on constraints. 
+    Returns a random selection to ensure variety every time.
+    """
+    valid = []
     
-    # --- A. Setup Variables based on Profile ---
+    # Map user equipment choice to database keys
+    equip_map = []
+    if equipment == "Bodyweight Only": equip_map = ["Bodyweight"]
+    elif equipment == "Home Dumbbells": equip_map = ["Bodyweight", "Dumbbells"]
+    elif equipment == "Gym Membership": equip_map = ["Bodyweight", "Dumbbells", "Barbell", "Machine", "Rack", "Gym"]
+    else: equip_map = ["Bodyweight", "Dumbbells", "Barbell", "Rack"] # Full Home Gym
+
+    for name, targets, bad_for, min_age_limit, max_bmi_limit, req_equip in exercise_db[category]:
+        # 1. Injury Filter
+        if injury in bad_for: continue
+        # 2. Age Filter (e.g., Box squats for seniors)
+        if age > 60 and min_age_limit > 0 and age < min_age_limit: continue
+        # 3. BMI Filter (No burpees for obese users)
+        if bmi > max_bmi_limit: continue
+        # 4. Equipment Filter
+        if req_equip not in equip_map: continue
+        
+        valid.append(name)
     
-    # 1. Calculate suggested starting weights (Estimates)
-    strength_base = weight * 0.4 if gender == "Male" else weight * 0.25
-    if experience == "Beginner": strength_base *= 0.6
-    if experience == "Advanced": strength_base *= 1.4
+    # If list is empty (strict constraints), give a generic fallback
+    if not valid: return [f"Standard {category} Movement (Low Impact)"]
     
-    suggested_weight_db = f"{int(strength_base * 0.3)}kg" # Dumbbells
-    suggested_weight_bb = f"{int(strength_base * 0.8)}kg" # Barbell
+    # Return 2-3 random exercises for variety
+    count = 3 if len(valid) >= 3 else len(valid)
+    return random.sample(valid, count)
+
+def generate_finisher(goal, injury, equipment):
+    """
+    Generates a specialized finisher: Foam Rolling, Stretching, or Burnout.
+    """
+    options = []
     
-    if equipment == "Bodyweight Only":
-        weight_text = "Bodyweight"
-    elif equipment == "Home Dumbbells":
-        weight_text = f"Dumbbells (~{suggested_weight_db})"
+    # Option A: Foam Rolling (Recovery)
+    if injury == "Knee":
+        options.append("🌀 FOAM ROLL: Quads & IT Band (2 mins each leg)")
+    elif injury == "Back":
+        options.append("🌀 FOAM ROLL: Glutes & Hamstrings (Avoid lower back directly!)")
+    elif injury == "Shoulder":
+        options.append("🌀 LACROSSE BALL: Roll Pecs & Upper Traps")
     else:
-        weight_text = f"Moderate Weight (~{suggested_weight_bb} Barbell / {suggested_weight_db} DBs)"
+        options.append("🌀 FOAM ROLL: Full body flush (Calves, Quads, Upper Back)")
 
-    # 2. Adjust Volume (Sets/Reps) based on Goal
+    # Option B: Burnout (Metabolic) - Only if healthy
+    if goal in ["Weight Loss", "Athletic Performance"] and injury == "None":
+        if equipment == "Bodyweight Only":
+            options.append("🔥 BURNOUT: 50 Air Squats for time!")
+        else:
+            options.append("🔥 BURNOUT: 'Farmer Carry' max distance walking")
+
+    # Option C: Static Stretch (Flexibility)
+    if goal == "General Health" or injury != "None":
+        options.append("🧘 STRETCH: Deep Pigeon Pose (3 min hold total)")
+        options.append("🧘 STRETCH: Doorway Chest Stretch + Hang from bar")
+
+    return random.choice(options)
+
+# --- 4. MAIN GENERATOR ---
+
+def create_full_plan(plan_name, age, gender, weight, height, goal, equipment, injury, experience):
+    
+    # A. Calculate Stats
+    bmi = calculate_bmi(weight, height)
+    bmi_status = "Normal"
+    if bmi > 25: bmi_status = "Overweight" 
+    if bmi > 30: bmi_status = "Obese (Low Impact Priority)"
+    
+    # B. Define Protocol (Sets/Reps) based on Goal & Age
+    warmup_time = "5-7 mins"
+    if age > 50: warmup_time = "10-12 mins (Joint Focus)"
+    
+    main_scheme = "3 sets x 12 reps"
+    rest = "60s"
+    
     if goal == "Muscle Gain":
-        protocol = "4 sets x 8-12 reps"
-        rest = "90 sec rest"
-        focus = "Hypertrophy (Focus on the squeeze)"
+        main_scheme = "4 sets x 8-10 reps (Slow Eccentric)"
+        rest = "90s"
     elif goal == "Strength":
-        protocol = "5 sets x 5 reps"
-        rest = "3 min rest"
-        focus = "Max Force (Lift heavy safely)"
+        main_scheme = "5 sets x 5 reps (Heavy)"
+        rest = "3 mins"
     elif goal == "Weight Loss":
-        protocol = "3 sets x 15-20 reps"
-        rest = "30-45 sec rest"
-        focus = "High Heart Rate (Keep moving)"
-    else: # General Health / Endurance
-        protocol = "3 sets x 12 reps"
-        rest = "60 sec rest"
-        focus = "Consistency & Form"
-
-    # 3. Adjust for Age
-    warmup_duration = "5 mins"
-    if age > 50:
-        warmup_duration = "10-12 mins (Joint mobilization focus)"
-        rest += " + 30 sec extra for recovery"
-
-    # --- B. Build the Routine Components ---
-    
-    # 1. Warmup
-    warmup = f"""
-    ⏱️ DURATION: {warmup_duration}
-    1. Joint Rotations (Neck, Shoulders, Hips, Ankles)
-    2. 3 min Light Cardio ({'Jumping Jacks' if injury != 'Knee' else 'March in place'})
-    3. Dynamic Stretch: {'Arm Circles' if 'Upper' in plan_category else 'Leg Swings'}
-    """
+        main_scheme = "3 sets x 15-20 reps (Circuit Style)"
+        rest = "30s"
 
-    # 2. Main Exercises Selection (Logic Tree)
-    exercises = []
+    # C. Select Exercises
+    legs = filter_exercises("Legs", injury, age, bmi, equipment)
+    push = filter_exercises("Push", injury, age, bmi, equipment)
+    pull = filter_exercises("Pull", injury, age, bmi, equipment)
+    core = filter_exercises("Core", injury, age, bmi, equipment)
+    cardio = filter_exercises("Cardio", injury, age, bmi, equipment)
     
-    # -- IF INJURED (Safety Priority) --
-    if injury == "Knee":
-        exercises = [
-            f"1. Glute Bridges (Floor): {protocol} (Safe for knees)",
-            f"2. Seated Overhead Press: {protocol} ({weight_text})",
-            f"3. Lat Pulldowns or Pull-ups: {protocol}",
-            f"4. Plank Hold: 3 sets x 45 sec"
-        ]
-    elif injury == "Back":
-        exercises = [
-            f"1. Wall Sits: 3 sets x 45 sec (Spine neutral)",
-            f"2. Supported Single-Arm Row: {protocol} ({weight_text})",
-            f"3. Seated Chest Press/Pushups (Incline): {protocol}",
-            f"4. Bird-Dog: 3 sets x 12 reps (Stability)"
-        ]
-    elif injury == "Shoulder":
-        exercises = [
-            f"1. Leg Press or Goblet Squat (Hands low): {protocol}",
-            f"2. Lunges: {protocol} (Bodyweight)",
-            f"3. Bicep Curls: {protocol} ({weight_text})",
-            f"4. Tricep Pushdowns (Elbows tucked): {protocol}"
-        ]
-        
-    # -- IF HEALTHY (Goal & Equipment Priority) --
-    else:
-        # Leg Exercise
-        if equipment == "Bodyweight Only": leg_ex = "Air Squats"
-        elif equipment == "Home Dumbbells": leg_ex = "Dumbbell Goblet Squats"
-        else: leg_ex = "Barbell Back Squats"
-        
-        # Push Exercise
-        if equipment == "Bodyweight Only": push_ex = "Push-ups"
-        elif equipment == "Home Dumbbells": push_ex = "Dumbbell Bench Press"
-        else: push_ex = "Barbell Bench Press"
-        
-        # Pull Exercise
-        if equipment == "Bodyweight Only": pull_ex = "Inverted Rows (using table)"
-        elif equipment == "Home Dumbbells": pull_ex = "Dumbbell Rows"
-        else: pull_ex = "Barbell Bent Over Row"
-
-        exercises = [
-            f"1. {leg_ex}: {protocol} using {weight_text}",
-            f"2. {push_ex}: {protocol}",
-            f"3. {pull_ex}: {protocol}",
-            f"4. {'Shoulder Press' if goal != 'Weight Loss' else 'Burpees'}: {protocol}"
-        ]
-
-    # 3. Finisher (Core/Cardio)
-    if goal in ["Weight Loss", "Athletic Performance", "Endurance"]:
-        finisher = """
-        🔥 CARDIO FINISHER:
-        - 30 sec High Knees / Mountain Climbers
-        - 30 sec Rest
-        - Repeat 5 times
-        """
-    else: # Muscle/Strength/Health
-        finisher = """
-        🧱 CORE & STABILITY FINISHER:
-        - Plank: 3 sets x failure
-        - Russian Twists: 3 sets x 20 reps
-        """
-
-    # 4. Cooldown
-    cooldown = """
-    🧘 COOLDOWN (Don't skip!):
-    1. Hamstring Stretch (30 sec per leg)
-    2. Chest Opener (Doorway stretch)
-    3. Child's Pose (Relieve back tension)
-    """
+    finisher = generate_finisher(goal, injury, equipment)
 
-    # --- C. Assemble Final Text ---
-    full_text = f"""
-    📋 PERSONALIZED PLAN: {plan_category}
-    ----------------------------------------------------
-    👤 PROFILE: {age}y/o {gender} | Goal: {goal} | Exp: {experience}
-    ⚖️ REC. WEIGHTS: {weight_text}
-    🎯 FOCUS: {focus}
-    ----------------------------------------------------
+    # D. Construct the Text
+    plan = f"📋 YOUR AI CUSTOM PLAN: {plan_name}\n"
+    plan += f"👤 Profile: {age}y | BMI: {bmi:.1f} ({bmi_status}) | Goal: {goal}\n"
+    plan += "---------------------------------------------------\n\n"
     
-    🔥 WARM-UP:
-    {warmup}
+    plan += f"🌡️ WARM-UP ({warmup_time}):\n"
+    plan += "1. Joint Rotations (Neck, Shoulders, Wrists, Ankles)\n"
+    if injury != "Knee": plan += "2. Light Jumping Jacks or Skipping (2 mins)\n"
+    else: plan += "2. March in Place (High Knees) (2 mins)\n"
+    plan += "3. Dynamic Stretch: Arm Swings & Leg Swings\n\n"
     
-    💪 MAIN WORKOUT (Rest: {rest}):
-    {chr(10).join(exercises)}
+    plan += f"💪 MAIN WORKOUT (Protocol: {main_scheme} | Rest: {rest}):\n"
+    plan += f"1. LEG FOCUS: {legs[0]}\n"
+    plan += f"2. PUSH FOCUS: {push[0]}\n"
+    plan += f"3. PULL FOCUS: {pull[0]}\n"
+    if len(legs) > 1: plan += f"4. SECONDARY LEGS: {legs[1]}\n"
+    if len(push) > 1: plan += f"5. SECONDARY UPPER: {push[1]}\n"
     
-    {finisher}
+    plan += "\n🛡️ CORE & STABILITY:\n"
+    plan += f"1. {core[0]}: 3 sets\n"
+    if len(core) > 1: plan += f"2. {core[1]}: 3 sets\n"
     
-    {cooldown}
-    """
+    plan += f"\n🏁 THE FINISHER (One Click Bonus):\n{finisher}\n"
     
-    return full_text
+    return plan
 
-# 3. Prediction Wrapper
-def predict_plan(age, gender, weight, height, goal, equipment, injury, experience):
-    # Predict Category
-    input_data = pd.DataFrame({
+# 5. GRADIO WRAPPER
+def predict_wrapper(age, gender, weight, height, goal, equipment, injury, experience):
+    # 1. AI Logic (The CSV Model)
+    input_df = pd.DataFrame({
         'Age': [age], 'Gender': [gender], 'Weight_kg': [weight], 
         'Height_cm': [height], 'Goal': [goal], 'Equipment': [equipment], 
         'Injury': [injury], 'Experience': [experience]
     })
-    plan_name = model.predict(input_data)[0]
+    plan_category = model.predict(input_df)[0]
     
-    # Generate Unique Text
-    routine = generate_dynamic_workout(plan_name, age, gender, weight, height, goal, equipment, injury, experience)
+    # 2. Rule Logic (The Python Generator)
+    detailed_plan = create_full_plan(plan_category, age, gender, weight, height, goal, equipment, injury, experience)
     
-    # Generate Tip
-    tip = f"AI Insight: We selected '{plan_name}' based on your safety needs, but customized the intensity for {goal}."
+    # 3. Strategy Note
+    note = f"🧠 AI Logic: Model detected '{plan_category}'. We adjusted volume for {goal} and removed dangerous exercises for {injury}."
     
-    return routine, tip
+    return detailed_plan, note
 
-# 4. Interface
+# 6. UI LAUNCH
 iface = gr.Interface(
-    fn=predict_plan,
+    fn=predict_wrapper,
     inputs=[
         gr.Slider(18, 80, value=25, label="Age"),
         gr.Radio(["Male", "Female"], label="Gender", value="Male"),
         gr.Number(label="Weight (kg)", value=75),
         gr.Number(label="Height (cm)", value=175),
-        gr.Dropdown(['Weight Loss', 'Muscle Gain', 'Endurance', 'General Health', 'Strength'], label="Goal", value="Muscle Gain"),
+        gr.Dropdown(['Weight Loss', 'Muscle Gain', 'Endurance', 'General Health', 'Strength'], label="Goal", value="General Health"),
         gr.Dropdown(['Gym Membership', 'Home Dumbbells', 'Bodyweight Only', 'Full Home Gym (Rack+Barbell)'], label="Equipment", value="Gym Membership"),
         gr.Dropdown(['None', 'Knee', 'Back', 'Shoulder', 'Ankle'], label="Injury", value="None"),
         gr.Radio(['Beginner', 'Intermediate', 'Advanced'], label="Experience", value="Intermediate")
     ],
     outputs=[
-        gr.Textbox(label="Your Custom Routine", lines=22),
-        gr.Textbox(label="AI Coach Note")
+        gr.Textbox(label="Your Ultimate Plan", lines=25),
+        gr.Textbox(label="Behind the Scenes")
     ],
-    title="SmartFit AI Ultimate 🚀",
-    description="A fully dynamic workout generator that calculates weights, sets, and exercises based on your exact profile.",
+    title="SmartFit AI Ultimate ⚡",
+    description="One-Click Personalized Workout Generator. Adapts to BMI, Injuries, and Equipment instantly.",
     theme="soft"
 )