From b0c2d9384f33ebeec34c1f29d6e7487b7969c30d Mon Sep 17 00:00:00 2001 From: Chubbly Date: Tue, 22 Aug 2023 11:56:25 +1200 Subject: [PATCH] Update update_image_data.py to check for webui geninfo in ComfyUI images --- scripts/iib/db/update_image_data.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/scripts/iib/db/update_image_data.py b/scripts/iib/db/update_image_data.py index 1f1f136..54ac559 100644 --- a/scripts/iib/db/update_image_data.py +++ b/scripts/iib/db/update_image_data.py @@ -10,7 +10,8 @@ from scripts.iib.tool import ( is_dev, get_comfyui_exif_data, comfyui_exif_data_to_str, - is_img_created_by_comfyui + is_img_created_by_comfyui, + is_img_created_by_comfyui_with_webui_gen_info ) from scripts.iib.logger import logger @@ -23,8 +24,12 @@ def get_exif_data(file_path): try: with Image.open(file_path) as img: if is_img_created_by_comfyui(img): - params = get_comfyui_exif_data(img) - info = comfyui_exif_data_to_str(params) + 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) else: info = read_sd_webui_gen_info_from_image(img, file_path) params = parse_generation_parameters(info)