mirror of
https://github.com/karl0ss/ai_image_frame_server.git
synced 2025-04-28 19:23:41 +01:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
|
from PIL import Image
|
||
|
|
import os
|
||
|
|
|
||
|
|
def generate_thumbnail(image_path: str, size=(500, 500)) -> str:
|
||
|
|
"""
|
||
|
|
Generates a thumbnail for a given image with a max size of 500x500,
|
||
|
|
and saves it in a 'thumbnails' subdirectory alongside the original.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
image_path (str): Path to the original image.
|
||
|
|
size (tuple): Maximum width and height of the thumbnail.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
str: Path to the thumbnail image.
|
||
|
|
"""
|
||
|
|
image_dir = os.path.dirname(image_path)
|
||
|
|
thumbnail_dir = os.path.join(image_dir, "thumbnails")
|
||
|
|
os.makedirs(thumbnail_dir, exist_ok=True)
|
||
|
|
|
||
|
|
filename = os.path.basename(image_path)
|
||
|
|
thumbnail_path = os.path.join(thumbnail_dir, filename)
|
||
|
|
|
||
|
|
if not os.path.exists(thumbnail_path):
|
||
|
|
try:
|
||
|
|
img = Image.open(image_path)
|
||
|
|
img.thumbnail(size, Image.Resampling.LANCZOS)
|
||
|
|
img.save(thumbnail_path, optimize=True)
|
||
|
|
print(f"Created thumbnail: {thumbnail_path}")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Error creating thumbnail for {image_path}: {e}")
|
||
|
|
else:
|
||
|
|
print(f"Thumbnail already exists: {thumbnail_path}")
|
||
|
|
|
||
|
|
return thumbnail_path
|