Update app.py

app.py

@@ -6,196 +6,252 @@ import random
 # 1. Load Model
 model = joblib.load('fitness_model.joblib')
 
-# --- 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}
-]
-
-# --- 3. THE CALCULATOR ENGINE ---
-
-def calculate_weight(base_ratio, user_weight, gender, experience):
+# ==========================================
+#        CONFIG DICTIONARIES (THE BRAIN)
+# ==========================================
+
+# 1. GOAL CONFIGURATION
+# Defines the structure (slots), sets, reps, and specific instructions per goal.
+GOAL_CONFIG = {
+    "Strength": {
+        "sets": 5, "reps": "3-5", "rest": "3-5 min",
+        "intensity": "85-90% 1RM",
+        "structure_modifier": "Focus on Compound Lifts",
+        "finisher": "Core Static Hold"
+    },
+    "Muscle Gain": {
+        "sets": 3, "reps": "8-12", "rest": "60-90 sec",
+        "intensity": "70-80% 1RM",
+        "structure_modifier": "Mix of Compound and Isolation",
+        "finisher": "Pump Work (High Reps)"
+    },
+    "Weight Loss": {
+        "sets": 3, "reps": "15-20", "rest": "30-45 sec",
+        "intensity": "60% 1RM",
+        "structure_modifier": "Circuit Style (Minimal Rest)",
+        "finisher": "High Intensity Interval (HIIT)"
+    },
+    "Endurance": {
+        "sets": 2, "reps": "20+", "rest": "30 sec",
+        "intensity": "50% 1RM",
+        "structure_modifier": "High Volume",
+        "finisher": "Steady State Cardio"
+    },
+    "General Health": {
+        "sets": 3, "reps": "12-15", "rest": "60 sec",
+        "intensity": "Moderate",
+        "structure_modifier": "Balanced Full Body",
+        "finisher": "Mobility Flow"
+    }
+}
+
+# 2. AGE CONFIGURATION
+# Adjusts warmup and safety parameters based on age groups.
+AGE_CONFIG = {
+    "Young (18-35)": {"warmup": "5 min", "rest_mod": 0, "type": "Standard"},
+    "Middle (36-55)": {"warmup": "8 min", "rest_mod": 15, "type": "Joint Prep Focus"},
+    "Senior (56+)": {"warmup": "12 min", "rest_mod": 30, "type": "Low Impact & Stability"}
+}
+
+# 3. LEVEL CONFIGURATION
+# Defines which difficulty tier (0-2) a user can access.
+LEVEL_CONFIG = {
+    "Beginner": {"tier": 0, "complexity": "Low (Machines/Stable)"},
+    "Intermediate": {"tier": 1, "complexity": "Medium (Free Weights)"},
+    "Advanced": {"tier": 2, "complexity": "High (Complex/Explosive)"}
+}
+
+# 4. EXERCISE DATABASE (Categorized)
+# Level 0 = Easy/Safe, Level 1 = Medium, Level 2 = Hard
+EXERCISE_DB = {
+    "Chest": [
+        {"name": "Machine Chest Press", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.6},
+        {"name": "Push-ups (Knees)", "equip": ["Bodyweight"], "bad": ["Wrist"], "level": 0, "ratio": 0},
+        {"name": "Push-ups (Regular)", "equip": ["Bodyweight"], "bad": ["Wrist"], "level": 1, "ratio": 0},
+        {"name": "Dumbbell Bench Press", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.35},
+        {"name": "Barbell Bench Press", "equip": ["Barbell", "Gym"], "bad": ["Shoulder"], "level": 1, "ratio": 0.9},
+        {"name": "Incline Dumbbell Press", "equip": ["Dumbbells", "Gym"], "bad": ["Shoulder"], "level": 1, "ratio": 0.3},
+        {"name": "Weighted Dips", "equip": ["Gym"], "bad": ["Shoulder"], "level": 2, "ratio": 0},
+        {"name": "Cable Flys", "equip": ["Gym"], "bad": [], "level": 1, "ratio": 0.2},
+    ],
+    "Back": [
+        {"name": "Lat Pulldown", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.6},
+        {"name": "Seated Cable Row", "equip": ["Gym"], "bad": ["Back"], "level": 0, "ratio": 0.6},
+        {"name": "Chest Supported Row", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.5},
+        {"name": "Single Arm Dumbbell Row", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.3},
+        {"name": "Pull-ups", "equip": ["Bodyweight", "Gym"], "bad": ["Shoulder"], "level": 1, "ratio": 0},
+        {"name": "Deadlift", "equip": ["Barbell", "Gym"], "bad": ["Back"], "level": 2, "ratio": 1.4},
+        {"name": "Barbell Row", "equip": ["Barbell", "Gym"], "bad": ["Back"], "level": 1, "ratio": 0.8},
+        {"name": "Inverted Row", "equip": ["Bodyweight"], "bad": [], "level": 0, "ratio": 0},
+    ],
+    "Legs": [
+        {"name": "Bodyweight Squats", "equip": ["Bodyweight"], "bad": ["Knee"], "level": 0, "ratio": 0},
+        {"name": "Goblet Squat", "equip": ["Dumbbells", "Gym"], "bad": ["Knee"], "level": 0, "ratio": 0.4},
+        {"name": "Leg Press", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 1.8},
+        {"name": "Lunge Walk", "equip": ["Bodyweight", "Dumbbells"], "bad": ["Knee"], "level": 1, "ratio": 0},
+        {"name": "Bulgarian Split Squat", "equip": ["Dumbbells", "Bodyweight"], "bad": ["Knee"], "level": 2, "ratio": 0.2},
+        {"name": "Barbell Back Squat", "equip": ["Barbell", "Gym"], "bad": ["Back", "Knee"], "level": 2, "ratio": 1.2},
+        {"name": "Romanian Deadlift", "equip": ["Barbell", "Dumbbells"], "bad": ["Back"], "level": 1, "ratio": 0.9},
+        {"name": "Leg Curls", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.4},
+        {"name": "Glute Bridges", "equip": ["Bodyweight"], "bad": [], "level": 0, "ratio": 0},
+    ],
+    "Shoulders": [
+        {"name": "Seated Dumbbell Press", "equip": ["Dumbbells", "Gym"], "bad": ["Shoulder"], "level": 0, "ratio": 0.25},
+        {"name": "Lateral Raises", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.1},
+        {"name": "Face Pulls", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.2},
+        {"name": "Overhead Barbell Press", "equip": ["Barbell", "Gym"], "bad": ["Shoulder", "Back"], "level": 2, "ratio": 0.5},
+        {"name": "Pike Pushups", "equip": ["Bodyweight"], "bad": ["Shoulder"], "level": 1, "ratio": 0},
+    ],
+    "Arms": [
+        {"name": "Dumbbell Curls", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.15},
+        {"name": "Tricep Pushdowns", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.3},
+        {"name": "Barbell Curls", "equip": ["Barbell", "Gym"], "bad": ["Wrist"], "level": 0, "ratio": 0.25},
+        {"name": "Skullcrushers", "equip": ["Barbell", "Dumbbells"], "bad": ["Elbow"], "level": 1, "ratio": 0.2},
+        {"name": "Bench Dips", "equip": ["Bodyweight"], "bad": ["Shoulder"], "level": 0, "ratio": 0},
+    ],
+    "Core": [
+        {"name": "Plank", "equip": ["Bodyweight"], "bad": [], "level": 0, "ratio": 0},
+        {"name": "Dead Bug", "equip": ["Bodyweight"], "bad": [], "level": 0, "ratio": 0},
+        {"name": "Hanging Leg Raises", "equip": ["Gym"], "bad": ["Shoulder"], "level": 2, "ratio": 0},
+        {"name": "Cable Woodchoppers", "equip": ["Gym"], "bad": ["Back"], "level": 1, "ratio": 0.3},
+    ],
+    "Cardio": [
+        {"name": "Walking", "equip": ["Bodyweight", "Gym"], "bad": [], "level": 0, "ratio": 0},
+        {"name": "Elliptical", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0},
+        {"name": "Burpees", "equip": ["Bodyweight"], "bad": ["Back", "Knee"], "level": 2, "ratio": 0},
+        {"name": "Sprints", "equip": ["Bodyweight"], "bad": ["Knee"], "level": 1, "ratio": 0},
+    ]
+}
+
+# ==========================================
+#           LOGIC ENGINE
+# ==========================================
+
+def get_age_group(age):
+    if age <= 35: return "Young (18-35)"
+    if age <= 55: return "Middle (36-55)"
+    return "Senior (56+)"
+
+def calculate_weight(base_ratio, user_weight, gender, level_tier):
     if base_ratio == 0: return "Bodyweight"
     
     load = user_weight * base_ratio
     
