Update app.py

app.py

@@ -3,264 +3,230 @@ import joblib
 import pandas as pd
 import random
 
-# 1. Load the AI Model
+# 1. Load Model
 model = joblib.load('fitness_model.joblib')
 
-# --- 2. THE EXTENDED EXERCISE DATABASE ---
-# Format: Name, Equipment, Type, Main Muscle, Bad For (Injury), Min Level (0=All, 1=Int, 2=Adv)
+# --- 2. THE INTELLIGENT EXERCISE DATABASE ---
+# base_ratio: How much of bodyweight a standard male can lift in this exercise.
+# min_level: 0=Beginner, 1=Intermediate, 2=Advanced
+
+DB = [
+    # --- CHEST ---
+    {"name": "Barbell Bench Press", "muscle": "Chest", "type": "Compound", "equip": ["Barbell", "Gym"], "bad_for": ["Shoulder"], "min_level": 1, "base_ratio": 0.8},
+    {"name": "Dumbbell Chest Press", "muscle": "Chest", "type": "Compound", "equip": ["Dumbbells", "Gym"], "bad_for": [], "min_level": 0, "base_ratio": 0.35}, # Per hand
+    {"name": "Machine Chest Press", "muscle": "Chest", "type": "Isolation", "equip": ["Gym"], "bad_for": [], "min_level": 0, "base_ratio": 0.6},
+    {"name": "Incline Dumbbell Press", "muscle": "Chest", "type": "Compound", "equip": ["Dumbbells", "Gym"], "bad_for": ["Shoulder"], "min_level": 1, "base_ratio": 0.3},
+    {"name": "Cable Crossovers", "muscle": "Chest", "type": "Isolation", "equip": ["Gym"], "bad_for": [], "min_level": 1, "base_ratio": 0.15},
+    {"name": "Push-ups", "muscle": "Chest", "type": "Compound", "equip": ["Bodyweight"], "bad_for": ["Wrist"], "min_level": 0, "base_ratio": 0},
+    {"name": "Weighted Dips", "muscle": "Chest", "type": "Compound", "equip": ["Gym", "Bodyweight"], "bad_for": ["Shoulder"], "min_level": 2, "base_ratio": 0},
+
+    # --- BACK ---
+    {"name": "Deadlift", "muscle": "Back", "type": "Compound", "equip": ["Barbell", "Gym"], "bad_for": ["Back"], "min_level": 2, "base_ratio": 1.2},
+    {"name": "Lat Pulldown", "muscle": "Back", "type": "Compound", "equip": ["Gym"], "bad_for": [], "min_level": 0, "base_ratio": 0.6},
+    {"name": "Seated Cable Row", "muscle": "Back", "type": "Isolation", "equip": ["Gym"], "bad_for": ["Back"], "min_level": 0, "base_ratio": 0.6},
+    {"name": "Barbell Bent Over Row", "muscle": "Back", "type": "Compound", "equip": ["Barbell", "Gym"], "bad_for": ["Back"], "min_level": 2, "base_ratio": 0.7},
+    {"name": "Single Arm Dumbbell Row", "muscle": "Back", "type": "Isolation", "equip": ["Dumbbells", "Gym"], "bad_for": [], "min_level": 0, "base_ratio": 0.3},
+    {"name": "Pull-ups", "muscle": "Back", "type": "Compound", "equip": ["Bodyweight", "Gym"], "bad_for": ["Shoulder"], "min_level": 1, "base_ratio": 0},
+
+    # --- LEGS ---
+    {"name": "Barbell Squat", "muscle": "Legs", "type": "Compound", "equip": ["Barbell", "Gym"], "bad_for": ["Knee", "Back"], "min_level": 2, "base_ratio": 1.0},
+    {"name": "Leg Press", "muscle": "Legs", "type": "Compound", "equip": ["Gym"], "bad_for": [], "min_level": 0, "base_ratio": 1.5},
+    {"name": "Goblet Squat", "muscle": "Legs", "type": "Compound", "equip": ["Dumbbells", "Gym"], "bad_for": ["Knee"], "min_level": 0, "base_ratio": 0.3},
+    {"name": "Walking Lunges", "muscle": "Legs", "type": "Compound", "equip": ["Dumbbells", "Bodyweight"], "bad_for": ["Knee"], "min_level": 1, "base_ratio": 0.15},
+    {"name": "Leg Extensions", "muscle": "Legs", "type": "Isolation", "equip": ["Gym"], "bad_for": ["Knee"], "min_level": 0, "base_ratio": 0.4},
+    {"name": "Romanian Deadlift", "muscle": "Legs", "type": "Compound", "equip": ["Barbell", "Dumbbells"], "bad_for": ["Back"], "min_level": 1, "base_ratio": 0.8},
+    {"name": "Lying Leg Curl", "muscle": "Legs", "type": "Isolation", "equip": ["Gym"], "bad_for": [], "min_level": 0, "base_ratio": 0.3},
+
+    # --- SHOULDERS ---
+    {"name": "Overhead Barbell Press", "muscle": "Shoulders", "type": "Compound", "equip": ["Barbell", "Gym"], "bad_for": ["Shoulder", "Back"], "min_level": 2, "base_ratio": 0.5},
+    {"name": "Seated Dumbbell Press", "muscle": "Shoulders", "type": "Compound", "equip": ["Dumbbells", "Gym"], "bad_for": ["Shoulder"], "min_level": 0, "base_ratio": 0.2},
+    {"name": "Lateral Raises", "muscle": "Shoulders", "type": "Isolation", "equip": ["Dumbbells", "Gym"], "bad_for": [], "min_level": 0, "base_ratio": 0.05}, # Light!
+    {"name": "Face Pulls", "muscle": "Shoulders", "type": "Isolation", "equip": ["Gym"], "bad_for": [], "min_level": 0, "base_ratio": 0.2},
+
+    # --- ARMS ---
+    {"name": "Barbell Curl", "muscle": "Arms", "type": "Isolation", "equip": ["Barbell", "Gym"], "bad_for": ["Wrist"], "min_level": 0, "base_ratio": 0.3},
+    {"name": "Dumbbell Hammer Curl", "muscle": "Arms", "type": "Isolation", "equip": ["Dumbbells", "Gym"], "bad_for": [], "min_level": 0, "base_ratio": 0.15},
+    {"name": "Tricep Rope Pushdown", "muscle": "Arms", "type": "Isolation", "equip": ["Gym"], "bad_for": [], "min_level": 0, "base_ratio": 0.3},
+    {"name": "Skullcrushers", "muscle": "Arms", "type": "Isolation", "equip": ["Barbell", "Gym"], "bad_for": ["Elbow"], "min_level": 1, "base_ratio": 0.25},
+    
+    # --- CORE ---
+    {"name": "Plank", "muscle": "Core", "type": "Static", "equip": ["Bodyweight"], "bad_for": [], "min_level": 0, "base_ratio": 0},
+    {"name": "Cable Woodchoppers", "muscle": "Core", "type": "Isolation", "equip": ["Gym"], "bad_for": ["Back"], "min_level": 1, "base_ratio": 0.3},
+    {"name": "Hanging Leg Raises", "muscle": "Core", "type": "Isolation", "equip": ["Gym"], "bad_for": ["Shoulder"], "min_level": 2, "base_ratio": 0}
+]
 
