use movie for video compilation

requirements.txt

@@ -8,3 +8,4 @@ opencv-python
 torch
 torchvision
 tqdm
+moviepy

services/summarizer.py

@@ -5,6 +5,7 @@ from services.model_loader import load_model
 import subprocess
 import os
 import numpy as np
+from moviepy import ImageSequenceClip
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = load_model("Model/epoch-199.pkl")
@@ -31,54 +32,24 @@ def get_selected_indices(scores, picks, threshold=SCORE_THRESHOLD):
     return selected
 
 def save_summary_video(video_path, selected_indices, output_path, fps=15):
-    import cv2
-
     cap = cv2.VideoCapture(video_path)
     selected = set(selected_indices)
-    frame_id = 0
     frames = []
+    frame_id = 0
 
-    while cap.isOpened():
+    while True:
         ret, frame = cap.read()
         if not ret:
             break
         if frame_id in selected:
+            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
             frames.append(frame)
         frame_id += 1
     cap.release()
 
-    if len(frames) == 0:
+    if not frames:
         print("No frames selected.")
         return
 
-    h, w, _ = frames[0].shape
-    print("Video dimensions:", w, h)
-
-    ext = output_path.split('.')[-1]
-    raw_output_path = output_path.replace(f".{ext}", f"_raw.{ext}")
-
-    out = cv2.VideoWriter(raw_output_path, cv2.VideoWriter_fourcc(*'mp4v'), fps, (h, w))
-    for frame in frames:
-        out.write(frame)
-    out.release()
-
-    print("Fixing the video with ffmpeg")
-    fix_video_with_ffmpeg(raw_output_path, output_path)
-
-def fix_video_with_ffmpeg(input_path, output_path):
-    # 2️⃣ Use FFmpeg to fix video (browser-compatible)
-    try:
-        subprocess.run([
-            "ffmpeg",
-            "-y",  # overwrite if file exists
-            "-i", input_path,
-            "-vcodec", "libx264",
-            "-acodec", "aac",
-            output_path,
-        ], check=True)
-        print(f"✅ FFmpeg re-encoded video saved to: {output_path}")
-        os.remove(input_path)
-    except subprocess.CalledProcessError as e:
-        print("❌ FFmpeg failed:", e)
-        print("⚠️ Using raw video instead.")
-        os.rename(input_path, output_path)
+    clip = ImageSequenceClip(frames, fps=fps)
+    clip.write_videofile(output_path, codec="libx264")