Unverified Commit f517838c authored by Keith's avatar Keith Committed by GitHub
Browse files

Fix extra networks save preview image geninfo

parent 89f9faa6
Loading
Loading
Loading
Loading
+13 −3
Original line number Diff line number Diff line
@@ -334,9 +334,19 @@ 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)