-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},
-    
-    # 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},
-    
-    # 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},
-    
-    # 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},
+# --- 3. THE CALCULATOR ENGINE ---
+
+def calculate_weight(base_ratio, user_weight, gender, experience):
+    if base_ratio == 0: return "Bodyweight"
     
-    # 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},
+    load = user_weight * base_ratio
     
-    # 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},
+    # Adjust for Gender
+    if gender == "Female": load *= 0.6
     
-    # 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},
+    # Adjust for Experience
+    if experience == "Beginner": load *= 0.7
+    elif experience == "Advanced": load *= 1.3
     
-    # 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},
+    # Round to nearest 2.5kg (standard plate size)
+    load = round(load / 2.5) * 2.5
     
-    # 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}
-]
-
-# --- 3. LOGIC FUNCTIONS ---
-
-def get_level_score(experience):
-    if experience == "Beginner": return 0
-    if experience == "Intermediate": return 1
-    return 2
+    return f"{int(load)} kg"
 
-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"]
+def get_exercises(target_muscle, target_type, user_equip, user_injury, user_level_score):
+    
+    # Define Equipment Lists
+    my_gear = []
+    if user_equip == "Gym Membership": my_gear = ["Gym", "Barbell", "Dumbbells", "Bodyweight"]
+    elif user_equip == "Full Home Gym (Rack+Barbell)": my_gear = ["Barbell", "Dumbbells", "Bodyweight"]
+    elif user_equip == "Home Dumbbells": my_gear = ["Dumbbells", "Bodyweight"]
+    else: my_gear = ["Bodyweight"]
 
