summaryrefslogtreecommitdiffstats
path: root/g4f/image.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2024-01-13 18:10:43 +0100
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2024-01-13 18:10:43 +0100
commitceed364cb1ade47a29ccf698074be01b520fc82c (patch)
treefd1ddf447d186ee5e4c8977e5b937e4161392b4c /g4f/image.py
parentImprove FreeChatgpt Provider (diff)
downloadgpt4free-ceed364cb1ade47a29ccf698074be01b520fc82c.tar
gpt4free-ceed364cb1ade47a29ccf698074be01b520fc82c.tar.gz
gpt4free-ceed364cb1ade47a29ccf698074be01b520fc82c.tar.bz2
gpt4free-ceed364cb1ade47a29ccf698074be01b520fc82c.tar.lz
gpt4free-ceed364cb1ade47a29ccf698074be01b520fc82c.tar.xz
gpt4free-ceed364cb1ade47a29ccf698074be01b520fc82c.tar.zst
gpt4free-ceed364cb1ade47a29ccf698074be01b520fc82c.zip
Diffstat (limited to 'g4f/image.py')
-rw-r--r--g4f/image.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/g4f/image.py b/g4f/image.py
index 4a97247e..01664f4e 100644
--- a/g4f/image.py
+++ b/g4f/image.py
@@ -64,7 +64,6 @@ def get_orientation(image: Image.Image) -> int:
def process_image(img: Image.Image, new_width: int, new_height: int) -> Image.Image:
orientation = get_orientation(img)
- new_img = Image.new("RGB", (new_width, new_height), color="#FFFFFF")
if orientation:
if orientation > 4:
img = img.transpose(Image.FLIP_LEFT_RIGHT)
@@ -74,8 +73,8 @@ def process_image(img: Image.Image, new_width: int, new_height: int) -> Image.Im
img = img.transpose(Image.ROTATE_270)
if orientation in [7, 8]:
img = img.transpose(Image.ROTATE_90)
- new_img.paste(img, (0, 0))
- return new_img
+ img.thumbnail((new_width, new_height))
+ return img
def to_base64(image: Image.Image, compression_rate: float) -> str:
output_buffer = BytesIO()