-    # Adjust for Gender
+    # Adjustments
     if gender == "Female": load *= 0.6
+    if level_tier == 0: load *= 0.6  # Beginner
+    if level_tier == 2: load *= 1.3  # Advanced
     
-    # Adjust for Experience
-    if experience == "Beginner": load *= 0.7
-    elif experience == "Advanced": load *= 1.3
-    
-    # Round to nearest 2.5kg (standard plate size)
-    load = round(load / 2.5) * 2.5
-    
+    # Rounding
+    load = 2.5 * round(load / 2.5)
     return f"{int(load)} kg"
 
-def get_exercises(target_muscle, target_type, user_equip, user_injury, user_level_score):
-    
-    # Define Equipment Lists
+def get_exercises(muscle, equipment, injury, level_tier, count=1):
+    # 1. Define Equipment List
     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"]
+    if equipment == "Gym Membership": my_gear = ["Gym", "Barbell", "Dumbbells", "Bodyweight"]
+    elif equipment == "Full Home Gym (Rack+Barbell)": my_gear = ["Barbell", "Dumbbells", "Bodyweight"]
+    elif equipment == "Home Dumbbells": my_gear = ["Dumbbells", "Bodyweight"]
     else: my_gear = ["Bodyweight"]
 
-    candidates = []
+    # 2. Filter Pool
+    pool = EXERCISE_DB.get(muscle, [])
+    valid = []
     
