Update app.py

app.py

@@ -7,251 +7,204 @@ import random
 model = joblib.load('fitness_model.joblib')
 
 # ==========================================
-#        CONFIG DICTIONARIES (THE BRAIN)
+#        CONFIG DICTIONARIES
 # ==========================================
 
-# 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"
-    }
+    "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"}
 }
 
-# 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)"}
+    "Beginner": {"tier": 0},
+    "Intermediate": {"tier": 1},
+    "Advanced": {"tier": 2}
 }
 
-# 4. EXERCISE DATABASE (Categorized)
-# Level 0 = Easy/Safe, Level 1 = Medium, Level 2 = Hard
+# EXERCISE DATABASE with Base Strength Ratios
+# Ratio 1.0 means an intermediate male can lift 100% of his bodyweight in this exercise.
 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": "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": "Cable Flys", "equip": ["Gym"], "bad": [], "level": 1, "ratio": 0.2},
     ],
     "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": "Chest Supported Row", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.5},
+        {"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": "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": "Chest Supported Row", "equip": ["Gym"], "bad": [], "level": 0, "ratio": 0.5},
         {"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": "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": "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": "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},
     ],
     "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": "Lateral Raises", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.1},
+        {"name": "Lateral Raises", "equip": ["Dumbbells", "Gym"], "bad": [], "level": 0, "ratio": 0.08},
         {"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},
+        {"name": "Front Raises", "equip": ["Dumbbells"], "bad": [], "level": 0, "ratio": 0.08},
     ],
     "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": "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": "Bench Dips", "equip": ["Bodyweight"], "bad": ["Shoulder"], "level": 0, "ratio": 0},
+        {"name": "Dips", "equip": ["Bodyweight"], "bad": ["Shoulder"], "level": 1, "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},
+        {"name": "Russian Twists", "equip": ["Bodyweight"], "bad": ["Back"], "level": 0, "ratio": 0},
+        {"name": "Hanging Leg Raises", "equip": ["Gym"], "bad": ["Shoulder"], "level": 2, "ratio": 0},
     ],
     "Cardio": [
-        {"name": "Walking", "equip": ["Bodyweight", "Gym"], "bad": [], "level": 0, "ratio": 0},
+        {"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": "Sprints", "equip": ["Bodyweight"], "bad": ["Knee"], "level": 1, "ratio": 0},
+        {"name": "Jump Rope", "equip": ["Bodyweight"], "bad": ["Knee", "Ankle"], "level": 1, "ratio": 0},
     ]
 }
 
 # ==========================================
-#           LOGIC ENGINE
+#           LOGIC ENGINE (CALCULATOR)
 # ==========================================
 
-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):
+def calculate_weight(base_ratio, user_weight, gender, level_tier, muscle_group):
+    """
+    Calculates precise weight based on Gender Physiology.
+    """
     if base_ratio == 0: return "Bodyweight"
     
     load = user_weight * base_ratio
     
-    # Adjustments
-    if gender == "Female": load *= 0.6
-    if level_tier == 0: load *= 0.6  # Beginner
-    if level_tier == 2: load *= 1.3  # Advanced
-    
-    # Rounding
-    load = 2.5 * round(load / 2.5)
-    return f"{int(load)} kg"
+    # 1. Gender Adjustment (The Science)
+    if gender == "Female":
+        # Women are relatively stronger in lower body than upper body
+        if muscle_group in ["Legs", "Core"]:
+            load *= 0.75  # Stronger legs
+        else:
+            load *= 0.55  # Weaker upper body push/pull relative to men
+    
+    # 2. Level Adjustment
+    if level_tier == 0: load *= 0.6   # Beginner
+    if level_tier == 2: load *= 1.3   # Advanced
+    
+    # 3. Smart Rounding
+    # Round to nearest 2.5kg (plate size) or 1kg (dumbbell size)
+    if load < 10:
+        final_load = round(load)
+    else:
+        final_load = 2.5 * round(load / 2.5)
+        
+    return f"{int(final_load)} kg"
 
 def get_exercises(muscle, equipment, injury, level_tier, count=1):
-    # 1. Define Equipment List
+    # Filter available gear
     my_gear = []
     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"]
 
-    # 2. Filter Pool
     pool = EXERCISE_DB.get(muscle, [])
     valid = []
     
     for ex in pool:
-        # Equip check
+        # Filter Logic
         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
         
         valid.append(ex)
     
-    # 3. Shuffle for Variety
+    # SHUFFLE to ensure variety every time
     random.shuffle(valid)
     
