24 lines
670 B
Python
24 lines
670 B
Python
|
import base64
|
||
|
|
import io
|
||
|
|
|
||
|
|
from PIL import Image
|
||
|
|
|
||
|
|
|
||
|
|
def process_image(source_bytes: bytes, resize_px: int):
|
||
|
|
image = Image.open(io.BytesIO(source_bytes))
|
||
|
|
width, height = image.size
|
||
|
|
|
||
|
|
if min(width, height) > resize_px:
|
||
|
|
if width < height:
|
||
|
|
new_width = resize_px
|
||
|
|
new_height = int((height / width) * new_width)
|
||
|
|
else:
|
||
|
|
new_height = resize_px
|
||
|
|
new_width = int((width / height) * new_height)
|
||
|
|
image = image.resize((new_width, new_height))
|
||
|
|
|
||
|
|
byte_arr = io.BytesIO()
|
||
|
|
image.save(byte_arr, format='PNG')
|
||
|
|
image_bytes = byte_arr.getvalue()
|
||
|
|
return base64.b64encode(image_bytes).decode('utf-8')
|