Added functionality to make sure that the video has enough frames

app.py

@@ -47,9 +47,20 @@ def delete_files_in_directory(directory):
         except Exception as e:
             print(f"Failed to delete {file_path}. Reason: {e}")
 
+def get_total_frames(container):
+    stream = container.streams.video[0]
+    total_frames = stream.frames
+    return total_frames
+
 def process_video(file_path):
     container = av.open(file_path)
-    indices = sample_frame_indices(clip_len=32, frame_sample_rate=2, seg_len=container.streams.video[0].frames)
+    total_frames = get_total_frames(container)
+    
+    if total_frames < 64:
+        container.close()
+        raise ValueError("Video must have at least 64 frames.")
+    
+    indices = sample_frame_indices(clip_len=32, frame_sample_rate=2, seg_len=total_frames)
     video = read_video_pyav(container=container, indices=indices)
     container.close()
     return video
@@ -134,7 +145,7 @@ def weighted_average_method(video_prediction, audio_prediction):
 def confidence_level_method(video_prediction, audio_prediction, threshold=0.7):
     highest_video_label = max(video_prediction, key=video_prediction.get)
     highest_video_confidence = video_prediction[highest_video_label]
-    if highest_video_confidence >= threshold:
+    if (highest_video_confidence >= threshold):
         return highest_video_label
     combined_probabilities = {}
     for label in set(video_prediction) | set(audio_prediction):