Update app.py

app.py

@@ -3,219 +3,250 @@ import joblib
 import pandas as pd
 import random
 
-# 1. Load Model (The "Brain" trained on Synthetic Data)
+# 1. Load the AI Model
 model = joblib.load('fitness_model.joblib')
 
-# --- 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 = []
+# --- 2. THE EXTENDED EXERCISE DATABASE ---
+# Format: Name, Equipment, Type, Main Muscle, Bad For (Injury), Min Level (0=All, 1=Int, 2=Adv)
+
+EXERCISE_DB = [
+    # LEGS - QUAD DOMINANT
+    {"name": "Barbell Back Squat", "equip": ["Barbell", "Gym"], "type": "Compound", "muscle": "Quads", "bad_for": ["Back", "Knee"], "level": 2},
+    {"name": "Goblet Squat", "equip": ["Dumbbells", "Gym"], "type": "Compound", "muscle": "Quads", "bad_for": ["Knee"], "level": 0},
+    {"name": "Leg Press Machine", "equip": ["Gym"], "type": "Compound", "muscle": "Quads", "bad_for": [], "level": 0},
+    {"name": "Bulgarian Split Squat", "equip": ["Dumbbells", "Bodyweight"], "type": "Compound", "muscle": "Quads", "bad_for": ["Knee"], "level": 2},
+    {"name": "Step-Ups", "equip": ["Bodyweight", "Dumbbells"], "type": "Isolation", "muscle": "Quads", "bad_for": ["Knee"], "level": 0},
+    {"name": "Leg Extensions", "equip": ["Gym"], "type": "Isolation", "muscle": "Quads", "bad_for": ["Knee"], "level": 0},
     
-    # 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)
+    # LEGS - GLUTE/HAMSTRING DOMINANT
+    {"name": "Romanian Deadlift", "equip": ["Barbell", "Dumbbells"], "type": "Compound", "muscle": "Hamstrings", "bad_for": ["Back"], "level": 1},
+    {"name": "Glute Bridges", "equip": ["Bodyweight"], "type": "Isolation", "muscle": "Glutes", "bad_for": [], "level": 0},
+    {"name": "Hip Thrusts", "equip": ["Barbell", "Dumbbells", "Gym"], "type": "Compound", "muscle": "Glutes", "bad_for": ["Back"], "level": 1},
+    {"name": "Lying Leg Curl", "equip": ["Gym"], "type": "Isolation", "muscle": "Hamstrings", "bad_for": [], "level": 0},
+    {"name": "Walking Lunges", "equip": ["Bodyweight", "Dumbbells"], "type": "Compound", "muscle": "Glutes", "bad_for": ["Knee"], "level": 1},
     
-    # If list is empty (strict constraints), give a generic fallback
-    if not valid: return [f"Standard {category} Movement (Low Impact)"]
+    # PUSH - CHEST
+    {"name": "Barbell Bench Press", "equip": ["Barbell", "Gym"], "type": "Compound", "muscle": "Chest", "bad_for": ["Shoulder"], "level": 1},
+    {"name": "Dumbbell Chest Press", "equip": ["Dumbbells", "Gym"], "type": "Compound", "muscle": "Chest", "bad_for": [], "level": 0},
+    {"name": "Push-ups", "equip": ["Bodyweight"], "type": "Compound", "muscle": "Chest", "bad_for": ["Shoulder", "Wrist"], "level": 0},
+    {"name": "Incline Dumbbell Press", "equip": ["Dumbbells", "Gym"], "type": "Compound", "muscle": "Chest", "bad_for": ["Shoulder"], "level": 1},
+    {"name": "Cable Flys", "equip": ["Gym"], "type": "Isolation", "muscle": "Chest", "bad_for": [], "level": 1},
+    {"name": "Machine Chest Press", "equip": ["Gym"], "type": "Compound", "muscle": "Chest", "bad_for": [], "level": 0},
     
-    # 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 = []
+    # PUSH - SHOULDERS
+    {"name": "Overhead Barbell Press", "equip": ["Barbell", "Gym"], "type": "Compound", "muscle": "Shoulders", "bad_for": ["Shoulder", "Back"], "level": 2},
+    {"name": "Seated Dumbbell Press", "equip": ["Dumbbells", "Gym"], "type": "Compound", "muscle": "Shoulders", "bad_for": ["Shoulder"], "level": 0},
+    {"name": "Lateral Raises", "equip": ["Dumbbells", "Gym"], "type": "Isolation", "muscle": "Shoulders", "bad_for": [], "level": 0},
+    {"name": "Front Raises", "equip": ["Dumbbells", "Gym"], "type": "Isolation", "muscle": "Shoulders", "bad_for": [], "level": 0},
+    {"name": "Face Pulls", "equip": ["Gym"], "type": "Isolation", "muscle": "Rear Delts", "bad_for": [], "level": 0},
     
-    # 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:
-        options.append("🌀 FOAM ROLL: Full body flush (Calves, Quads, Upper Back)")
+    # PUSH - TRICEPS
+    {"name": "Tricep Dips", "equip": ["Bodyweight", "Gym"], "type": "Compound", "muscle": "Triceps", "bad_for": ["Shoulder"], "level": 1},
+    {"name": "Skullcrushers", "equip": ["Barbell", "Dumbbells"], "type": "Isolation", "muscle": "Triceps", "bad_for": ["Elbow"], "level": 1},
+    {"name": "Cable Tricep Pushdown", "equip": ["Gym"], "type": "Isolation", "muscle": "Triceps", "bad_for": [], "level": 0},
+    {"name": "Diamond Push-ups", "equip": ["Bodyweight"], "type": "Compound", "muscle": "Triceps", "bad_for": ["Wrist"], "level": 2},
+    
+    # PULL - BACK
+    {"name": "Pull-ups", "equip": ["Bodyweight", "Gym"], "type": "Compound", "muscle": "Back", "bad_for": ["Shoulder"], "level": 2},
+    {"name": "Lat Pulldown Machine", "equip": ["Gym"], "type": "Compound", "muscle": "Back", "bad_for": [], "level": 0},
+    {"name": "Barbell Bent Over Row", "equip": ["Barbell", "Gym"], "type": "Compound", "muscle": "Back", "bad_for": ["Back"], "level": 2},
+    {"name": "One Arm Dumbbell Row", "equip": ["Dumbbells", "Gym"], "type": "Compound", "muscle": "Back", "bad_for": [], "level": 0},
+    {"name": "Seated Cable Row", "equip": ["Gym"], "type": "Compound", "muscle": "Back", "bad_for": ["Back"], "level": 1},
+    {"name": "Inverted Row", "equip": ["Bodyweight"], "type": "Compound", "muscle": "Back", "bad_for": [], "level": 0},
+    
+    # PULL - BICEPS
+    {"name": "Barbell Curl", "equip": ["Barbell", "Gym"], "type": "Isolation", "muscle": "Biceps", "bad_for": ["Wrist"], "level": 0},
+    {"name": "Hammer Curls", "equip": ["Dumbbells", "Gym"], "type": "Isolation", "muscle": "Biceps", "bad_for": [], "level": 0},
+    {"name": "Incline Dumbbell Curl", "equip": ["Dumbbells", "Gym"], "type": "Isolation", "muscle": "Biceps", "bad_for": [], "level": 0},
+    {"name": "Chin-ups", "equip": ["Bodyweight", "Gym"], "type": "Compound", "muscle": "Biceps", "bad_for": ["Shoulder"], "level": 2},
+    
+    # CORE
+    {"name": "Plank", "equip": ["Bodyweight"], "type": "Isolation", "muscle": "Core", "bad_for": [], "level": 0},
+    {"name": "Hanging Leg Raises", "equip": ["Gym", "Barbell"], "type": "Isolation", "muscle": "Core", "bad_for": ["Shoulder"], "level": 2},
+    {"name": "Russian Twists", "equip": ["Bodyweight", "Dumbbells"], "type": "Isolation", "muscle": "Core", "bad_for": ["Back"], "level": 0},
+    {"name": "Ab Wheel Rollout", "equip": ["Gym", "Dumbbells"], "type": "Isolation", "muscle": "Core", "bad_for": ["Back"], "level": 2},
+    {"name": "Dead Bug", "equip": ["Bodyweight"], "type": "Isolation", "muscle": "Core", "bad_for": [], "level": 0},
+    
+    # CARDIO
+    {"name": "Treadmill Run", "equip": ["Gym"], "type": "Cardio", "muscle": "Cardio", "bad_for": ["Knee"], "level": 0},
+    {"name": "Burpees", "equip": ["Bodyweight"], "type": "Cardio", "muscle": "Cardio", "bad_for": ["Knee", "Back"], "level": 2},
+    {"name": "Stationary Bike", "equip": ["Gym"], "type": "Cardio", "muscle": "Cardio", "bad_for": [], "level": 0},
+    {"name": "Jumping Jacks", "equip": ["Bodyweight"], "type": "Cardio", "muscle": "Cardio", "bad_for": ["Knee", "Ankle"], "level": 0},
+    {"name": "Mountain Climbers", "equip": ["Bodyweight"], "type": "Cardio", "muscle": "Cardio", "bad_for": ["Shoulder"], "level": 1}
+]
 
