diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-17 20:42:37 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-17 20:42:37 +0300 |
| commit | 9c91a867201bf7570adb49ec857c8e50dd055c9f (patch) | |
| tree | 89d82b4e3fd2860eccf4961bc381963eba80be99 /modules/ui_extra_networks.py | |
| parent | 6b51cc7530dd70037681c108a65c55d52ed918ef (diff) | |
| parent | f517838c75014f981ae1c41f1bc776d74daf9a23 (diff) | |
Merge pull request #10395 from wk5ovc/patch-4
Fix extra networks save preview image geninfo
Diffstat (limited to 'modules/ui_extra_networks.py')
| -rw-r--r-- | modules/ui_extra_networks.py | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py index 27cb9807..471df23b 100644 --- a/modules/ui_extra_networks.py +++ b/modules/ui_extra_networks.py @@ -344,9 +344,19 @@ def setup_ui(ui, gallery): assert is_allowed, f'writing to {filename} is not allowed'
if geninfo:
- pnginfo_data = PngImagePlugin.PngInfo()
- pnginfo_data.add_text('parameters', geninfo)
- image.save(filename, pnginfo=pnginfo_data)
+ 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)
|