From 77f2d6c5706e5919e18bcd8426ab09cb584d3efd Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Sun, 24 Mar 2024 09:04:49 -0400
Subject: [PATCH] fix api generate save metadata

---
 CHANGELOG.md           | 2 ++
 cli/simple-txt2img.py  | 9 ++++-----
 modules/api/helpers.py | 3 +--
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index a58087dbc..2efa77fbc 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -9,6 +9,7 @@
 - Model load from dropdown select variant
 - VAE preview
 - SC LoRA
+- Gallery urlencode/decode
 
 ## Update for 2024-03-23
 
@@ -28,6 +29,7 @@
   - Prompt params parser
   - Fix image save without metadata
   - fix ROCm compatibility, thanks @Disty0
+  - Fix API generate save metadata
 
 ## Update for 2024-03-19
 
diff --git a/cli/simple-txt2img.py b/cli/simple-txt2img.py
index a9696c600..53565690c 100755
--- a/cli/simple-txt2img.py
+++ b/cli/simple-txt2img.py
@@ -12,16 +12,15 @@ from PIL import Image
 sd_url = os.environ.get('SDAPI_URL', "http://127.0.0.1:7860")
 sd_username = os.environ.get('SDAPI_USR', None)
 sd_password = os.environ.get('SDAPI_PWD', None)
+options = {
+    "save_images": True,
+    "send_images": True,
+}
 
 logging.basicConfig(level = logging.INFO, format = '%(asctime)s %(levelname)s: %(message)s')
 log = logging.getLogger(__name__)
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
-options = {
-    "save_images": False,
-    "send_images": True,
-}
-
 
 def auth():
     if sd_username is not None and sd_password is not None:
diff --git a/modules/api/helpers.py b/modules/api/helpers.py
index 8e11df5c7..6388d17e9 100644
--- a/modules/api/helpers.py
+++ b/modules/api/helpers.py
@@ -36,8 +36,7 @@ def encode_pil_to_base64(image):
         shared.log.error('API cannot encode image: not a PIL image')
         return ''
     buffered = io.BytesIO()
-    image_format = Image.registered_extensions()[f'.{shared.opts.samples_format}']
-    image.save(buffered, format=image_format)
+    save_image(image, fn=buffered, ext=shared.opts.samples_format)
     b64 = base64.b64encode(buffered.getvalue())
     return b64