-    # 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")
+# --- 3. LOGIC FUNCTIONS ---
 
-    # 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")
+def get_level_score(experience):
+    if experience == "Beginner": return 0
+    if experience == "Intermediate": return 1
+    return 2
 
-    return random.choice(options)
+def get_available_equipment(user_equip):
+    if user_equip == "Gym Membership": return ["Gym", "Barbell", "Dumbbells", "Bodyweight"]
+    if user_equip == "Full Home Gym (Rack+Barbell)": return ["Barbell", "Dumbbells", "Bodyweight"]
+    if user_equip == "Home Dumbbells": return ["Dumbbells", "Bodyweight"]
+    return ["Bodyweight"]
 
-# --- 4. MAIN GENERATOR ---
+def filter_exercises(target_muscle, equipment_list, injury, user_level_score, exercise_type=None):
+    valid = []
+    for ex in EXERCISE_DB:
+        # Muscle Match
+        if ex["muscle"] != target_muscle: continue
+        
+        # Equipment Match
+        has_equip = any(eq in equipment_list for eq in ex["equip"])
+        if not has_equip: continue
+        
+        # Injury Match
+        if injury in ex["bad_for"]: continue
+        
+        # Level Match (Don't give advanced moves to beginners)
+        if ex["level"] > user_level_score: continue
+        
+        # Type Match (Compound vs Isolation) - Optional
+        if exercise_type and ex["type"] != exercise_type: continue
+        
+        valid.append(ex["name"])
+        
+    return valid
 
