Update app.py

app.py

@@ -5,40 +5,296 @@ import pandas as pd
 # 1. Load the trained model
 model = joblib.load('fitness_model.joblib')
 
-# 2. Define the prediction function
+# 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
+    """,
+
+    "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
+    """,
+
+    "HIIT with Dumbbells (Tabata Style)": """
+    🎯 GOAL: Fast Fat Burn
+    
+    TABATA PROTOCOL: 20 sec Work / 10 sec Rest. 8 Rounds per exercise.
+    
+    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
+    
+    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
+    
+    Time Cap: 15 mins. Go!
+    """
+}
+
+default_plan_text = "Please select a plan or adjust inputs."
+
+# 3. Prediction Function
 def predict_plan(age, gender, weight, height, goal, equipment, injury, experience):
-    # Create DataFrame with exact column names as trained
+    # Prepare Input
     input_data = pd.DataFrame({
-        'Age': [age],
-        'Gender': [gender],
-        'Weight_kg': [weight],
-        'Height_cm': [height],
-        'Goal': [goal],
-        'Equipment': [equipment],
-        'Injury': [injury],
-        'Experience': [experience]
+        'Age': [age], 'Gender': [gender], 'Weight_kg': [weight], 
+        'Height_cm': [height], 'Goal': [goal], 'Equipment': [equipment], 
+        'Injury': [injury], 'Experience': [experience]
     })
     
-    # Predict
-    prediction = model.predict(input_data)[0]
+    # Predict Label
+    plan_name = model.predict(input_data)[0]
     
-    # Generate Smart Tip based on logic
-    tip = f"Goal: {goal}. "
+    # Get Detailed Text (Lookup in dictionary)
+    full_routine = workout_database.get(plan_name, f"Routine for {plan_name} coming soon!")
     
+    # Generate Tip
+    tip = f"Plan Selected: {plan_name}\n"
     if injury != 'None':
-        tip += f"Note: Because of your {injury} injury, this plan avoids high impact stress on that area."
+        tip += f"⚠️ SAFETY MODE: Adjusted for {injury} injury. Avoid heavy impact."
     elif goal == 'Muscle Gain':
-        tip += "Focus on progressive overload and sufficient protein intake."
+        tip += "💪 TIP: Eat 20g protein post-workout."
     elif goal == 'Weight Loss':
-        tip += "Combine this workout with a slight caloric deficit for best results."
-    elif goal == 'Strength':
-        tip += "Rest periods of 3-5 minutes between sets are recommended for strength."
-    else:
-        tip += "Stay consistent and listen to your body!"
+        tip += "🔥 TIP: Keep heart rate up during rests."
         
-    return prediction, tip
+    return full_routine, tip
 
-# 3. Create the Gradio Interface
+# 4. Interface
 iface = gr.Interface(
     fn=predict_plan,
     inputs=[
@@ -49,16 +305,15 @@ iface = gr.Interface(
         gr.Dropdown(['Weight Loss', 'Muscle Gain', 'Endurance', 'General Health', 'Strength', 'Athletic Performance'], label="Goal", value="General Health"),
         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="Beginner")
+        gr.Radio(['Beginner', 'Intermediate', 'Advanced'], label="Experience", value="Intermediate")
     ],
     outputs=[
-        gr.Textbox(label="Recommended Workout Plan", type="text"),
-        gr.Textbox(label="AI Coach Tip", type="text")
+        gr.Textbox(label="Your Personalized Routine", lines=15),
+        gr.Textbox(label="AI Coach Notes")
     ],
-    title="SmartFit AI - Workout Recommender",
-    description="Enter your profile details to get a machine-learning optimized workout plan based on 10,000 synthetic profiles.",
+    title="SmartFit AI 🏋️‍♂️",
+    description="Get a safe, goal-oriented workout plan generated by AI.",
     theme="soft"
 )
 
-# 4. Launch App
 iface.launch()