From eb75d4e583aede141a1f07c2d7c2b500d8912439 Mon Sep 17 00:00:00 2001 From: zanllp Date: Wed, 27 Mar 2024 01:27:45 +0800 Subject: [PATCH] feat: Add exception handling --- scripts/iib/parsers/comfyui.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/scripts/iib/parsers/comfyui.py b/scripts/iib/parsers/comfyui.py index 0c4aea1..0998e16 100644 --- a/scripts/iib/parsers/comfyui.py +++ b/scripts/iib/parsers/comfyui.py @@ -21,12 +21,15 @@ class ComfyUIParser: params = None if not clz.test(img): raise Exception("The input image does not match the current parser.") - if is_img_created_by_comfyui_with_webui_gen_info(img): - info = read_sd_webui_gen_info_from_image(img, file_path) - params = parse_generation_parameters(info) - else: - params = get_comfyui_exif_data(img) - info = comfyui_exif_data_to_str(params) + try: + if is_img_created_by_comfyui_with_webui_gen_info(img): + info = read_sd_webui_gen_info_from_image(img, file_path) + params = parse_generation_parameters(info) + else: + params = get_comfyui_exif_data(img) + info = comfyui_exif_data_to_str(params) + except Exception: + return ImageGenerationInfo() return ImageGenerationInfo( info, ImageGenerationParams(