Bladeren bron

Serialize PIL Image in its format (#2201)

Masen Furer 1 jaar geleden
bovenliggende
commit
5eb4c4e62a
1 gewijzigde bestanden met toevoegingen van 3 en 2 verwijderingen
  1. 3 2
      reflex/components/media/image.py

+ 3 - 2
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