From a6b618d07248267de36f0e8f4a847d997285e272 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Wed, 17 May 2023 21:03:41 +0300
Subject: use a single function for saving images with metadata both in extra
 networks and main mode for #10395

---
 modules/ui_extra_networks.py | 19 ++-----------------
 1 file changed, 2 insertions(+), 17 deletions(-)

(limited to 'modules/ui_extra_networks.py')

diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 471df23b..c6e45fb1 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -4,7 +4,7 @@ from pathlib import Path
 from PIL import PngImagePlugin
 
 from modules import shared
-from modules.images import read_info_from_image
+from modules.images import read_info_from_image, save_image_with_geninfo
 import gradio as gr
 import json
 import html
@@ -343,22 +343,7 @@ def setup_ui(ui, gallery):
 
         assert is_allowed, f'writing to {filename} is not allowed'
 
-        if geninfo:
-            ext = os.path.splitext(filename)[1].lower()
-            if ext == '.png':
-                pnginfo_data = PngImagePlugin.PngInfo()
-                pnginfo_data.add_text('parameters', geninfo)
-                image.save(filename, pnginfo=pnginfo_data)
-            elif ext in ('.jpg', '.jpeg', '.webp'):
-                exif_bytes = piexif.dump({
-                    'Exif': {piexif.ExifIFD.UserComment: piexif.helper.UserComment.dump(geninfo or '',
-                                                                                        encoding='unicode')}
-                })
-                image.save(filename, exif=exif_bytes, quality=shared.opts.jpeg_quality)
-            else:
-                image.save(filename)
-        else:
-            image.save(filename)
+        save_image_with_geninfo(image, geninfo, filename)
 
         return [page.create_html(ui.tabname) for page in ui.stored_extra_pages]
 
-- 
cgit v1.2.1