Update video_processing.py

video_processing.py

@@ -50,8 +50,8 @@ 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)
-    for frame_time in range(0, int(video_clip.duration), 5):
-        frame = video_clip.get_frame(frame_time)
+    for frame_time in range(0, int(video_clip.duration * video_clip.fps), 5):
+        frame = video_clip.get_frame(frame_time / video_clip.fps)
         frames.append(frame)
     return frames
 
@@ -61,6 +61,10 @@ def analyze_scenes(video_path, scenes, description):
 
     for scene_num, (start_time, end_time) in enumerate(scenes):
         frames = extract_frames(video_path, start_time, end_time)
+        if not frames:
+            print(f"Scene {scene_num + 1}: Start={start_time}, End={end_time} - No frames extracted")
+            continue
+
         scene_prob = 0.0
         for frame in frames:
             image = Image.fromarray(frame[..., ::-1])
@@ -71,7 +75,6 @@ def analyze_scenes(video_path, scenes, description):
                 probs = logits_per_image.softmax(dim=1)
                 scene_prob += max(probs[0]).item()
 
-        # Average the probabilities over the frames
         scene_prob /= len(frames)
         print(f"Scene {scene_num + 1}: Start={start_time}, End={end_time}, Probability={scene_prob}")
 
@@ -79,7 +82,11 @@ def analyze_scenes(video_path, scenes, description):
             highest_prob = scene_prob
             best_scene = (start_time, end_time)
 
-    print(f"Best Scene: Start={best_scene[0]}, End={best_scene[1]}, Probability={highest_prob}")
+    if best_scene:
+        print(f"Best Scene: Start={best_scene[0]}, End={best_scene[1]}, Probability={highest_prob}")
+    else:
+        print("No suitable scene found")
+
     return best_scene
 
 def extract_best_scene(video_path, scene):