fix: rgb bug

michaelfromyeg · Jan 6, 2024 · b3d50b0 · b3d50b0
1 parent 955ab1b
commit b3d50b0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/bereal/images.py b/bereal/images.py
@@ -87,6 +87,10 @@ def process_image(
 
     # Save the result in the output folder
     output_path = os.path.join(output_folder, f"combined_{primary_filename}")
+
+    # ensure the photo is jpg ready
+    primary_image = primary_image.convert("RGB")
+
     primary_image.save(output_path, quality=IMAGE_QUALITY)
 
     logger.debug("Combined image saved at %s", output_path)

diff --git a/bereal/videos.py b/bereal/videos.py
@@ -54,6 +54,8 @@ def create_endcard(phone: str, year: str, n_images: int, font_size: int = 50, of
     draw.text((x, y), text, font=font, fill="white")
 
     encard_image_path = os.path.join(CONTENT_PATH, phone, year, "endcard.png")
+
+    img = img.convert("RGB")
     img.save(encard_image_path, quality=IMAGE_QUALITY)
 
     return encard_image_path