-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"
+def generate_parameters(goal, experience, age):
+    # Calculates Sets/Reps/Rest based on exact inputs
+    sets = 3
+    reps = "10-12"
     rest = "60s"
     
-    if goal == "Muscle Gain":
-        main_scheme = "4 sets x 8-10 reps (Slow Eccentric)"
+    if goal == "Strength":
+        sets = 5
+        reps = "3-5"
+        rest = "3 min"
+        if experience == "Beginner": sets = 3; reps = "5-8" # Safe start
+    
+    elif goal == "Muscle Gain":
+        sets = 4
+        reps = "8-12"
         rest = "90s"
-    elif goal == "Strength":
-        main_scheme = "5 sets x 5 reps (Heavy)"
-        rest = "3 mins"
+    
     elif goal == "Weight Loss":
-        main_scheme = "3 sets x 15-20 reps (Circuit Style)"
+        sets = 3
+        reps = "15-20"
         rest = "30s"
+    
+    elif goal == "Endurance":
+        sets = 2
+        reps = "20+"
+        rest = "15s"
+        
+    # Age Adjustment
+    if age >= 50:
+        rest = f"{rest} (+30s)"
+        if goal == "Strength": sets -= 1 # Reduce volume for seniors
+        
+    return f"{sets} Sets | {reps} Reps | {rest} Rest"
+
+def select_exercise(muscle, equip, injury, level, type_pref, fallback_name):
+    # Tries to find specific type (Compound), falls back to any valid
+    options = filter_exercises(muscle, equip, injury, level, type_pref)
+    if not options:
+        options = filter_exercises(muscle, equip, injury, level) # Try any type
+    
+    if not options:
+        return f"Standard {muscle} Exercise (Safe Mode)"
+    
+    return random.choice(options)
+
+# --- 4. MAIN WORKOUT BUILDER ---
+
+def generate_workout(plan_name, age, weight, goal, equipment, injury, experience):
+    # A. Setup
+    level_score = get_level_score(experience)
+    equip_list = get_available_equipment(equipment)
+    params = generate_parameters(goal, experience, age)
+    
+    # B. Define Routine Skeleton based on Body Part Split
+    # Logic: AI Predicted "Upper Body" -> We give Chest+Back+Shoulders
+    
+    skeleton = [] # List of (Muscle, Type) tuples
+    
+    if "Upper" in plan_name or "Push" in plan_name:
+        skeleton = [
+            ("Chest", "Compound"), ("Back", "Compound"), 
+            ("Shoulders", "Compound"), ("Chest", "Isolation"), 
+            ("Back", "Isolation"), ("Triceps", "Isolation"), ("Biceps", "Isolation")
+        ]
+        title = "UPPER BODY POWER"
+        
+    elif "Lower" in plan_name or "Legs" in plan_name:
+        skeleton = [
+            ("Quads", "Compound"), ("Hamstrings", "Compound"),
+            ("Glutes", "Compound"), ("Quads", "Isolation"),
+            ("Hamstrings", "Isolation"), ("Core", "Isolation")
+        ]
+        title = "LOWER BODY & CORE"
+        
+    else: # Full Body / Generic
+        skeleton = [
+            ("Quads", "Compound"), ("Chest", "Compound"),
+            ("Back", "Compound"), ("Hamstrings", "Isolation"),
+            ("Shoulders", "Isolation"), ("Core", "Isolation")
+        ]
+        title = "FULL BODY CONDITIONING"
+
+    # C. Build Text
+    text = f"WORKOUT PLAN: {title}\n"
+    text += f"Profile: {int(age)} years old | Level: {experience} | Goal: {goal}\n"
+    text += f"Config: {equipment} | Injury Avoidance: {injury}\n"
+    text += "-" * 50 + "\n\n"
+    
+    text += f"PROTOCOL: {params}\n\n"
+    
+    text += "WARM-UP (5-7 mins):\n"
+    if injury == "Knee":
+        text += "1. Arm Circles & Torso Rotation\n2. Glute Bridges\n3. Cat-Cow Stretch\n\n"
+    else:
+        text += "1. Jumping Jacks (2 mins)\n2. Bodyweight Squats\n3. Arm Swings\n\n"
+    
+    text += "MAIN EXERCISES:\n"
+    
+    for i, (muscle, ex_type) in enumerate(skeleton):
+        ex_name = select_exercise(muscle, equip_list, injury, level_score, ex_type, "Generic Move")
+        text += f"{i+1}. {ex_name} ({muscle} - {ex_type})\n"
+        
+    text += "\nFINISHER (Burnout):\n"
+    if goal in ["Weight Loss", "Cardio"]:
+        cardio_opts = filter_exercises("Cardio", equip_list, injury, level_score)
+        finisher = random.choice(cardio_opts) if cardio_opts else "March in Place"
+        text += f"--> {finisher}: 3 Rounds x 45 Seconds (Max Effort)"
+    else:
+        core_opts = filter_exercises("Core", equip_list, injury, level_score)
+        finisher = random.choice(core_opts) if core_opts else "Plank"
+        text += f"--> {finisher}: 3 Sets x Failure"
+
+    return text
 
