mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
29 lines
663 B
Python
29 lines
663 B
Python
|
import os
|
||
|
|
from PIL import Image
|
||
|
|
|
||
|
|
|
||
|
|
def save_thumbnail(
|
||
|
|
image: Image.Image,
|
||
|
|
filename: str,
|
||
|
|
path: str,
|
||
|
|
size: int = 256,
|
||
|
|
) -> str:
|
||
|
|
"""
|
||
|
|
Saves a thumbnail of an image, returning its path.
|
||
|
|
"""
|
||
|
|
base_filename = os.path.splitext(filename)[0]
|
||
|
|
thumbnail_path = os.path.join(path, base_filename + ".webp")
|
||
|
|
|
||
|
|
if os.path.exists(thumbnail_path):
|
||
|
|
return thumbnail_path
|
||
|
|
|
||
|
|
thumbnail_width = size
|
||
|
|
thumbnail_height = round(size * (image.height / image.width))
|
||
|
|
|
||
|
|
image_copy = image.copy()
|
||
|
|
image_copy.thumbnail(size=(thumbnail_width, thumbnail_height))
|
||
|
|
|
||
|
|
image_copy.save(thumbnail_path, "WEBP")
|
||
|
|
|
||
|
|
return thumbnail_path
|