Update prompt_builder.py

prompt_builder.py

@@ -13,7 +13,7 @@ def bmi_category(bmi):
         return "Obese"
 
 
-def build_prompt(name, gender, height, weight, goal, fitness_level, equipment):
+def build_prompt(name, age, gender, height, weight, goal, fitness_level, equipment):
 
     bmi = calculate_bmi(weight, height)
     bmi_status = bmi_category(bmi)
@@ -22,11 +22,11 @@ def build_prompt(name, gender, height, weight, goal, fitness_level, equipment):
 
     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
@@ -40,9 +40,8 @@ Instructions:
 2. Include exercise name.
 3. Include sets and reps.
 4. Include rest period.
-5. Adjust intensity based on BMI category.
-6. Avoid unsafe exercises for beginners.
+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.
 """
-
     return prompt, bmi, bmi_status