From fb04e436c26d4b6126acee0644f66d6bfddc76ec Mon Sep 17 00:00:00 2001
From: sakthivel <37198254+sakthivelj@users.noreply.github.com>
Date: Sun, 5 May 2024 09:57:09 +0530
Subject: [PATCH] Fixed issue with video file detection (fixes #1)

---
 video2image/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/video2image/__init__.py b/video2image/__init__.py
index 9100fa3..5551885 100644
--- 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: