diff --git a/.gitignore b/.gitignore index c514c5d..f2c4413 100644 --- a/.gitignore +++ b/.gitignore @@ -168,4 +168,8 @@ poetry.toml .ruff_cache/ # LSP config files -pyrightconfig.json \ No newline at end of file +pyrightconfig.json + +rejects/ +done/ +.vscode/ \ No newline at end of file diff --git a/cli.py b/cli.py index 9bc4e67..18af5cf 100644 --- a/cli.py +++ b/cli.py @@ -1,5 +1,6 @@ import argparse from pathlib import Path +import cv2 from sorter.image_sorter import process_image # CLI argument parsing @@ -28,7 +29,13 @@ def main(): # Process the images for image_path in image_paths: - process_image(image_path, pass_folder, reject_folder) + """ + try catch block to handle errors in processing individual images. + """ + try: + process_image(image_path, pass_folder, reject_folder) + except cv2.error as e: + print(f"Error processing image {image_path.name}: {str(e)}") print("Sorting complete.")