Add cleaning logic

modules/utils/video_helper.py

@@ -23,11 +23,15 @@ class VideoInfo:
 def extract_frames(
     vid_input: str,
     output_temp_dir: str = TEMP_VIDEO_FRAMES_DIR,
-    start_number: int = 0
+    start_number: int = 0,
+    clean=True
 ):
     """
     Extract frames as jpg files and save them into output_temp_dir. This needs FFmpeg installed.
     """
+    if clean:
+        clean_temp_dir(temp_dir=output_temp_dir)
+
     os.makedirs(output_temp_dir, exist_ok=True)
     output_path = os.path.join(output_temp_dir, "%05d.jpg")
 
@@ -258,7 +262,9 @@ def clean_temp_dir(temp_dir: Optional[str] = None):
 
     clean_files_with_extension(temp_dir, SOUND_FILE_EXT)
     clean_files_with_extension(temp_dir, IMAGE_FILE_EXT)
-    clean_files_with_extension(temp_out_dir, IMAGE_FILE_EXT)
+
+    if os.path.exists(temp_out_dir):
+        clean_files_with_extension(temp_out_dir, IMAGE_FILE_EXT)
 
 
 def clean_files_with_extension(dir_path: str, extensions: List):