Update video_processing.py

video_processing.py

@@ -103,19 +103,18 @@ def convert_timestamp_to_seconds(timestamp):
     h, m, s = map(float, timestamp.split(':'))
     return int(h) * 3600 + int(m) * 60 + s
 
-def extract_frames(video_path, start_time, end_time):
-    start_seconds = convert_timestamp_to_seconds(start_time)
-    end_seconds = convert_timestamp_to_seconds(end_time)
-    video_clip = VideoFileClip(video_path).subclip(start_seconds, end_seconds)
-
-    def extract_frame_at_time(t):
-        return video_clip.get_frame(t / video_clip.fps)
+def extract_frame_at_time(args):
+    video_clip, t = args
+    return video_clip.get_frame(t / video_clip.fps)
 
+def extract_frames(video_path, start_time, end_time):
+    video_clip = VideoFileClip(video_path).subclip(start_time, end_time)
     frame_times = range(0, int(video_clip.duration * video_clip.fps), int(video_clip.fps / 10))
     
     # Create a pool of workers to extract frames in parallel
     with Pool() as pool:
-        frames = pool.map(extract_frame_at_time, frame_times)
+        # We need to pass video_clip and t to the function, wrap in tuple
+        frames = pool.map(extract_frame_at_time, ((video_clip, t) for t in frame_times))
 
     return frames