Update app.py

app.py

@@ -2,22 +2,38 @@ import gradio as gr
 import joblib
 import pandas as pd
 import random
-
-# 1. Load Model
-# loading the specific file name you uploaded
-model = joblib.load('fitness_model (1).joblib')
-
-# ==========================================
-#        CONFIG DICTIONARIES
-# ==========================================
-
+import os
+
+# 1. Load Model with the CORRECT filename
+# We check if the file exists just to be safe
+model_filename = 'fitness_model (2).joblib'
+
+if os.path.exists(model_filename):
+    model = joblib.load(model_filename)
+    print(f"Successfully loaded: {model_filename}")
+else:
+    # Fallback in case the name changed during upload
+    try:
+        model = joblib.load('fitness_model.joblib')
+        print("Loaded standard 'fitness_model.joblib' instead.")
+    except:
+        raise FileNotFoundError(f"Could not find {model_filename} or fitness_model.joblib")
+
+#        CONFIG DICTIONARIES (UPDATED)
+# Added new goals from the dataset
 GOAL_CONFIG = {
     "Strength": {"sets": 5, "reps": "3-5", "rest": "3 min", "intensity": "Heavy"},
     "Muscle Gain": {"sets": 3, "reps": "8-12", "rest": "90 sec", "intensity": "Moderate"},
     "Weight Loss": {"sets": 4, "reps": "15-20", "rest": "45 sec", "intensity": "High Tempo"},
     "Endurance": {"sets": 2, "reps": "20-25", "rest": "30 sec", "intensity": "Light"},
     "General Health": {"sets": 3, "reps": "12-15", "rest": "60 sec", "intensity": "Moderate"},
-    "Athletic Performance": {"sets": 4, "reps": "6-10", "rest": "2 min", "intensity": "Explosive"}
+    "Athletic Performance": {"sets": 4, "reps": "6-10", "rest": "2 min", "intensity": "Explosive"},
+    # NEW GOALS
+    "Flexibility": {"sets": 2, "reps": "Hold 30s", "rest": "None", "intensity": "Low Impact"},
+    "Tone Up": {"sets": 3, "reps": "15-20", "rest": "45 sec", "intensity": "Burnout"},
+    "Powerlifting": {"sets": 5, "reps": "1-3", "rest": "3-5 min", "intensity": "Max Effort"},
+    "CrossFit": {"sets": "AMRAP", "reps": "High", "rest": "Minimal", "intensity": "Very High"},
+    "Marathon Training": {"sets": "Distance", "reps": "Km", "rest": "Pacing", "intensity": "Steady State"}
 }
 
 LEVEL_CONFIG = {
@@ -26,92 +42,65 @@ LEVEL_CONFIG = {
     "Advanced": {"tier": 2}
 }
 
+# Expanded Exercise Database
 EXERCISE_DB = {
     "Chest": [
         {"name": "Barbell Bench Press", "equip": ["Barbell", "Gym"], "bad": ["Shoulder"], "level": 1, "ratio": 0.9},
         {"name": "Dumbbell Chest Press", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.35},
-        {"name": "Incline Dumbbell Press", "equip": ["Dumbbells", "Gym"], "bad": ["Shoulder"], "level": 1, "ratio": 0.3},
-        {"name": "Machine Chest Press", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.6},
         {"name": "Push-ups", "equip": ["Bodyweight"], "bad": ["Wrist"], "level": 0, "ratio": 0},
-        {"name": "Cable Flys", "equip": ["Gym"], "bad": [], "level": 1, "ratio": 0.15},
-        {"name": "Dumbbell Flys", "equip": ["Dumbbells"], "bad": ["Shoulder"], "level": 0, "ratio": 0.12},
-        {"name": "Weighted Dips", "equip": ["Gym"], "bad": ["Shoulder"], "level": 2, "ratio": 0},
+        {"name": "Burpees", "equip": ["Bodyweight"], "bad": ["Knee"], "level": 1, "ratio": 0},
     ],
     "Back": [
         {"name": "Deadlift", "equip": ["Barbell", "Gym"], "bad": ["Back"], "level": 2, "ratio": 1.4},
         {"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": "Barbell Bent Over Row", "equip": ["Barbell", "Gym"], "bad": ["Back"], "level": 1, "ratio": 0.7},
-        {"name": "Single Arm Dumbbell Row", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.3},
+        {"name": "Dumbbell Row", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.3},
         {"name": "Pull-ups", "equip": ["Bodyweight", "Gym"], "bad": ["Shoulder"], "level": 1, "ratio": 0},
-        {"name": "Chest Supported Row", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.5},
-        {"name": "Inverted Row", "equip": ["Bodyweight"], "bad": [], "level": 0, "ratio": 0},
     ],
     "Legs": [
         {"name": "Barbell Squat", "equip": ["Barbell", "Gym"], "bad": ["Back", "Knee"], "level": 2, "ratio": 1.1},
         {"name": "Leg Press", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 1.8},
         {"name": "Goblet Squat", "equip": ["Dumbbells", "Gym"], "bad": ["Knee"], "level": 0, "ratio": 0.4},
-        {"name": "Walking Lunges", "equip": ["Dumbbells", "Bodyweight"], "bad": ["Knee"], "level": 1, "ratio": 0.15},
-        {"name": "Romanian Deadlift", "equip": ["Barbell", "Dumbbells"], "bad": ["Back"], "level": 1, "ratio": 0.8},
-        {"name": "Lying Leg Curl", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.35},
-        {"name": "Leg Extensions", "equip": ["Gym"], "bad": ["Knee"], "level": 0, "ratio": 0.35},
-        {"name": "Bulgarian Split Squat", "equip": ["Dumbbells"], "bad": ["Knee"], "level": 2, "ratio": 0.2},
-        {"name": "Glute Bridges", "equip": ["Bodyweight"], "bad": [], "level": 0, "ratio": 0},
+        {"name": "Box Jumps", "equip": ["Bodyweight", "Gym"], "bad": ["Knee", "Ankle"], "level": 1, "ratio": 0},
+        {"name": "Lunges", "equip": ["Dumbbells", "Bodyweight"], "bad": ["Knee"], "level": 0, "ratio": 0.2},
     ],
     "Shoulders": [
-        {"name": "Overhead Barbell Press", "equip": ["Barbell", "Gym"], "bad": ["Shoulder", "Back"], "level": 2, "ratio": 0.5},
-        {"name": "Seated Dumbbell Press", "equip": ["Dumbbells", "Gym"], "bad": ["Shoulder"], "level": 0, "ratio": 0.25},
+        {"name": "Overhead Press", "equip": ["Barbell", "Gym"], "bad": ["Shoulder", "Back"], "level": 2, "ratio": 0.5},
         {"name": "Lateral Raises", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.08},
-        {"name": "Face Pulls", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.2},
-        {"name": "Front Raises", "equip": ["Dumbbells"], "bad": [], "level": 0, "ratio": 0.08},
-    ],
-    "Arms": [
-        {"name": "Barbell Curls", "equip": ["Barbell", "Gym"], "bad": ["Wrist"], "level": 0, "ratio": 0.25},
-        {"name": "Dumbbell Hammer Curls", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.12},
-        {"name": "Tricep Rope Pushdown", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.25},
-        {"name": "Skullcrushers", "equip": ["Barbell", "Dumbbells"], "bad": ["Elbow"], "level": 1, "ratio": 0.2},
-        {"name": "Dips", "equip": ["Bodyweight"], "bad": ["Shoulder"], "level": 1, "ratio": 0},
+        {"name": "Handstand Pushups", "equip": ["Bodyweight"], "bad": ["Shoulder", "Wrist"], "level": 2, "ratio": 0},
     ],
     "Core": [
         {"name": "Plank", "equip": ["Bodyweight"], "bad": [], "level": 0, "ratio": 0},
-        {"name": "Cable Woodchoppers", "equip": ["Gym"], "bad": ["Back"], "level": 1, "ratio": 0.3},
-        {"name": "Russian Twists", "equip": ["Bodyweight"], "bad": ["Back"], "level": 0, "ratio": 0},
         {"name": "Hanging Leg Raises", "equip": ["Gym"], "bad": ["Shoulder"], "level": 2, "ratio": 0},
+        {"name": "Crunches", "equip": ["Bodyweight"], "bad": ["Back"], "level": 0, "ratio": 0},
+    ],
+    "Mobility": [
+        {"name": "Downward Dog", "equip": ["Bodyweight", "Yoga Mat"], "bad": ["Wrist"], "level": 0, "ratio": 0},
+        {"name": "Cat-Cow Stretch", "equip": ["Bodyweight"], "bad": [], "level": 0, "ratio": 0},
+        {"name": "Pigeon Pose", "equip": ["Bodyweight"], "bad": ["Knee"], "level": 1, "ratio": 0},
     ],
     "Cardio": [
-        {"name": "Treadmill Run", "equip": ["Gym"], "bad": ["Knee"], "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": "Jump Rope", "equip": ["Bodyweight"], "bad": ["Knee", "Ankle"], "level": 1, "ratio": 0},
+        {"name": "5km Run", "equip": ["Bodyweight", "Gym"], "bad": ["Knee"], "level": 1, "ratio": 0},
+        {"name": "Rowing Machine", "equip": ["Gym"], "bad": ["Back"], "level": 0, "ratio": 0},
+        {"name": "Cycling", "equip": ["Gym", "Stationary"], "bad": [], "level": 0, "ratio": 0},
     ]
 }
 
-# ==========================================
 #            LOGIC ENGINE
-# ==========================================
-
 def calculate_weight(base_ratio, user_weight, gender, level_tier, muscle_group):
-    if base_ratio == 0: return "Bodyweight"
-    
+    if base_ratio == 0: return "Bodyweight / NA"
+
     load = user_weight * base_ratio
-    
+
     # Gender Logic
     if gender == "Female":
-        if muscle_group in ["Legs", "Core"]:
-            load *= 0.75
-        else:
-            load *= 0.55
-    
+        if muscle_group in ["Legs", "Core"]: load *= 0.75
+        else: load *= 0.55
+
     # Level Logic
     if level_tier == 0: load *= 0.6
     if level_tier == 2: load *= 1.3
-    
-    # Rounding
-    if load < 10:
-        final_load = round(load)
-    else:
-        final_load = 2.5 * round(load / 2.5)
-        
+
+    final_load = round(load / 2.5) * 2.5 if load > 10 else round(load)
     return f"{int(final_load)} kg"
 
 def get_exercises(muscle, equipment, injury, level_tier, count=1):
@@ -119,95 +108,105 @@ def get_exercises(muscle, equipment, injury, level_tier, count=1):
     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"]
+    else: my_gear = ["Bodyweight", "Yoga Mat"]
 
     pool = EXERCISE_DB.get(muscle, [])
     valid = []
-    
+
     for ex in pool:
+        # Check gear
         if not any(g in my_gear for g in ex["equip"]): continue
+        # Check injury
         if injury in ex["bad"]: continue
+        # Check level
         if level_tier < ex["level"]: continue
         valid.append(ex)
-    
+
     random.shuffle(valid)
-    
-    if not valid: return [{"name": f"Standard {muscle} Move", "ratio": 0}]
+    if not valid: return [{"name": f"Generic {muscle} Move", "ratio": 0}]
     return valid[:count]
 
 def generate_routine(plan_name, age, gender, weight, goal, equipment, injury, experience):
-    
+
     goal_settings = GOAL_CONFIG.get(goal, GOAL_CONFIG["General Health"])
     level_tier = LEVEL_CONFIG.get(experience, {"tier": 1})["tier"]
-    
-    slots = []
-    if "Upper" in plan_name or "Push" in plan_name:
-        slots = ["Chest", "Back", "Shoulders", "Arms", "Core"]
-        title = "Upper Body Focus"
+
+    # PLAN LOGIC SELECTOR 
+    # We map the AI's predicted plan name to a structure of muscles to work
+    
+    if "Yoga" in plan_name:
+        slots = ["Mobility", "Mobility", "Core", "Mobility", "Core"]
+        title = "Yoga & Mobility Flow"
+    elif "Running" in plan_name or "Marathon" in plan_name:
+        slots = ["Cardio", "Legs", "Core", "Mobility"]
+        title = "Endurance Run & Core"
+    elif "CrossFit" in plan_name or "WOD" in plan_name:
+        slots = ["Legs", "Chest", "Cardio", "Shoulders", "Core"]
+        title = "Metabolic Conditioning (WOD)"
+    elif "Upper" in plan_name or "Push" in plan_name:
+        slots = ["Chest", "Back", "Shoulders", "Core"]
+        title = "Upper Body Strength"
     elif "Lower" in plan_name or "Legs" in plan_name:
-        slots = ["Legs", "Legs", "Legs", "Core", "Cardio"]
-        title = "Lower Body Focus"
+        slots = ["Legs", "Legs", "Core", "Cardio"]
+        title = "Lower Body Power"
     else:
+        # Default Full Body
         slots = ["Legs", "Chest", "Back", "Shoulders", "Core"]
         title = "Full Body Mix"
 
-    text = f"WORKOUT PLAN: {title}\n"
+    # --- TEXT GENERATION ---
+    text = f"AI PLAN: {title}\n"
+    text += f"Strategy: {plan_name}\n"
     text += f"User: {age}y | {gender} | {weight}kg | {experience}\n"
-    text += f"Goal: {goal} | Equip: {equipment}\n"
     text += "="*40 + "\n\n"
-    
-    text += f"WARM-UP (5-8 mins):\n"
-    if injury == "Knee": text += "- Arm Circles & Torso Twists\n- Glute Bridges\n\n"
-    else: text += "- Jumping Jacks\n- Dynamic Stretching\n\n"
-    
-    text += f"MAIN WORKOUT:\n"
-    text += f"Sets: {goal_settings['sets']} | Reps: {goal_settings['reps']} | Rest: {goal_settings['rest']}\n\n"
-    
+
+    text += f"PROTOCOL:\n"
+    text += f"Sets: {goal_settings['sets']} | Reps: {goal_settings['reps']}\n"
+    text += f"Intensity: {goal_settings['intensity']} | Rest: {goal_settings['rest']}\n\n"
+
+    text += f"WORKOUT:\n"
     used_names = []
-    
+
     for i, muscle in enumerate(slots):
         candidates = get_exercises(muscle, equipment, injury, level_tier, count=3)
         selected = candidates[0]
+        # Avoid duplicates
         for cand in candidates:
             if cand['name'] not in used_names:
                 selected = cand
                 break
         used_names.append(selected['name'])
-        
+
         load = calculate_weight(selected['ratio'], weight, gender, level_tier, muscle)
         
         text += f"{i+1}. {selected['name']} ({muscle})\n"
-        text += f"   Rec. Load: {load}\n\n"
-        
-    text += f"FINISHER:\n"
-    if goal in ["Weight Loss", "Endurance"]:
-        text += "HIIT: 30s Work / 30s Rest (5 Rounds)"
-    else:
-        text += "Core Stability: 3 Sets to Failure"
+        if load != "Bodyweight / NA":
+            text += f"   Target Load: {load}\n"
+        text += "\n"
 
     return text
 
-# ==========================================
 #            MAIN PREDICTOR
-# ==========================================
 
 def predict_wrapper(age, gender, weight, height, goal, equipment, injury, experience):
     input_df = pd.DataFrame({
-        'Age': [age], 'Gender': [gender], 'Weight_kg': [weight], 
-        'Height_cm': [height], 'Goal': [goal], 'Equipment': [equipment], 
+        'Age': [age], 'Gender': [gender], 'Weight_kg': [weight],
+        'Height_cm': [height], 'Goal': [goal], 'Equipment': [equipment],
         'Injury': [injury], 'Experience': [experience]
     })
-    
-    # Use the trained Random Forest model
+
+    # Predict using the NEW model (trained on 10k rows)
     plan_name = model.predict(input_df)[0]
     
+    # Generate the readable text routine
     routine = generate_routine(plan_name, age, gender, weight, goal, equipment, injury, experience)
     
-    return routine, f"AI Strategy: {plan_name}"
+    return routine
 
-# ==========================================
 #            UI LAUNCH (GRADIO)
-# ==========================================
+
+goal_options = list(GOAL_CONFIG.keys()) # Auto-update from config
+equip_options = ['Gym Membership', 'Home Dumbbells', 'Bodyweight Only', 'Full Home Gym (Rack+Barbell)', 'Yoga Mat & Bands']
 
 iface = gr.Interface(
     fn=predict_wrapper,
@@ -216,27 +215,16 @@ 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(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(goal_options, label="Goal", value="Muscle Gain"),
+        gr.Dropdown(equip_options, label="Equipment", value="Gym Membership"),
         gr.Dropdown(['None', 'Knee', 'Back', 'Shoulder', 'Ankle'], label="Injury", value="None"),
         gr.Dropdown(list(LEVEL_CONFIG.keys()), label="Experience", value="Intermediate")
     ],
     outputs=[
-        gr.Textbox(label="Generated Workout", lines=20),
-        gr.Textbox(label="AI Note")
-    ],
-    title="SmartFit AI - Biological Engine",
-    description="Advanced generator with gender-specific strength standards.",
-    examples=[
-        # Example 1: The Bodybuilder (Young Male, Gym, No Injury)
-        [24, "Male", 80, 180, "Muscle Gain", "Gym Membership", "None", "Advanced"],
-        
-        # Example 2: The Rehab/Home User (Female, Knee Injury, Home)
-        [35, "Female", 65, 165, "Weight Loss", "Home Dumbbells", "Knee", "Beginner"],
-        
-        # Example 3: The Senior Health (Older Male, Bodyweight, Back safe)
-        [60, "Male", 85, 175, "General Health", "Bodyweight Only", "Back", "Beginner"]
+        gr.Textbox(label="AI Personalized Workout", lines=18)
     ],
+    title="SmartFit AI 2.0 - Hyper-Personalized Engine",
+    description="Now supporting CrossFit, Yoga, Marathon Training, and Rehab protocols.",
     theme="soft"
 )