Update prompt_builder.py

prompt_builder.py

@@ -23,26 +23,56 @@ 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.
+IMPORTANT:
+- Do NOT write introduction.
+- Do NOT write explanation.
+- Do NOT give nutrition advice.
+- Output ONLY the workout plan.
+- Keep it concise.
 
 User Profile:
-- Name: {name}
-- 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.
-6. Avoid unsafe exercises for beginners.
-7. Keep the plan professional and easy to follow.
+Name: {name}
+Gender: {gender}
+Height: {height} cm
+Weight: {weight} kg
+BMI: {bmi:.2f} ({bmi_status})
+Goal: {goal}
+Fitness Level: {fitness_level}
+Available Equipment: {equipment_list}
+
+STRICT FORMAT:
+
+Day 1:
+Warm-up:
+Main Workout (sets x reps):
+Rest:
+Cooldown:
+
+Day 2:
+Warm-up:
+Main Workout (sets x reps):
+Rest:
+Cooldown:
+
+Day 3:
+Warm-up:
+Main Workout (sets x reps):
+Rest:
+Cooldown:
+
+Day 4:
+Warm-up:
+Main Workout (sets x reps):
+Rest:
+Cooldown:
+
+Day 5:
+Warm-up:
+Main Workout (sets x reps):
+Rest:
+Cooldown:
+
+Each day must contain at least 4 exercises.
 """
 
     return prompt, bmi, bmi_status