Make live previews use JPEG only when the image is lorge enough

1 files changed, 10 insertions, 2 deletions

diff --git a/modules/progress.py b/modules/progress.py
index 289dd311..c2e37834 100644
--- a/modules/progress.py
+++ b/modules/progress.py
@@ -95,9 +95,17 @@ def progressapi(req: ProgressRequest):
         image = shared.state.current_image

         if image is not None:

             buffered = io.BytesIO()

-            image.save(buffered, format=opts.live_previews_format)

+            format = opts.live_previews_format

+            save_kwargs = {}

+            if format == "auto":

+                if max(*image.size) > 256:

+                    format = "jpeg"

+                else:

+                    format = "png"

+                    save_kwargs = {"optimize": True}

+            image.save(buffered, format=format, **save_kwargs)

             base64_image = base64.b64encode(buffered.getvalue()).decode('ascii')

-            live_preview = f"data:image/{opts.live_previews_format};base64,{base64_image}"

+            live_preview = f"data:image/{format};base64,{base64_image}"

             id_live_preview = shared.state.id_live_preview

         else:

             live_preview = None