Fixed issue with video file detection (fixes #1)

sakthivelj · May 5, 2024 · fb04e43 · fb04e43
1 parent 708f237
commit fb04e43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/video2image/__init__.py b/video2image/__init__.py
@@ -24,7 +24,7 @@ def main():
 
     # Handle directory with multiple video files
     elif os.path.isdir(args.input):
-        video_files = glob.glob(os.path.join(args.input, "*.mp4"))
+        video_files = glob.glob(os.path.join(args.input, "*.[mp4|mkv|avi|mov]"))  # Updated line
         if len(video_files) == 0:
             print("No video files found in the input directory")
         else: