update path for flux

This commit is contained in:
Karl Hudgell 2025-04-24 20:26:56 +01:00
parent 9aea4e63fc
commit 4acf28e485

4
lib.py
View File

@ -276,8 +276,8 @@ def get_prompt_from_png(path):
with Image.open(path) as img: with Image.open(path) as img:
try: try:
# Flux workflow # Flux workflow
meta = img.info.get("parameters").split("Negative")[0] meta = json.loads(img.info["prompt"])['44']['inputs']['text']
except AttributeError: except KeyError:
# SDXL workflow # SDXL workflow
meta = json.loads(img.info["prompt"])['6']['inputs']['text'] meta = json.loads(img.info["prompt"])['6']['inputs']['text']
return meta or "" return meta or ""