-    # 4. Return
-    if not valid:
-        return [{"name": f"Standard {muscle} Move", "ratio": 0}]
+    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):
     
-    # --- STEP 1: RESOLVE CONFIGS ---
+    # Configs
     goal_settings = GOAL_CONFIG[goal]
-    level_settings = LEVEL_CONFIG[experience]
-    age_group = get_age_group(age)
-    age_settings = AGE_CONFIG[age_group]
+    level_tier = LEVEL_CONFIG[experience]["tier"]
     
-    # --- STEP 2: BUILD STRUCTURE BASED ON PLAN ---
-    # This determines WHICH muscles we hit
+    # Structure based on Plan Category
     slots = []
-    
     if "Upper" in plan_name or "Push" in plan_name:
-        slots = ["Chest", "Back", "Shoulders", "Chest", "Back", "Arms"]
-        routine_name = "Upper Body Focus"
+        slots = ["Chest", "Back", "Shoulders", "Arms", "Core"]
+        title = "Upper Body Focus"
     elif "Lower" in plan_name or "Legs" in plan_name:
-        slots = ["Legs", "Legs", "Legs", "Core", "Core", "Cardio"]
-        routine_name = "Lower Body Focus"
-    else: # Full Body
-        slots = ["Legs", "Chest", "Back", "Shoulders", "Legs", "Core"]
-        routine_name = "Full Body Mix"
+        slots = ["Legs", "Legs", "Legs", "Core", "Cardio"]
+        title = "Lower Body Focus"
+    else:
+        slots = ["Legs", "Chest", "Back", "Shoulders", "Core"]
+        title = "Full Body Mix"
 
-    # --- STEP 3: GENERATE TEXT ---
-    text = f"WORKOUT PLAN: {routine_name}\n"
+    # Build Text
+    text = f"WORKOUT PLAN: {title}\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"
+    text += f"Goal: {goal} | Equip: {equipment}\n"
+    text += "="*40 + "\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"
+    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"
     
-    # 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"
+    text += f"Sets: {goal_settings['sets']} | Reps: {goal_settings['reps']} | 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)
+        # Fetch exercise
+        candidates = get_exercises(muscle, equipment, injury, level_tier, count=3)
         
+        # Pick unique
         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'])
         
-        # Calc Weight
-        load = calculate_weight(selected['ratio'], weight, gender, level_settings['tier'])
+        # Calculate Load
+        load = calculate_weight(selected['ratio'], weight, gender, level_tier, muscle)
         
         text += f"{i+1}. {selected['name']} ({muscle})\n"
-        text += f"   Load: {load}\n\n"
+        text += f"   Rec. 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"
+        text += "HIIT: 30s Work / 30s Rest (5 Rounds)"
     else:
-        core = get_exercises("Core", equipment, injury, level_settings['tier'])[0]
-        text += f"Activity: {core['name']} - 3 sets to failure"
+        text += "Core Stability: 3 Sets to Failure"
 
     return text
 
 # 6. WRAPPER
 def predict_wrapper(age, gender, weight, height, goal, equipment, injury, experience):
-    # AI Prediction
     input_df = pd.DataFrame({
         'Age': [age], 'Gender': [gender], 'Weight_kg': [weight], 
         'Height_cm': [height], 'Goal': [goal], 'Equipment': [equipment], 
@@ -259,13 +212,9 @@ def predict_wrapper(age, gender, weight, height, goal, equipment, injury, experi
     })
     plan_name = model.predict(input_df)[0]
     
-    # Logic Generation
     routine = generate_routine(plan_name, age, gender, weight, goal, equipment, injury, experience)
     
-    info = f"AI Strategy: {plan_name}\n"
-    info += f"Adjusted for {experience} level and {goal} goal."
-    
-    return routine, info
+    return routine, f"AI Strategy: {plan_name}"
 
 # 7. LAUNCH
 iface = gr.Interface(
@@ -281,11 +230,11 @@ iface = gr.Interface(
         gr.Dropdown(list(LEVEL_CONFIG.keys()), label="Experience", value="Intermediate")
     ],
     outputs=[
-        gr.Textbox(label="Result", lines=22),
-        gr.Textbox(label="System Note")
+        gr.Textbox(label="Generated Workout", lines=20),
+        gr.Textbox(label="AI Note")
     ],
-    title="SmartFit AI - Matrix Edition",
-    description="Fully modular workout engine controlled by strict Logic Dictionaries.",
+    title="SmartFit AI - Biological Engine",
+    description="Advanced generator with gender-specific strength standards.",
     theme="soft"
 )