From 7fa9193112dc243ae9d520a138cc0c30929f1b4d Mon Sep 17 00:00:00 2001
From: normalizedwater546 <normalizedwater546@ak.gy>
Date: Sun, 22 Sep 2024 02:05:32 +0000
Subject: [PATCH] fix: non-image files in pdf conversion causing crash

---
 nhentai/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nhentai/utils.py b/nhentai/utils.py
index 46ca752..d1fa5fe 100644
--- a/nhentai/utils.py
+++ b/nhentai/utils.py
@@ -237,7 +237,7 @@ def generate_pdf(output_dir='.', doujinshi_obj=None, rm_origin_dir=False, move_t
             logger.info(f'Skipped download: {doujinshi_dir} already exists')
             return
 
-        file_list = os.listdir(doujinshi_dir)
+        file_list = [f for f in os.listdir(doujinshi_dir) if f.lower().endswith(('.png', '.jpg', '.jpeg', '.gif'))]
         file_list.sort()
 
         logger.info(f'Writing PDF file to path: {filename}')