-    # 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)
-    
-    finisher = generate_finisher(goal, injury, equipment)
-
-    # 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"
-    
-    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"
-    
-    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"
-    
-    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"
-    
-    plan += f"\n🏁 THE FINISHER (One Click Bonus):\n{finisher}\n"
-    
-    return plan
-
-# 5. GRADIO WRAPPER
-def predict_wrapper(age, gender, weight, height, goal, equipment, injury, experience):
-    # 1. AI Logic (The CSV Model)
-    input_df = pd.DataFrame({
+# 5. GRADIO INTERFACE
+def predict(age, gender, weight, height, goal, equipment, injury, experience):
+    # AI Prediction
+    input_data = pd.DataFrame({
         'Age': [age], 'Gender': [gender], 'Weight_kg': [weight], 
         'Height_cm': [height], 'Goal': [goal], 'Equipment': [equipment], 
         'Injury': [injury], 'Experience': [experience]
     })
-    plan_category = model.predict(input_df)[0]
+    plan_name = model.predict(input_data)[0]
     
-    # 2. Rule Logic (The Python Generator)
-    detailed_plan = create_full_plan(plan_category, age, gender, weight, height, goal, equipment, injury, experience)
+    # Python Generation
+    final_routine = generate_workout(plan_name, age, weight, goal, equipment, injury, experience)
     
-    # 3. Strategy Note
-    note = f"🧠 AI Logic: Model detected '{plan_category}'. We adjusted volume for {goal} and removed dangerous exercises for {injury}."
+    logic_note = f"System detected '{plan_name}'. Parameters adjusted for {experience} level."
     
-    return detailed_plan, note
+    return final_routine, logic_note
 
-# 6. UI LAUNCH
+# Launch
 iface = gr.Interface(
-    fn=predict_wrapper,
+    fn=predict,
     inputs=[
-        gr.Slider(18, 80, value=25, label="Age"),
+        gr.Slider(18, 80, step=1, value=25, label="Age"), # Step=1 forces integers
         gr.Radio(["Male", "Female"], label="Gender", value="Male"),
         gr.Number(label="Weight (kg)", value=75),
         gr.Number(label="Height (cm)", value=175),
@@ -225,11 +256,11 @@ iface = gr.Interface(
         gr.Radio(['Beginner', 'Intermediate', 'Advanced'], label="Experience", value="Intermediate")
     ],
     outputs=[
-        gr.Textbox(label="Your Ultimate Plan", lines=25),
-        gr.Textbox(label="Behind the Scenes")
+        gr.Textbox(label="Generated Workout", lines=18),
+        gr.Textbox(label="Logic Trace")
     ],
-    title="SmartFit AI Ultimate ⚡",
-    description="One-Click Personalized Workout Generator. Adapts to BMI, Injuries, and Equipment instantly.",
+    title="SmartFit AI - Pro Edition",
+    description="Generates strict, equipment-specific workouts based on AI analysis.",
     theme="soft"
 )