refix

backend/app/services/ai_summary.py

@@ -10,11 +10,32 @@ Be specific with numbers. Use a friendly, coaching tone.
 Respond in Russian.
 Use a HTML text formatting."""
 
+SYSTEM_PROMPT_WITH_PLAN = """You are VeloBrain, an AI cycling coach.
+The athlete completed a workout that was part of their training plan.
 
-def _build_activity_prompt(activity: Activity, rider_ftp: float | None = None) -> str:
+Your analysis MUST include two sections:
+
+1. **Анализ тренировки** — General performance summary (2-3 sentences): highlights, pacing, training effect.
+2. **Соответствие плану** — How well the actual workout matched the planned workout (2-4 sentences). Compare:
+   - Actual duration vs planned duration
+   - Actual intensity (IF, TSS, avg power) vs planned targets
+   - Workout type match (was it the right type of effort?)
+   - Give a compliance verdict: fully matched / partially matched / deviated significantly
+   - If deviated, explain what was different and whether the deviation was acceptable or needs attention.
+
+Be specific with numbers. Use a friendly, coaching tone.
+Respond in Russian.
+Use HTML text formatting."""
+
+
+def _build_activity_prompt(
+    activity: Activity,
+    rider_ftp: float | None = None,
+    planned_workout: dict | None = None,
+) -> str:
     m: ActivityMetrics | None = activity.metrics
     lines = [
-        f"Activity: {activity.name or 'Ride'}",
+        f"Activity: {activity.name or 'Workout'}",
         f"Type: {activity.activity_type}",
         f"Duration: {activity.duration // 3600}h {(activity.duration % 3600) // 60}m",
     ]
@@ -41,10 +62,25 @@ def _build_activity_prompt(activity: Activity, rider_ftp: float | None = None) -
             lines.append(f"Max HR: {m.max_hr} bpm")
         if m.avg_cadence:
             lines.append(f"Avg Cadence: {m.avg_cadence} rpm")
+        if m.calories:
+            lines.append(f"Calories: {m.calories} kcal")
 
     if rider_ftp:
         lines.append(f"Rider FTP: {rider_ftp:.0f} W")
 
+    # Exercise sets for strength workouts
+    if activity.exercise_sets:
+        lines.append(f"\nExercise sets ({len(activity.exercise_sets)} total):")
+        for s in activity.exercise_sets:
+            parts = [s.get("exercise_name", "Unknown")]
+            if s.get("repetitions"):
+                parts.append(f"{s['repetitions']} reps")
+            if s.get("weight"):
+                parts.append(f"{s['weight']} kg")
+            if s.get("duration") and not s.get("repetitions"):
+                parts.append(f"{s['duration']:.0f}s")
+            lines.append(f"  - {' / '.join(parts)}")
+
     intervals = activity.intervals or []
     work_intervals = [i for i in intervals if i.interval_type == "work"]
     if work_intervals:
@@ -53,10 +89,29 @@ def _build_activity_prompt(activity: Activity, rider_ftp: float | None = None) -
         if powers:
             lines.append(f"Interval avg powers: {', '.join(f'{p:.0f}W' for p in powers)}")
 
+    # Planned workout from training plan
+    if planned_workout:
+        lines.append("\n--- PLANNED WORKOUT ---")
+        lines.append(f"Planned type: {planned_workout.get('workout_type', '?')}")
+        lines.append(f"Planned title: {planned_workout.get('title', '?')}")
+        if planned_workout.get("description"):
+            lines.append(f"Planned description: {planned_workout['description']}")
+        if planned_workout.get("duration_minutes"):
+            lines.append(f"Planned duration: {planned_workout['duration_minutes']} min")
+        if planned_workout.get("target_tss"):
+            lines.append(f"Planned TSS: {planned_workout['target_tss']}")
+        if planned_workout.get("target_if"):
+            lines.append(f"Planned IF: {planned_workout['target_if']}")
+
     return "\n".join(lines)
 
 
-async def generate_summary(activity: Activity, rider_ftp: float | None = None) -> str:
-    prompt = _build_activity_prompt(activity, rider_ftp)
+async def generate_summary(
+    activity: Activity,
+    rider_ftp: float | None = None,
+    planned_workout: dict | None = None,
+) -> str:
+    prompt = _build_activity_prompt(activity, rider_ftp, planned_workout)
     messages = [{"role": "user", "text": prompt}]
-    return await chat_async(messages, system_instruction=SYSTEM_PROMPT, temperature=0.5)
+    system = SYSTEM_PROMPT_WITH_PLAN if planned_workout else SYSTEM_PROMPT
+    return await chat_async(messages, system_instruction=system, temperature=0.5)