-    for ex in DB:
-        # 1. Muscle Match
-        if ex["muscle"] != target_muscle: continue
-        
-        # 2. Type Match (Compound vs Isolation) - Optional but good for structure
-        if target_type and ex["type"] != target_type: continue
+    for ex in pool:
+        # Equip check
+        if not any(g in my_gear for g in ex["equip"]): continue
+        # Injury check
+        if injury in ex["bad"]: continue
+        # Level check (User Tier must be >= Exercise Level)
+        if level_tier < ex["level"]: 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
-        
-        # 4. Injury Check
-        if user_injury in ex["bad_for"]: continue
-        
-        # 5. Level Check (Strict)
-        # Beginner (0) cannot do Advanced (2)
-        if user_level_score < ex["min_level"]: continue
-        
-        candidates.append(ex)
-        
-    return candidates
-
-# --- 4. ROUTINE GENERATOR ---
+        valid.append(ex)
+    
+    # 3. Shuffle for Variety
+    random.shuffle(valid)
+    
+    # 4. Return
+    if not valid:
+        return [{"name": f"Standard {muscle} Move", "ratio": 0}]
+    return valid[:count]
 
 def generate_routine(plan_name, age, gender, weight, goal, equipment, injury, experience):
     
-    # A. Parse Level
-    level_score = 0
-    if experience == "Intermediate": level_score = 1
-    if experience == "Advanced": level_score = 2
+    # --- STEP 1: RESOLVE CONFIGS ---
+    goal_settings = GOAL_CONFIG[goal]
+    level_settings = LEVEL_CONFIG[experience]
+    age_group = get_age_group(age)
+    age_settings = AGE_CONFIG[age_group]
     
-    # B. Define Structure based on AI Plan
-    # We create slots: (Muscle, Type)
+    # --- STEP 2: BUILD STRUCTURE BASED ON PLAN ---
+    # This determines WHICH muscles we hit
     slots = []
     
     if "Upper" in plan_name or "Push" in plan_name:
-        slots = [
-            ("Chest", "Compound"), ("Back", "Compound"), 
-            ("Shoulders", "Compound"), ("Chest", "Isolation"), 
-            ("Back", "Isolation"), ("Arms", "Isolation")
-        ]
+        slots = ["Chest", "Back", "Shoulders", "Chest", "Back", "Arms"]
+        routine_name = "Upper Body Focus"
     elif "Lower" in plan_name or "Legs" in plan_name:
-        slots = [
-            ("Legs", "Compound"), ("Legs", "Compound"), # Two big lifts
-            ("Legs", "Isolation"), ("Legs", "Isolation"),
-            ("Core", "Isolation")
-        ]
+        slots = ["Legs", "Legs", "Legs", "Core", "Core", "Cardio"]
+        routine_name = "Lower Body Focus"
     else: # Full Body