-def filter_exercises(target_muscle, equipment_list, injury, user_level_score, exercise_type=None):
-    valid = []
-    for ex in EXERCISE_DB:
-        # Muscle Match
+    candidates = []
+    
+    for ex in DB:
+        # 1. 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
+        # 2. Type Match (Compound vs Isolation) - Optional but good for structure
+        if target_type and ex["type"] != target_type: continue
         
-        # Type Match (Compound vs Isolation) - Optional
-        if exercise_type and ex["type"] != exercise_type: continue
+        # 3. Equipment Match
+        has_gear = False
+        for gear in ex["equip"]:
+            if gear in my_gear:
+                has_gear = True
+                break
+        if not has_gear: continue
         
-        valid.append(ex["name"])
+        # 4. Injury Check
+        if user_injury in ex["bad_for"]: continue
         
-    return valid
-
-def generate_parameters(goal, experience, age):
-    # Calculates Sets/Reps/Rest based on exact inputs
-    sets = 3
-    reps = "10-12"
-    rest = "60s"
-    
-    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 == "Weight Loss":
-        sets = 3
-        reps = "15-20"
-        rest = "30s"
-    
-    elif goal == "Endurance":
-        sets = 2
-        reps = "20+"
-        rest = "15s"
+        # 5. Level Check (Strict)
+        # Beginner (0) cannot do Advanced (2)
+        if user_level_score < ex["min_level"]: continue
         
-    # Age Adjustment
-    if age >= 50:
-        rest = f"{rest} (+30s)"
-        if goal == "Strength": sets -= 1 # Reduce volume for seniors
+        candidates.append(ex)
         
-    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)
+    return candidates
 
-# --- 4. MAIN WORKOUT BUILDER ---
+# --- 4. ROUTINE GENERATOR ---
 
-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)
+def generate_routine(plan_name, age, gender, weight, goal, equipment, injury, experience):
     
-    # B. Define Routine Skeleton based on Body Part Split
-    # Logic: AI Predicted "Upper Body" -> We give Chest+Back+Shoulders
+    # A. Parse Level
+    level_score = 0
+    if experience == "Intermediate": level_score = 1
+    if experience == "Advanced": level_score = 2
     
-    skeleton = [] # List of (Muscle, Type) tuples
+    # B. Define Structure based on AI Plan
+    # We create slots: (Muscle, Type)
+    slots = []
     
     if "Upper" in plan_name or "Push" in plan_name:
