From b2f1c7301902f9c951316c1cdfa07d3ee31278a6 Mon Sep 17 00:00:00 2001
From: Rafael Fonseca <ramofojabuka@gmail.com>
Date: Sat, 10 Feb 2024 11:42:53 -0300
Subject: Fix process_image to handle transparency correctly

---
 g4f/image.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'g4f')

diff --git a/g4f/image.py b/g4f/image.py
index f0ee0395..29194d6e 100644
--- a/g4f/image.py
+++ b/g4f/image.py
@@ -166,7 +166,7 @@ def process_image(img: Image, new_width: int, new_height: int) -> Image:
     if img.mode != "RGB":
         img.load()
         white = new_image('RGB', img.size, (255, 255, 255))
-        white.paste(img, mask=img.split()[3]) 
+        white.paste(img, mask=img.convert('RGBA').split()[-1])
         return white
     return img
 
-- 
cgit v1.2.3