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

decode webp metadata for images generated with comfyui

pull/798/head
Ancions-dot 2025-05-07 23:42:04 +02:00
parent 847aec69e7
commit a8cff90617
1 changed files with 28 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
scripts/iib/tool.py
View File

@ -394,14 +394,39 @@ def get_img_geninfo_txt_path(path: str):
return txt_path
def is_img_created_by_comfyui(img: Image):
prompt = img.info.get('prompt')
return prompt and (img.info.get('workflow') or ("class_type" in prompt)) # ermanitu
if img.format == "PNG":
prompt = img.info.get('prompt')
return prompt and (img.info.get('workflow') or ("class_type" in prompt)) # ermanitu
elif img.format == "WEBP":
exif = img.info.get("exif")
split = [x.decode("utf-8", errors="ignore") for x in exif.split(b"\x00")]
workflow_str = find(split, lambda x: x.lower().startswith("workflow:"))
prompt_str = find(split, lambda x: x.lower().startswith("prompt:"))
if workflow_str and prompt_str:
workflow = json.loads(workflow_str.split(":", 1)[1])
prompt = json.loads(prompt_str.split(":", 1)[1])
return (
workflow
and prompt
and any("class_type" in x.keys() for x in prompt.values())
)
else:
return False
else:
return False # unsupported format
def is_img_created_by_comfyui_with_webui_gen_info(img: Image):
return is_img_created_by_comfyui(img) and img.info.get('parameters')
def get_comfyui_exif_data(img: Image):
prompt = img.info.get('prompt')
if img.format == "PNG":
prompt = img.info.get('prompt')
elif img.format == "WEBP":
exif = img.info.get("exif")
split = [x.decode("utf-8", errors="ignore") for x in exif.split(b"\x00")]
prompt_str = find(split, lambda x: x.lower().startswith("prompt:"))
if prompt_str:
prompt = prompt_str.split(":", 1)[1] if prompt_str else None
if not prompt:
return {}
meta_key = '3'