-        skeleton = [
+        slots = [
             ("Chest", "Compound"), ("Back", "Compound"), 
             ("Shoulders", "Compound"), ("Chest", "Isolation"), 
-            ("Back", "Isolation"), ("Triceps", "Isolation"), ("Biceps", "Isolation")
+            ("Back", "Isolation"), ("Arms", "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")
+        slots = [
+            ("Legs", "Compound"), ("Legs", "Compound"), # Two big lifts
+            ("Legs", "Isolation"), ("Legs", "Isolation"),
+            ("Core", "Isolation")
         ]
-        title = "LOWER BODY & CORE"
-        
-    else: # Full Body / Generic
-        skeleton = [
-            ("Quads", "Compound"), ("Chest", "Compound"),
-            ("Back", "Compound"), ("Hamstrings", "Isolation"),
-            ("Shoulders", "Isolation"), ("Core", "Isolation")
+    else: # Full Body
+        slots = [
+            ("Legs", "Compound"), ("Chest", "Compound"),
+            ("Back", "Compound"), ("Shoulders", "Compound"),
+            ("Arms", "Isolation"), ("Core", "Static")
         ]
-        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"
+    # C. Build Output
+    text = f"CUSTOM WORKOUT PLAN\n"
+    text += f"Config: {gender}, {weight}kg, {experience}, {equipment}\n"
+    text += f"Focus: {goal} | Avoid: {injury}\n"
+    text += "="*40 + "\n\n"
     
-    text += f"PROTOCOL: {params}\n\n"
+    # Sets/Reps Logic
+    sets = "3"
+    reps = "10"
+    rest = "60s"
+    if goal == "Strength": sets="5"; reps="5"; rest="3 min"
+    if goal == "Muscle Gain": sets="4"; reps="8-12"; rest="90s"
+    if goal == "Weight Loss": sets="3"; reps="15+"; rest="30s"
     
-    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 += f"GLOBAL PROTOCOL: {sets} Sets | {reps} Reps | {rest} Rest\n\n"
+    text += "EXERCISES:\n"
     
-    text += "MAIN EXERCISES:\n"
+    used_names = [] # To prevent duplicates
     
-    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"
+    for i, (muscle, ex_type) in enumerate(slots):
+        # 1. Get Options
+        options = get_exercises(muscle, ex_type, equipment, injury, level_score)
+        
+        # 2. Fallback if empty (e.g., no compound chest for bodyweight beginner)
+        if not options:
+            options = get_exercises(muscle, None, equipment, injury, level_score)
+            
+        # 3. Random Selection (Variety!)
+        if not options:
+            selected = {"name": f"Generic {muscle} Move", "base_ratio": 0}
+        else:
+            # Shuffle and pick one that hasn't been used
+            random.shuffle(options)
+            selected = options[0]
+            for opt in options:
+                if opt["name"] not in used_names:
+                    selected = opt
+                    break
+        
+        used_names.append(selected["name"])
+        
+        # 4. Calculate Weight
+        rec_weight = calculate_weight(selected.get("base_ratio", 0), weight, gender, experience)
+        
+        text += f"{i+1}. {selected['name']}\n"
+        text += f"   Target Load: {rec_weight}\n"
+        text += f"   Type: {muscle} {selected.get('type', '')}\n\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
 
-# 5. GRADIO INTERFACE
-def predict(age, gender, weight, height, goal, equipment, injury, experience):
-    # AI Prediction
-    input_data = pd.DataFrame({
+# 5. WRAPPER
+def predict_wrapper(age, gender, weight, height, goal, equipment, injury, experience):
+    # AI Logic
+    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]
-    
-    # Python Generation
-    final_routine = generate_workout(plan_name, age, weight, goal, equipment, injury, experience)
+    plan_name = model.predict(input_df)[0]
     
-    logic_note = f"System detected '{plan_name}'. Parameters adjusted for {experience} level."
+    # Generator Logic
+    routine = generate_routine(plan_name, age, gender, weight, goal, equipment, injury, experience)
     
-    return final_routine, logic_note
+    return routine, f"AI Strategy: Selected '{plan_name}' based on injury profile."
 
-# Launch
+# 6. LAUNCH
 iface = gr.Interface(
-    fn=predict,
+    fn=predict_wrapper,
     inputs=[
-        gr.Slider(18, 80, step=1, value=25, label="Age"), # Step=1 forces integers
+        gr.Slider(18, 80, step=1, 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="General Health"),
+        gr.Dropdown(['Weight Loss', 'Muscle Gain', 'Endurance', 'General Health', 'Strength'], label="Goal", value="Muscle Gain"),
         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="Generated Workout", lines=18),
-        gr.Textbox(label="Logic Trace")
+        gr.Textbox(label="Result", lines=20),
+        gr.Textbox(label="Note")
     ],
-    title="SmartFit AI - Pro Edition",
-    description="Generates strict, equipment-specific workouts based on AI analysis.",
+    title="SmartFit AI - Calculator Edition",
+    description="Generates strict workouts with specific weight recommendations.",
     theme="soft"
 )