From 5eb4c4e62a4222ba094847183cd0ff4c67c49f7b Mon Sep 17 00:00:00 2001 From: Masen Furer Date: Mon, 20 Nov 2023 16:17:09 -0800 Subject: [PATCH] Serialize PIL Image in its format (#2201) --- reflex/components/media/image.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/reflex/components/media/image.py b/reflex/components/media/image.py index f1ff34bfd..62e71df2a 100644 --- a/reflex/components/media/image.py +++ b/reflex/components/media/image.py @@ -95,10 +95,11 @@ try: The serialized image. """ buff = io.BytesIO() - image.save(buff, format="PNG") + image.save(buff, format=getattr(image, "format", None) or "PNG") image_bytes = buff.getvalue() base64_image = base64.b64encode(image_bytes).decode("utf-8") - return f"data:image/png;base64,{base64_image}" + mime_type = getattr(image, "get_format_mimetype", lambda: "image/png")() + return f"data:{mime_type};base64,{base64_image}" except ImportError: pass