Update app.py

app.py

@@ -1,318 +1,197 @@
 import gradio as gr
 import joblib
 import pandas as pd
+import random
 
-# 1. Load the trained model
+# 1. Load Model
 model = joblib.load('fitness_model.joblib')
 
-# 2. COMPLETE Workout Database (Covers ALL 19 plans from the dataset)
-workout_database = {
-    # --- Injury Friendly / Rehab ---
-    "Lower Body Power & Stationary Bike": """
-    🎯 GOAL: Legs & Cardio (Shoulder Injury Safe)
-    
-    WARM-UP:
-    - 5 min Light Cycling
-    - Leg Swings (20 per leg)
-    
-    THE WORKOUT:
-    1. Leg Press (Machine): 3 sets x 12-15 reps (Moderate weight)
-    2. Goblet Squats (Hold weight at chest): 3 sets x 10 reps
-    3. Walking Lunges: 3 sets x 12 steps per leg
-    4. Seated Leg Extensions: 3 sets x 15 reps
-    5. Standing Calf Raises: 4 sets x 20 reps
-    
-    CARDIO FINISHER:
-    - Stationary Bike: 15 mins total (30 sec sprint / 30 sec slow)
-    """,
-
-    "Seated Upper Body Strength & Rowing": """
-    🎯 GOAL: Upper Body Power (Ankle/Leg Injury Safe)
-    
-    WARM-UP:
-    - 5 min Rowing Machine (Light pace)
-    - Arm Circles & Torso Twists
-    
-    THE WORKOUT (All Seated):
-    1. Seated Dumbbell Overhead Press: 3 sets x 10 reps
-    2. Seated Chest Press (Machine): 3 sets x 12 reps
-    3. Lat Pulldowns (Wide Grip): 3 sets x 12 reps
-    4. Seated Bicep Curls (Dumbbells): 3 sets x 12 reps
-    5. Seated Tricep Overhead Extension: 3 sets x 15 reps
-    
-    CORE:
-    - Seated Russian Twists (feet hooked): 3 sets x 20 reps
-    """,
-
-    "Upper Body Calisthenics & Core": """
-    🎯 GOAL: Bodyweight Strength (Knee Injury Safe)
-    
-    WARM-UP:
-    - Arm Swings, Wrist Rotations
-    
-    THE WORKOUT:
-    1. Push-ups (Classic or Knees): 3 sets x MAX reps
-    2. Pull-ups (or Bodyweight Rows under table): 3 sets x 6-10 reps
-    3. Dips (Chair or Bar): 3 sets x 10 reps
-    4. Plank Hold: 3 sets x 45-60 seconds
-    5. Side Plank: 3 sets x 30 sec per side
-    6. Supermans (Lower back): 3 sets x 15 reps
-    """,
-
-    "Core Stability, Yoga & Walking": """
-    🎯 GOAL: Active Recovery & Back Health (Back Injury Safe)
-    
-    WARM-UP:
-    - 5 min Slow Walk
-    
-    YOGA FLOW (Hold each for 30-45 sec):
-    1. Cat-Cow Stretch (Mobilize spine)
-    2. Bird-Dog (Stabilize core): 3 sets x 10 per side
-    3. Child's Pose (Rest lower back)
-    4. Glute Bridges: 3 sets x 15 reps
-    5. Cobra Stretch (Gentle extension)
-    
-    CARDIO:
-    - Walking: 30 minutes brisk pace (Outdoor or Treadmill 1% incline)
-    """,
-
-    # --- Strength & Muscle Building ---
-    "Push-Pull-Legs Split (Hypertrophy)": """
-    🎯 GOAL: Maximum Muscle Growth (Advanced)
-    
-    TODAY'S SESSION: PUSH (Chest, Shoulders, Triceps)
-    1. Barbell Bench Press: 4 sets x 8 reps (Heavy)
-    2. Incline Dumbbell Press: 3 sets x 10 reps
-    3. Overhead Military Press: 3 sets x 10 reps
-    4. Lateral Raises (Dumbbells): 4 sets x 15 reps (Burnout)
-    5. Cable Tricep Pushdowns: 3 sets x 12 reps
-    """,
-
-    "5x5 Powerlifting Program": """
-    🎯 GOAL: Pure Strength (Powerlifting Style)
-    
-    PROTOCOL: Rest 3-5 mins between sets. Lift Heavy.
-    
-    THE BIG 3:
-    1. Barbell Squat: 5 sets x 5 reps
-    2. Bench Press: 5 sets x 5 reps
-    3. Deadlift: 1 set x 5 reps (Very Heavy)
-    
-    ACCESSORY WORK:
-    - Pull-ups: 3 sets x Failure
-    - Dips: 3 sets x Failure
-    """,
-
-    "Starting Strength (3x5)": """
-    🎯 GOAL: Beginner Strength Foundation
-    
-    PROTOCOL: Focus on perfect form. Add weight each workout.
-    
-    THE WORKOUT:
-    1. Squat: 3 sets x 5 reps
-    2. Overhead Press: 3 sets x 5 reps
-    3. Deadlift: 1 set x 5 reps
-    """,
-
-    "Dumbbell Upper/Lower Split": """
-    🎯 GOAL: Balanced Physique with Dumbbells
-    
-    TODAY: UPPER BODY
-    1. Dumbbell Floor Press: 3 sets x 10 reps
-    2. Single-Arm Dumbbell Row: 3 sets x 12 per side
-    3. Dumbbell Shoulder Press: 3 sets x 10 reps
-    4. Dumbbell Shrugs: 3 sets x 15 reps
-    5. Bicep Curls vs Skullcrushers (Superset): 3 sets x 12 reps
-    """,
-
-    "3-Day Full Body Compound Lifts": """
-    🎯 GOAL: Efficiency (3 days/week)
-    
-    THE WORKOUT:
-    1. Goblet Squat: 3 sets x 10 reps
-    2. Push-ups: 3 sets x 15 reps
-    3. Dumbbell Lunges: 3 sets x 10 per leg
-    4. Dumbbell Shoulder Press: 3 sets x 10
-    5. Plank: 3 sets x 60 sec
-    """,
-
-    # --- Cardio & Athletic Performance ---
-    "Outdoor Running (5k-10k Prep)": """
-    🎯 GOAL: Endurance Running
-    
-    TODAY'S RUN: Interval Training
-    - Warm-up: 5 min Brisk Walk + Dynamic Stretching
-    - Run 1 km Easy Pace
-    - Interval: 4 rounds of [500m Fast / 2 mins Walk]
-    - Run 1 km Easy Pace to finish
-    - Cool down: Static Stretching (Calves, Hamstrings)
-    
-    TOTAL DISTANCE: Approx 4-5 km
-    """,
-
-    "Treadmill Intervals & Circuit Training": """
-    🎯 GOAL: Fat Burn & Cardio
-    
-    TREADMILL BLOCK (15 mins):
-    - 5 min Warmup walk
-    - 10 rounds of: 30 sec Sprint / 30 sec Rest (Stand on rails)
-    
-    CIRCUIT BLOCK (Repeat 3 times):
-    1. 15 Kettlebell/Dumbbell Swings
-    2. 10 Pushups
-    3. 15 Box Jumps (or Step ups)
-    4. Rest 1 min
-    """,
-
-    "Sprints & Plyometrics": """
-    🎯 GOAL: Explosive Speed & Power
-    
-    WARM-UP (Critical): 10 mins jogging, high knees, butt kicks.
-    
-    PLYOMETRICS:
-    1. Box Jumps: 3 sets x 6 reps (Max height)
-    2. Broad Jumps: 3 sets x 5 reps
-    
-    SPRINTS:
-    - 4 x 40 meter sprints (100% effort)
-    - Rest 2 mins between sprints
-    """,
-
-    "Jump Rope & Burpee Intervals": """
-    🎯 GOAL: High Intensity Cardio (No Equipment)
-    
-    FORMAT: 30 sec work / 15 sec rest. Repeat for 20 mins.
-    
-    THE LOOPS:
-    1. Jump Rope (Regular bounce)
-    2. Burpees
-    3. Jump Rope (High knees)
-    4. Mountain Climbers
-    5. Jump Rope (Double unders or fast)
-    6. Air Squats
-    (Repeat cycle)
-    """,
-
-    "Metabolic Resistance Training (Weights + Cardio)": """
-    🎯 GOAL: Burn Fat while Toning
-    
-    CIRCUIT (Do exercises back-to-back, rest 90s after full round. Do 4 Rounds):
-    1. Dumbbell Thrusters (Squat + Press): 15 reps
-    2. Renegade Rows (Plank + Row): 10 per side
-    3. Mountain Climbers: 30 reps
-    4. Kettlebell Swings: 20 reps
-    """,
-
-    # --- Functional & General Health ---
-    "Full Body Dumbbell Functional Flow": """
-    🎯 GOAL: Daily movement & health
-    
-    THE WORKOUT:
-    1. Dumbbell Deadlifts: 3 sets x 12 reps
-    2. Z-Press (Seated on floor shoulder press): 3 sets x 10
-    3. Reverse Lunges: 3 sets x 10 per leg
-    4. Farmer's Carries: 3 sets x 40 steps
-    5. Bird-Dogs: 3 sets x 12 reps
-    """,
-
-    "Daily Yoga & Bodyweight Basics": """
-    🎯 GOAL: Flexibility & Core
-    
-    FLOW SEQUENCE (Repeat 2 times):
-    1. Sun Salutation A (5 rounds)
-    2. Warrior II pose (Hold 30 sec per side)
-    3. Tree Pose (Balance): 30 sec per side
-    4. Bodyweight Squats: 15 reps
-    5. Downward Dog -> Plank flow: 10 reps
-    """,
+# 2. Smart Logic Generator
+def generate_dynamic_workout(plan_category, age, gender, weight, height, goal, equipment, injury, experience):
+    
+    # --- A. Setup Variables based on Profile ---
+    
+    # 1. Calculate suggested starting weights (Estimates)
+    strength_base = weight * 0.4 if gender == "Male" else weight * 0.25
+    if experience == "Beginner": strength_base *= 0.6
+    if experience == "Advanced": strength_base *= 1.4
+    
+    suggested_weight_db = f"{int(strength_base * 0.3)}kg" # Dumbbells
+    suggested_weight_bb = f"{int(strength_base * 0.8)}kg" # Barbell
+    
+    if equipment == "Bodyweight Only":
+        weight_text = "Bodyweight"
+    elif equipment == "Home Dumbbells":
+        weight_text = f"Dumbbells (~{suggested_weight_db})"
+    else:
+        weight_text = f"Moderate Weight (~{suggested_weight_bb} Barbell / {suggested_weight_db} DBs)"
+
+    # 2. Adjust Volume (Sets/Reps) based on Goal
+    if goal == "Muscle Gain":
+        protocol = "4 sets x 8-12 reps"
+        rest = "90 sec rest"
+        focus = "Hypertrophy (Focus on the squeeze)"
+    elif goal == "Strength":
+        protocol = "5 sets x 5 reps"
+        rest = "3 min rest"
+        focus = "Max Force (Lift heavy safely)"
+    elif goal == "Weight Loss":
+        protocol = "3 sets x 15-20 reps"
+        rest = "30-45 sec rest"
+        focus = "High Heart Rate (Keep moving)"
+    else: # General Health / Endurance
+        protocol = "3 sets x 12 reps"
+        rest = "60 sec rest"
+        focus = "Consistency & Form"
+
+    # 3. Adjust for Age
+    warmup_duration = "5 mins"
+    if age > 50:
+        warmup_duration = "10-12 mins (Joint mobilization focus)"
+        rest += " + 30 sec extra for recovery"
+
+    # --- B. Build the Routine Components ---
+    
+    # 1. Warmup
+    warmup = f"""
+    ⏱️ DURATION: {warmup_duration}
+    1. Joint Rotations (Neck, Shoulders, Hips, Ankles)
+    2. 3 min Light Cardio ({'Jumping Jacks' if injury != 'Knee' else 'March in place'})
+    3. Dynamic Stretch: {'Arm Circles' if 'Upper' in plan_category else 'Leg Swings'}
+    """
 
-    "Advanced Calisthenics Progression": """
-    🎯 GOAL: Bodyweight Mastery
-    
-    SKILL WORK:
-    1. Handstand Hold (Wall assisted): 3 sets x 30 sec
-    2. L-Sit Practice: 3 sets x 10-15 sec
-    
-    STRENGTH:
-    1. Pistol Squats (or assisted): 3 sets x 5 per leg
-    2. Archer Pushups: 3 sets x 8 per side
-    3. Wide Grip Pullups: 3 sets x 8
-    """,
+    # 2. Main Exercises Selection (Logic Tree)
+    exercises = []
+    
+    # -- IF INJURED (Safety Priority) --
+    if injury == "Knee":
+        exercises = [
+            f"1. Glute Bridges (Floor): {protocol} (Safe for knees)",
+            f"2. Seated Overhead Press: {protocol} ({weight_text})",
+            f"3. Lat Pulldowns or Pull-ups: {protocol}",
+            f"4. Plank Hold: 3 sets x 45 sec"
+        ]
+    elif injury == "Back":
+        exercises = [
+            f"1. Wall Sits: 3 sets x 45 sec (Spine neutral)",
+            f"2. Supported Single-Arm Row: {protocol} ({weight_text})",
+            f"3. Seated Chest Press/Pushups (Incline): {protocol}",
+            f"4. Bird-Dog: 3 sets x 12 reps (Stability)"
+        ]
+    elif injury == "Shoulder":
+        exercises = [
+            f"1. Leg Press or Goblet Squat (Hands low): {protocol}",
+            f"2. Lunges: {protocol} (Bodyweight)",
+            f"3. Bicep Curls: {protocol} ({weight_text})",
+            f"4. Tricep Pushdowns (Elbows tucked): {protocol}"
+        ]
+        
+    # -- IF HEALTHY (Goal & Equipment Priority) --
+    else:
+        # Leg Exercise
+        if equipment == "Bodyweight Only": leg_ex = "Air Squats"
+        elif equipment == "Home Dumbbells": leg_ex = "Dumbbell Goblet Squats"
+        else: leg_ex = "Barbell Back Squats"
+        
+        # Push Exercise
+        if equipment == "Bodyweight Only": push_ex = "Push-ups"
+        elif equipment == "Home Dumbbells": push_ex = "Dumbbell Bench Press"
+        else: push_ex = "Barbell Bench Press"
+        
+        # Pull Exercise
+        if equipment == "Bodyweight Only": pull_ex = "Inverted Rows (using table)"
+        elif equipment == "Home Dumbbells": pull_ex = "Dumbbell Rows"
+        else: pull_ex = "Barbell Bent Over Row"
+
+        exercises = [
+            f"1. {leg_ex}: {protocol} using {weight_text}",
+            f"2. {push_ex}: {protocol}",
+            f"3. {pull_ex}: {protocol}",
+            f"4. {'Shoulder Press' if goal != 'Weight Loss' else 'Burpees'}: {protocol}"
+        ]
+
+    # 3. Finisher (Core/Cardio)
+    if goal in ["Weight Loss", "Athletic Performance", "Endurance"]:
+        finisher = """
+        🔥 CARDIO FINISHER:
+        - 30 sec High Knees / Mountain Climbers
+        - 30 sec Rest
+        - Repeat 5 times
+        """
+    else: # Muscle/Strength/Health
+        finisher = """
+        🧱 CORE & STABILITY FINISHER:
+        - Plank: 3 sets x failure
+        - Russian Twists: 3 sets x 20 reps
+        """
+
+    # 4. Cooldown
+    cooldown = """
+    🧘 COOLDOWN (Don't skip!):
+    1. Hamstring Stretch (30 sec per leg)
+    2. Chest Opener (Doorway stretch)
+    3. Child's Pose (Relieve back tension)
+    """
 
-    "HIIT with Dumbbells (Tabata Style)": """
-    🎯 GOAL: Fast Fat Burn
+    # --- C. Assemble Final Text ---
+    full_text = f"""
+    📋 PERSONALIZED PLAN: {plan_category}
+    ----------------------------------------------------
+    👤 PROFILE: {age}y/o {gender} | Goal: {goal} | Exp: {experience}
+    ⚖️ REC. WEIGHTS: {weight_text}
+    🎯 FOCUS: {focus}
+    ----------------------------------------------------
     
-    TABATA PROTOCOL: 20 sec Work / 10 sec Rest. 8 Rounds per exercise.
+    🔥 WARM-UP:
+    {warmup}
     
-    EXERCISE 1: Dumbbell Goblet Squats (4 mins)
-    -- Rest 1 min --
-    EXERCISE 2: Dumbbell Push Press (4 mins)
-    -- Rest 1 min --
-    EXERCISE 3: Russian Twists (4 mins)
-    """,
-
-    "CrossFit Style WODs": """
-    🎯 GOAL: Functional Fitness
+    💪 MAIN WORKOUT (Rest: {rest}):
+    {chr(10).join(exercises)}
     
-    WOD (Workout of the Day): "The Chipper"
-    For Time (Complete as fast as possible):
-    - 50 Air Squats
-    - 40 Sit-ups
-    - 30 Dumbbell Snatches (alternate arms)
-    - 20 Push-ups
-    - 10 Burpees
+    {finisher}
     
-    Time Cap: 15 mins. Go!
+    {cooldown}
     """
-}
-
-default_plan_text = "Please select a plan or adjust inputs."
+    
+    return full_text
 
-# 3. Prediction Function
+# 3. Prediction Wrapper
 def predict_plan(age, gender, weight, height, goal, equipment, injury, experience):
-    # Prepare Input
+    # Predict Category
     input_data = pd.DataFrame({
         'Age': [age], 'Gender': [gender], 'Weight_kg': [weight], 
         'Height_cm': [height], 'Goal': [goal], 'Equipment': [equipment], 
         'Injury': [injury], 'Experience': [experience]
     })
-    
-    # Predict Label
     plan_name = model.predict(input_data)[0]
     
-    # Get Detailed Text (Lookup in dictionary)
-    full_routine = workout_database.get(plan_name, f"Routine for {plan_name} coming soon!")
+    # Generate Unique Text
+    routine = generate_dynamic_workout(plan_name, age, gender, weight, height, goal, equipment, injury, experience)
     
     # Generate Tip
-    tip = f"Plan Selected: {plan_name}\n"
-    if injury != 'None':
-        tip += f"⚠️ SAFETY MODE: Adjusted for {injury} injury. Avoid heavy impact."
-    elif goal == 'Muscle Gain':
-        tip += "💪 TIP: Eat 20g protein post-workout."
-    elif goal == 'Weight Loss':
-        tip += "🔥 TIP: Keep heart rate up during rests."
-        
-    return full_routine, tip
+    tip = f"AI Insight: We selected '{plan_name}' based on your safety needs, but customized the intensity for {goal}."
+    
+    return routine, tip
 
 # 4. Interface
 iface = gr.Interface(
     fn=predict_plan,
     inputs=[
-        gr.Slider(18, 80, value=30, label="Age"),
+        gr.Slider(18, 80, 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', 'Athletic Performance'], 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="Your Personalized Routine", lines=15),
-        gr.Textbox(label="AI Coach Notes")
+        gr.Textbox(label="Your Custom Routine", lines=22),
+        gr.Textbox(label="AI Coach Note")
     ],
-    title="SmartFit AI 🏋️‍♂️",
-    description="Get a safe, goal-oriented workout plan generated by AI.",
+    title="SmartFit AI Ultimate 🚀",
+    description="A fully dynamic workout generator that calculates weights, sets, and exercises based on your exact profile.",
     theme="soft"
 )