Merge pull request #11596 from akx/use-read-info

postprocessing: use read_info_from_image

1 files changed, 12 insertions, 5 deletions

diff --git a/modules/images.py b/modules/images.py
index cdfa6719..fdccec09 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -1,3 +1,5 @@
+from __future__ import annotations

+

 import datetime

 

 import pytz

@@ -678,8 +680,15 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
     return fullfn, txt_fullfn

 

 

-def read_info_from_image(image):

-    items = image.info or {}

+IGNORED_INFO_KEYS = {

+    'jfif', 'jfif_version', 'jfif_unit', 'jfif_density', 'dpi', 'exif',

+    'loop', 'background', 'timestamp', 'duration', 'progressive', 'progression',

+    'icc_profile', 'chromaticity', 'photoshop',

+}

+

+

+def read_info_from_image(image: Image.Image) -> tuple[str | None, dict]:

+    items = (image.info or {}).copy()

 

     geninfo = items.pop('parameters', None)

 

@@ -695,9 +704,7 @@ def read_info_from_image(image):
             items['exif comment'] = exif_comment

             geninfo = exif_comment

 

-    for field in ['jfif', 'jfif_version', 'jfif_unit', 'jfif_density', 'dpi', 'exif',

-                    'loop', 'background', 'timestamp', 'duration', 'progressive', 'progression',

-                    'icc_profile', 'chromaticity']:

+    for field in IGNORED_INFO_KEYS:

         items.pop(field, None)

 

     if items.get("Software", None) == "NovelAI":