Update utils/ai.py

utils/ai.py

@@ -1,10 +1,41 @@
 """
-ai.py — Groq-powered workout + diet plan generation.
+ai.py — Groq-powered workout + diet plan generation with exercise descriptions.
 """
 import os, time
 
 GROQ_MODEL = "llama-3.3-70b-versatile"
 
+# ── Exercise description database ─────────────────────────────────────────────
+EXERCISE_DESCRIPTIONS = {
+    "push": "Keep body straight, lower chest to floor, push back up explosively.",
+    "pull": "Hang from bar, drive elbows down to pull chin above the bar.",
+    "squat": "Feet shoulder-width, push knees out, descend until thighs parallel.",
+    "deadlift": "Hinge hips back, grip bar outside knees, drive through heels to stand.",
+    "lunge": "Step forward, lower back knee toward floor, keep torso upright.",
+    "plank": "Forearms on floor, body in straight line, brace core throughout.",
+    "curl": "Upper arms fixed, curl weight toward shoulder, control the descent.",
+    "press": "Start at shoulders, press overhead until arms fully extended.",
+    "row": "Hinge forward, retract shoulder blade, pull elbow past torso.",
+    "fly": "Slight elbow bend, open arms wide, squeeze chest to close.",
+    "raise": "Keep slight bend in elbow, raise to shoulder height, lower slowly.",
+    "dip": "Grip parallel bars, lower until upper arm parallel, press back up.",
+    "crunch": "Feet flat, curl upper body off floor, exhale at the top.",
+    "leg raise": "Lie flat, raise legs to 90°, lower slowly without touching floor.",
+    "glute bridge": "Feet flat, drive hips up, squeeze glutes at the top for 1s.",
+    "mountain climber": "High plank, drive knees alternately toward chest at pace.",
+    "burpee": "Squat to floor, jump feet back, push-up, jump feet in, leap up.",
+    "jump": "Load hips, explode upward, land softly with bent knees.",
+    "stretch": "Ease into position, hold without bouncing, breathe steadily.",
+    "band": "Anchor band, maintain tension throughout full range of motion.",
+}
+
+def get_exercise_description(name):
+    name_lower = name.lower()
+    for key, desc in EXERCISE_DESCRIPTIONS.items():
+        if key in name_lower:
+            return desc
+    return "Perform with controlled movement through full range of motion."
+
 def _groq_client():
     key = os.environ.get("GROQ_API_KEY", "").strip()
     if not key:
@@ -109,23 +140,25 @@ GOAL: {goal} | LEVEL: {level} | EQUIPMENT: {eq_str}
 {context}
 
 FORMAT — use EXACTLY this for each day:
-## Day N — [Focus Area]
-**Warm-Up**
-- Exercise: 2×10
-- Exercise: 2×10
+## Day N — [Focus Area e.g. Chest & Triceps]
+**Warm-Up** (5–8 min)
+- Exercise Name — 2×10 reps (rest 30s)
+- Exercise Name — 2×10 reps (rest 30s)
+
 **Main Workout**
 - Exercise Name — 3×12 reps (rest 60s)
 - Exercise Name — 3×12 reps (rest 60s)
 - Exercise Name — 3×12 reps (rest 60s)
-- Exercise Name — 3×12 reps (rest 60s)
-- Exercise Name — 3×12 reps (rest 60s)
-**Cool-Down**
-- Stretch 1 — 30s each side
-- Stretch 2 — 30s
+- Exercise Name — 3×10 reps (rest 75s)
+- Exercise Name — 3×15 reps (rest 45s)
+
+**Cool-Down & Stretching**
+- Stretch Name — hold 30s each side
+- Stretch Name — hold 30s
 
 RULES: {intensity} | BMI note: {advice} | Only use: {eq_str}
-{'End with one motivational line for '+name+'.' if end_day==days else 'No closing text — more days follow.'}
-Output ONLY Days {start_day}–{end_day}. No preamble."""
+{'End with one short motivational sentence for '+name+'.' if end_day==days else 'No closing text — more days follow.'}
+Output ONLY Days {start_day}–{end_day}. No preamble. No extra text."""
 
         chunk = _call_groq(prompt, max_tokens=1800)
         all_workout_chunks.append(chunk.strip())
@@ -134,7 +167,7 @@ Output ONLY Days {start_day}–{end_day}. No preamble."""
             time.sleep(1)
 
     if progress_cb:
-        progress_cb(total_chunks, total_chunks, "Generating nutrition plan…")
+        progress_cb(total_chunks, total_chunks, "Generating personalised nutrition plan…")
 
     # Diet plan
     diet_prompt = f"""Create a detailed {days}-day nutrition plan for:
@@ -143,38 +176,47 @@ Output ONLY Days {start_day}–{end_day}. No preamble."""
 FORMAT:
 ## Daily Nutrition Targets
 - Calories: X kcal
-- Protein: Xg | Carbs: Xg | Fats: Xg
+- Protein: Xg | Carbs: Xg | Fats: Xg | Fibre: Xg
 - Water: X litres/day
+- Meal timing: brief note
 
 ## Sample Day Meal Plan
-**Breakfast** — [meal name]
-- Item: quantity/amount (Xcal, Xg protein)
+**Breakfast** — [Meal name] (~Xcal)
+- Food item: quantity (Xcal, Xg protein)
+- Food item: quantity
 
-**Mid-Morning Snack**
-- Item: quantity
+**Mid-Morning Snack** (~Xcal)
+- Food item: quantity
 
-**Lunch** — [meal name]
-- Item: quantity/amount (Xcal, Xg protein)
+**Lunch** — [Meal name] (~Xcal)
+- Food item: quantity (Xcal, Xg protein)
+- Food item: quantity
 
 **Pre-Workout** (if applicable)
-- Item: quantity
+- Food item: quantity
 
-**Dinner** — [meal name]
-- Item: quantity/amount (Xcal, Xg protein)
+**Post-Workout / Dinner** — [Meal name] (~Xcal)
+- Food item: quantity (Xcal, Xg protein)
+- Food item: quantity
 
-**Evening Snack**
-- Item: quantity
+**Evening Snack** (~Xcal)
+- Food item: quantity
 
 ## Key Supplements (optional)
-- Supplement: timing + dosage
+- Supplement: timing + dosage + benefit
 
 ## Foods to Prioritise
-- Food 1, Food 2, Food 3...
+- List 6–8 foods with brief reasons
 
 ## Foods to Limit
-- Food 1, Food 2, Food 3...
+- List 4–6 foods with brief reasons
+
+## Weekly Meal Prep Tips
+- Tip 1
+- Tip 2
+- Tip 3
 
-Be specific with quantities. Tailor everything to the {goal} goal and {bmi_cat} BMI status.
+Be specific with quantities. Tailor everything to the {goal} goal and {bmi_cat} BMI.
 Note: {advice}"""
 
     diet = _call_groq(diet_prompt, max_tokens=1500)