ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-01-01 18:36:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded utility

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-01-09 02:04:37 +01:00
parent 918fcf7c00
commit cc2c3e18ee
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lollms/utilities.py
View File

@ -98,15 +98,19 @@ def encode_image(image_path, max_image_width=-1):
image = Image.open(image_path) image = Image.open(image_path)
width, height = image.size width, height = image.size
# Check and convert image format if needed
if image.format not in ['PNG', 'JPEG', 'GIF', 'WEBP']:
image = image.convert('JPEG')
if max_image_width != -1 and width > max_image_width: if max_image_width != -1 and width > max_image_width:
ratio = max_image_width / width ratio = max_image_width / width
new_width = max_image_width new_width = max_image_width
new_height = int(height * ratio) new_height = int(height * ratio)
f = image.format
image = image.resize((new_width, new_height)) image = image.resize((new_width, new_height))
image.format = f
# Check and convert image format if needed
if image.format not in ['PNG', 'JPEG', 'GIF', 'WEBP']:
image = image.convert('JPEG')
# Save the image to a BytesIO object # Save the image to a BytesIO object
byte_arr = io.BytesIO() byte_arr = io.BytesIO()