From 3527fac3a085654eedcd9b80a33809d35a43209c Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Tue, 4 Nov 2025 10:03:22 -0500
Subject: [PATCH] keep exif on thumbnail generation

Signed-off-by: Vladimir Mandic <mandic00@live.com>
---
 CHANGELOG.md                 |  2 ++
 modules/ui_extra_networks.py | 25 ++++++++++++++++++++++---
 2 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 359b2aa53..0d89b88d6 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -3,6 +3,7 @@
 ## Update for 2025-11-04
 
 - **Features**
+  - **ROCm for Windows** switch to using **TheRock** torch builds when available  
   - **detailer**
     optional include detection image to output results  
     optional sort detection objects left-to-right for improved prompt consistency  
@@ -44,6 +45,7 @@
   - `ui` mobile layout for video tab  
   - `video` save to subfolder  
   - `taesd` warn on long decode times  
+  - `metadata` keep exif on thumbnail generation  
 
 ## Update for 2025-10-31
 
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index f18e31ab8..333e3cd13 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -190,6 +190,22 @@ class ExtraNetworksPage:
         preview = f"{shared.opts.subpath}/sdapi/v1/network/thumb?filename={quoted_filename}&mtime={mtime}"
         return preview
 
+    def get_exif(self, image: Image.Image):
+        import piexif
+        try:
+            exifinfo = image.getexif()
+            if exifinfo is not None and len(exifinfo) > 0:
+                return piexif.dump({ "Exif": { piexif.ExifIFD.UserComment: piexif.helper.UserComment.dump(exifinfo, encoding="unicode") } })
+        except Exception:
+            pass
+        try:
+            exifinfo = image.info.get('parameters', None)
+            if exifinfo is not None and len(exifinfo) > 0:
+                return piexif.dump({ "Exif": { piexif.ExifIFD.UserComment: piexif.helper.UserComment.dump(exifinfo, encoding="unicode") } })
+        except Exception:
+            pass
+        return piexif.dump({ "Exif": { piexif.ExifIFD.UserComment: piexif.helper.UserComment.dump('', encoding="unicode") } })
+
     def create_thumb(self):
         debug(f'EN create-thumb: {self.name}')
         created = 0
@@ -203,6 +219,7 @@ class ExtraNetworksPage:
             img = None
             try:
                 img = Image.open(f)
+                img.load()
             except Exception as e:
                 img = None
                 shared.log.warning(f'Network removing invalid: image={f} {e}')
@@ -210,16 +227,18 @@ class ExtraNetworksPage:
                 if img is None:
                     img = None
                     os.remove(f)
-                elif img.width > 1024 or img.height > 1024 or os.path.getsize(f) > 65536:
+                elif (img.width > 1024) or (img.height > 1024) or (os.path.getsize(f) > 65536):
+                    exif = self.get_exif(img)
                     img = img.convert('RGB')
                     img.thumbnail((512, 512), Image.Resampling.HAMMING)
-                    img.save(fn, quality=50)
+                    img.save(fn, quality=50, exif=exif)
                     img.close()
                     created += 1
             except Exception as e:
                 shared.log.warning(f'Network create thumbnail={f} {e}')
+                errors.display(e, 'thumbnail')
         if created > 0:
-            shared.log.info(f'Network thumbnails: {self.name} created={created}')
+            shared.log.info(f'Network thumbnails: type={self.name} created={created}')
             self.missing_thumbs.clear()
 
     def create_items(self, tabname):