From fb0bd6dd31dad43669403baf132ed8e96f03baff Mon Sep 17 00:00:00 2001
From: Mel Massadian <mel@melmassadian.com>
Date: Tue, 30 Jul 2024 21:54:49 +0200
Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9B=20limit=20loadImage=20dropd?=
 =?UTF-8?q?own=20to=20supported=20formats?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nodes.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index 16f5c9b00..eeaa00b90 100644
--- a/nodes.py
+++ b/nodes.py
@@ -9,6 +9,7 @@ import math
 import time
 import random
 import logging
+from pathlib import Path
 
 from PIL import Image, ImageOps, ImageSequence, ImageFile
 from PIL.PngImagePlugin import PngInfo
@@ -1523,8 +1524,9 @@ class PreviewImage(SaveImage):
 class LoadImage:
     @classmethod
     def INPUT_TYPES(s):
-        input_dir = folder_paths.get_input_directory()
-        files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))]
+        input_dir = Path(folder_paths.get_input_directory())
+        pillow_formats = {ext.lower() for ext in Image.registered_extensions().keys()}
+        files = sorted([f.name for f in input_dir.iterdir() if f.is_file() and f.suffix.lower() in pillow_formats])
         return {"required":
                     {"image": (sorted(files), {"image_upload": True})},
                 }