fix: convert image to RGB after cropping if needed

openfoodfacts · Dec 25, 2024 · 77ce5a4 · 77ce5a4
1 parent da3fbc5
commit 77ce5a4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/robotoff/app/api.py b/robotoff/app/api.py
@@ -949,6 +949,9 @@ def on_get(self, req: falcon.Request, resp: falcon.Response):
 def image_response(image: Image.Image, resp: falcon.Response) -> None:
     resp.content_type = "image/jpeg"
     fp = io.BytesIO()
+    # JPEG doesn't support RGBA, so we convert to RGB if needed
+    if image.mode != "RGB":
+        image = image.convert("RGB")
     image.save(fp, "JPEG")
     stream_len = fp.tell()
     fp.seek(0)