Update analyzer/process_video.py

analyzer/process_video.py

@@ -120,5 +120,22 @@ def analyze_video(video_path, video_id, user_id):
 
     print(f"✅ Saved result to: {output_file}")
     print(f"✅ Returning filename: {video_id}_popup_analysis.png")
-    return f"{video_id}_popup_analysis.png"
+    
+    # Save 2–3 key frames as examples
+    frame_filenames = []
+    for i, frame in enumerate(frames[:3]):  # Adjust how many you want
+        frame_filename = f"frame_{i}.jpg"
+        frame_path = os.path.join(output_folder, frame_filename)
+        cv2.imwrite(frame_path, frame)
+        frame_filenames.append(frame_filename)
+
+    # Log output
+    print(f"✅ Saved plot to: {output_file}")
+    print(f"📤 Returning frames: {frame_filenames}")
+
+    # Return both plot and frame image names
+    return {
+        "plot": f"{video_id}_popup_analysis.png",
+        "frames": frame_filenames
+    }