From eb7c4b68ff4e586579d0b05bf6db51ea977d8bd5 Mon Sep 17 00:00:00 2001
From: SillyLossy <titan0599@gmail.com>
Date: Sun, 11 Jun 2023 13:42:30 +0300
Subject: [PATCH] #53 Fix buffering of SD output image

---
 server.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/server.py b/server.py
index 852408a..f4b890f 100644
--- a/server.py
+++ b/server.py
@@ -388,9 +388,10 @@ def generate_image(data: dict) -> Image:
 
 
 def image_to_base64(image: Image, quality: int = 75) -> str:
-    buffered = BytesIO()
-    image.save(buffered, format="JPEG", quality=quality)
-    img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    buffer = BytesIO()
+    image.convert("RGB")
+    image.save(buffer, format="JPEG", quality=quality)
+    img_str = base64.b64encode(buffer.getvalue()).decode("utf-8")
     return img_str
 
 # Reads an API key from an already existing file. If that file doesn't exist, create it.