1580158
/
sd-webui-infinite-image-browsing
mirror of https://github.com/zanllp/sd-webui-infinite-image-browsing.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update update_image_data.py to check for webui geninfo in ComfyUI images

pull/379/head
Chubbly 2023-08-22 11:56:25 +12:00 committed by GitHub
parent 2609cdce94
commit b0c2d9384f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/iib/db/update_image_data.py
View File

@ -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)