From fdfe7c29ed31f3432839e7c0ac5e20b9105880d4 Mon Sep 17 00:00:00 2001 From: SoClose <33631880+SoClosee@users.noreply.github.com> Date: Wed, 4 Mar 2026 16:22:41 +0100 Subject: [PATCH] fix: improve image discovery performance --- main.py | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) diff --git a/main.py b/main.py index 886b7dc..37d6866 100644 --- a/main.py +++ b/main.py @@ -133,23 +133,17 @@ def load_csv_metadata(csv_path): def discover_images(folder_path): - """Return a sorted list of image file paths in *folder_path*.""" + """Return a generator of image file paths in *folder_path*.""" folder = Path(folder_path) if not folder.is_dir(): logger.error("Images folder not found: %s", folder) sys.exit(1) - images = sorted( - str(p) for p in folder.iterdir() - if p.is_file() and p.suffix.lower() in SUPPORTED_EXTENSIONS - ) - - if not images: - logger.error("No images found in %s", folder) - sys.exit(1) + for p in folder.iterdir(): + if p.is_file() and p.suffix.lower() in SUPPORTED_EXTENSIONS: + yield str(p) - logger.info("Found %d image(s) in %s", len(images), folder) - return images + logger.info("Found images in %s", folder) def create_driver(headless=False):