From b6b9faa779ca4239ef3c06abddd92d4520effd0e Mon Sep 17 00:00:00 2001 From: AUTOMATIC <16777216c@gmail.com> Date: Tue, 13 Sep 2022 19:23:55 +0300 Subject: add support for reading saved jpeg comments --- modules/extras.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) (limited to 'modules/extras.py') diff --git a/modules/extras.py b/modules/extras.py index 596cd172..cb083544 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -6,6 +6,8 @@ from modules.shared import opts import modules.gfpgan_model from modules.ui import plaintext_to_html import modules.codeformer_model +import piexif + cached_images = {} @@ -73,8 +75,20 @@ def run_extras(image, gfpgan_visibility, codeformer_visibility, codeformer_weigh def run_pnginfo(image): + items = image.info + + if "exif" in image.info: + exif = piexif.load(image.info["exif"]) + exif_comment = (exif or {}).get("Exif", {}).get(piexif.ExifIFD.UserComment, b'') + exif_comment = exif_comment.decode("utf8", 'ignore') + items['exif comment'] = exif_comment + + for field in ['jfif', 'jfif_version', 'jfif_unit', 'jfif_density', 'dpi', 'exif']: + del items[field] + + info = '' - for key, text in image.info.items(): + for key, text in items.items(): info += f"""

{plaintext_to_html(str(key))}

-- cgit v1.2.3