-        slots = [
-            ("Legs", "Compound"), ("Chest", "Compound"),
-            ("Back", "Compound"), ("Shoulders", "Compound"),
-            ("Arms", "Isolation"), ("Core", "Static")
-        ]
-
-    # 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"
-    
-    # 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 += f"GLOBAL PROTOCOL: {sets} Sets | {reps} Reps | {rest} Rest\n\n"
-    text += "EXERCISES:\n"
-    
-    used_names = [] # To prevent duplicates
-    
-    for i, (muscle, ex_type) in enumerate(slots):
-        # 1. Get Options
-        options = get_exercises(muscle, ex_type, equipment, injury, level_score)
+        slots = ["Legs", "Chest", "Back", "Shoulders", "Legs", "Core"]
+        routine_name = "Full Body Mix"
+
+    # --- STEP 3: GENERATE TEXT ---
+    text = f"WORKOUT PLAN: {routine_name}\n"
+    text += f"User: {age}y | {gender} | {weight}kg | {experience}\n"
+    text += f"Config: {goal} ({goal_settings['intensity']}) | Equip: {equipment}\n"
+    text += "="*45 + "\n\n"
+    
+    # Warmup
+    text += f"WARM-UP ({age_settings['warmup']}):\n"
+    text += f"Type: {age_settings['type']}\n"
+    if injury == "Knee": text += "- Avoid high impact jumping.\n"
+    text += "- 3 mins Light Cardio\n- Dynamic Stretching\n\n"
+    
+    # Main Block
+    text += f"MAIN WORKOUT:\n"
+    text += f"Sets: {goal_settings['sets']} | Reps: {goal_settings['reps']}\n"
+    text += f"Rest: {goal_settings['rest']}\n\n"
+    
+    used_names = []
+    
+    for i, muscle in enumerate(slots):
+        # Fetch 1 exercise, exclude used ones if possible
+        candidates = get_exercises(muscle, equipment, injury, level_settings['tier'], count=5)
         
-        # 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
+        selected = candidates[0]
+        # Try to find one not used yet
+        for cand in candidates:
+            if cand['name'] not in used_names:
+                selected = cand
+                break
         
-        used_names.append(selected["name"])
+        used_names.append(selected['name'])
         
-        # 4. Calculate Weight
-        rec_weight = calculate_weight(selected.get("base_ratio", 0), weight, gender, experience)
+        # Calc Weight
+        load = calculate_weight(selected['ratio'], weight, gender, level_settings['tier'])
         
-        text += f"{i+1}. {selected['name']}\n"
-        text += f"   Target Load: {rec_weight}\n"
-        text += f"   Type: {muscle} {selected.get('type', '')}\n\n"
+        text += f"{i+1}. {selected['name']} ({muscle})\n"
+        text += f"   Load: {load}\n\n"
         
+    # Finisher
+    text += f"FINISHER:\n"
+    text += f"Style: {goal_settings['finisher']}\n"
+    
+    if goal in ["Weight Loss", "Endurance"]:
+        cardio = get_exercises("Cardio", equipment, injury, level_settings['tier'])[0]
+        text += f"Activity: {cardio['name']} - 5 mins burnout"
+    else:
+        core = get_exercises("Core", equipment, injury, level_settings['tier'])[0]
+        text += f"Activity: {core['name']} - 3 sets to failure"
+
     return text
 
-# 5. WRAPPER
+# 6. WRAPPER
 def predict_wrapper(age, gender, weight, height, goal, equipment, injury, experience):
-    # AI Logic
+    # AI Prediction
     input_df = pd.DataFrame({
         'Age': [age], 'Gender': [gender], 'Weight_kg': [weight], 
         'Height_cm': [height], 'Goal': [goal], 'Equipment': [equipment], 
@@ -203,12 +259,15 @@ def predict_wrapper(age, gender, weight, height, goal, equipment, injury, experi
     })
     plan_name = model.predict(input_df)[0]
     
-    # Generator Logic
+    # Logic Generation
     routine = generate_routine(plan_name, age, gender, weight, goal, equipment, injury, experience)
     
-    return routine, f"AI Strategy: Selected '{plan_name}' based on injury profile."
+    info = f"AI Strategy: {plan_name}\n"
+    info += f"Adjusted for {experience} level and {goal} goal."
+    
+    return routine, info
 
-# 6. LAUNCH
+# 7. LAUNCH
 iface = gr.Interface(
     fn=predict_wrapper,
     inputs=[
@@ -216,18 +275,18 @@ iface = gr.Interface(
         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(list(GOAL_CONFIG.keys()), 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")
+        gr.Dropdown(list(LEVEL_CONFIG.keys()), label="Experience", value="Intermediate")
     ],
     outputs=[
-        gr.Textbox(label="Result", lines=20),
-        gr.Textbox(label="Note")
+        gr.Textbox(label="Result", lines=22),
+        gr.Textbox(label="System Note")
     ],
-    title="SmartFit AI - Calculator Edition",
-    description="Generates strict workouts with specific weight recommendations.",
+    title="SmartFit AI - Matrix Edition",
+    description="Fully modular workout engine controlled by strict Logic Dictionaries.",
     theme="soft"
 )
 
-iface.launch()
+iface.launch()