Update prompt_builder.py

prompt_builder.py

@@ -2,6 +2,7 @@ def calculate_bmi(weight, height):
     height_m = height / 100
     return weight / (height_m ** 2)
 
+
 def bmi_category(bmi):
     if bmi < 18.5:
         return "Underweight"
@@ -22,26 +23,71 @@ def build_prompt(name, age, gender, height, weight, goal, fitness_level, equipme
 
     prompt = f"""
 You are a certified professional fitness trainer.
-Create a structured 5-day personalized workout plan.
-
-User Profile:
-- Name: {name}
-- Age: {age}
-- Gender: {gender}
-- Height: {height} cm
-- Weight: {weight} kg
-- BMI: {bmi:.2f} ({bmi_status})
-- Goal: {goal}
-- Fitness Level: {fitness_level}
-- Available Equipment: {equipment_list}
-
-Instructions:
-1. Divide clearly into Day 1 to Day 5.
-2. Include exercise name.
-3. Include sets and reps.
-4. Include rest period.
-5. Adjust intensity based on BMI category AND age.
-6. Avoid unsafe exercises for beginners or older users.
-7. Keep the plan professional and easy to follow.
+
+Create a 5-day personalized workout plan for the user.
+
+USER PROFILE
+Name: {name}
+Age: {age}
+Gender: {gender}
+Height: {height} cm
+Weight: {weight} kg
+BMI: {bmi:.2f} ({bmi_status})
+Goal: {goal}
+Fitness Level: {fitness_level}
+Available Equipment: {equipment_list}
+
+INSTRUCTIONS
+
+1. Create a workout plan from Day 1 to Day 5.
+2. Each day must contain 3–4 exercises.
+3. Each exercise must include:
+   - Exercise Name
+   - Sets
+   - Reps
+   - Rest Time
+4. Adjust intensity based on BMI, age, and fitness level.
+5. Avoid unsafe exercises for beginners.
+
+OUTPUT FORMAT (VERY IMPORTANT)
+
+Write the workout like step-by-step instructions.
+
+Example:
+
+Day 1 – Upper Body
+
+1. Push-Ups  
+Sets: 3  
+Reps: 10-12  
+Rest: 60 seconds  
+
+2. Dumbbell Shoulder Press  
+Sets: 3  
+Reps: 8-10  
+Rest: 60 seconds  
+
+3. Bent Over Rows  
+Sets: 3  
+Reps: 10  
+Rest: 45 seconds  
+
+Day 2 – Lower Body
+
+1. Bodyweight Squats  
+Sets: 3  
+Reps: 12  
+Rest: 60 seconds  
+
+2. Lunges  
+Sets: 3  
+Reps: 10 each leg  
+Rest: 45 seconds  
+
+Continue until Day 5.
+
+Do not include explanations.
+Only provide the workout instructions.
 """
+
     return prompt, bmi, bmi_status