Update prompt_builder.py

prompt_builder.py

@@ -4,45 +4,33 @@ def calculate_bmi(weight, height):
 
 def bmi_category(bmi):
     if bmi < 18.5:
-        return "Underweight"
+        return "Underweight", "#3498db"
     elif bmi < 25:
-        return "Normal Weight"
+        return "Normal Weight", "#2ecc71"
     elif bmi < 30:
-        return "Overweight"
+        return "Overweight", "#f1c40f"
     else:
-        return "Obese"
-
+        return "Obese", "#e74c3c"
 
 def build_prompt(name, gender, height, weight, goal, fitness_level, equipment):
-
     bmi = calculate_bmi(weight, height)
-    bmi_status = bmi_category(bmi)
-
+    bmi_status, status_color = bmi_category(bmi)
     equipment_list = ", ".join(equipment) if equipment else "No Equipment"
 
-    prompt = f"""
-You are a certified professional fitness trainer.
-
-Create a structured 5-day personalized workout plan.
-
+    # Strict instructions for Mistral-7B
+    prompt = f"""Create a professional 5-day workout plan for {name}.
 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}
+- 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.
-"""
+Requirements:
+1. Provide a specific routine for Day 1, Day 2, Day 4, and Day 5.
+2. Day 3 MUST be a Rest and Recovery day.
+3. For every exercise, include the Name, Sets, Reps, and Rest Time.
+4. Do not include any introductory text, start directly with Day 1.
+5. Stop exactly after finishing Day 5."""
 
-    return prompt, bmi, bmi_status
+    return prompt, bmi, bmi_status, status_color