Update extract_frames.py

extract_frames.py

@@ -1,8 +1,11 @@
 import cv2
-import os , shutil
+import os, shutil
+
 
 def ExtractFrames(videoPath):
-    shutil.rmtree("data/output")
+    directory = "data/output"
+    if os.path.exists(directory):
+        shutil.rmtree(directory)
     # Path to the video file
     video_path = videoPath
 
@@ -32,7 +35,7 @@ def ExtractFrames(videoPath):
             break
 
         # Construct the filename for the frame
-        frame_filename = os.path.join(output_dir, f'frame_{frame_index:04d}.png')
+        frame_filename = os.path.join(output_dir, f'frames_{frame_index:04d}.png')
 
         # Save the current frame as an image file
         cv2.imwrite(frame_filename, frame)