327 lines
13 KiB
Python
327 lines
13 KiB
Python
import logging
|
|
import os
|
|
from typing import List, Dict
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
from app.server.utils.minio_client import oss_get_image, minio_client
|
|
from app.server.utils.minio_config import MINIO_LC_DATA_PATH
|
|
from app.config import settings
|
|
|
|
logger = logging.getLogger(__name__)
|
|
# 9个 341x341 左右的单元格 (ALL_9_CELLS)
|
|
# 布局顺序: 从上到下，从左到右 (1 -> 9)
|
|
ALL_9_CELLS = [
|
|
# Top Row (Y=0, H=341)
|
|
(0, 0, 341, 341), # 1. Top-Left (341x341)
|
|
(341, 0, 341, 341), # 2. Top-Middle (341x341)
|
|
(682, 0, 342, 341), # 3. Top-Right (342x341)
|
|
# Middle Row (Y=341, H=341)
|
|
(0, 341, 341, 341), # 4. Mid-Left (341x341)
|
|
(341, 341, 341, 341), # 5. Center (341x341)
|
|
(682, 341, 342, 341), # 6. Mid-Right (342x341)
|
|
# Bottom Row (Y=682, H=342)
|
|
(0, 682, 341, 342), # 7. Bottom-Left (341x342)
|
|
(341, 682, 341, 342), # 8. Bottom-Middle (341x342)
|
|
(682, 682, 342, 342) # 9. Bottom-Right (342x342)
|
|
]
|
|
|
|
|
|
def merge_images_to_square(outfit_items: List[Dict[str, str]], max_len=9, add_text=True):
|
|
"""
|
|
Loads up to 4 images from the given paths, resizes them while maintaining
|
|
aspect ratio, and merges them onto a 1024x1024 white background JPG.
|
|
|
|
The layout depends on the number of images:
|
|
1: Center the single image on the 1024x1024 canvas.
|
|
2: Place side-by-side, each scaled to fit a 512x1024 half.
|
|
3: Place in top-left (512x512), top-right (512x512), and bottom-left (512x512).
|
|
4: Place in all four 512x512 quadrants.
|
|
|
|
Args:
|
|
outfit_items: A list of item metadata (max length 9).
|
|
|
|
Returns:
|
|
The file path of the temporary merged JPG image.
|
|
"""
|
|
|
|
# Define the final canvas size
|
|
CANVAS_SIZE = 1024
|
|
|
|
# 定义每个 item 的外边距
|
|
MARGIN = 5 # 5像素外边距
|
|
|
|
# 1. Create the final white canvas
|
|
# Using 'RGB' mode for JPG output
|
|
canvas = Image.new('RGB', (CANVAS_SIZE, CANVAS_SIZE), 'white')
|
|
draw = ImageDraw.Draw(canvas)
|
|
font = ImageFont.load_default()
|
|
|
|
# 2. Define the quadrants/target areas (x, y, w, h)
|
|
# The positions are based on a 512x512 quadrant size
|
|
quadrants = {
|
|
1: [(0, 0, CANVAS_SIZE, CANVAS_SIZE)], # Single full-size placement
|
|
2: [(0, 0, 512, CANVAS_SIZE), (512, 0, 512, CANVAS_SIZE)], # Left, Right
|
|
3: [(0, 0, 512, 512), (512, 0, 512, 512), (0, 512, 512, 512)], # Top-Left, Top-Right, Bottom-Left
|
|
4: [(0, 0, 512, 512), (512, 0, 512, 512), (0, 512, 512, 512), (512, 512, 512, 512)], # All Four
|
|
5: ALL_9_CELLS[:5], # 布局前5个单元格 (1-5)
|
|
6: ALL_9_CELLS[:6], # 布局前6个单元格 (1-6)
|
|
7: ALL_9_CELLS[:7], # 布局前7个单元格 (1-7)
|
|
8: ALL_9_CELLS[:8], # 布局前8个单元格 (1-8)
|
|
9: ALL_9_CELLS[:9] # 布局全部9个单元格 (1-9)
|
|
}
|
|
|
|
# 3. Load and Filter Images
|
|
valid_images = []
|
|
image_paths = [item['image_path'] for item in outfit_items]
|
|
for path in image_paths:
|
|
try:
|
|
# We use Image.open() and convert to 'RGB' to handle potential transparency (RGBA)
|
|
# and ensure compatibility with the final 'RGB' canvas and JPG output.
|
|
if settings.LOCAL == 1:
|
|
image_file_path = os.path.join(settings.LOCAL_IMAGE_DIR, path)
|
|
img = Image.open(image_file_path).convert('RGB')
|
|
else:
|
|
img = oss_get_image(oss_client=minio_client, path=f"{MINIO_LC_DATA_PATH}/{path}", data_type="PIL").convert('RGB')
|
|
# img = Image.open(path).convert('RGB')
|
|
valid_images.append(img)
|
|
except Exception as e:
|
|
logger.error(f"Error loading image {path}. Skipping: {e}")
|
|
|
|
num_images = len(valid_images)
|
|
|
|
if num_images == 0:
|
|
raise ValueError("No valid images were loaded.")
|
|
|
|
if num_images > max_len:
|
|
raise ValueError(f"Valid item number {num_images} exceed max limit {max_len}")
|
|
|
|
# Get the correct list of target areas based on the number of valid images
|
|
target_areas = quadrants.get(num_images, [])
|
|
|
|
if not target_areas:
|
|
raise ValueError(f"No layout defined for {num_images} images.")
|
|
|
|
# 4. Resize and Paste
|
|
for i, (img, item) in enumerate(zip(valid_images, outfit_items)):
|
|
item_id = item['item_id']
|
|
category = item['category']
|
|
if i >= len(target_areas):
|
|
# This should not happen if num_images <= 4
|
|
break
|
|
|
|
# 原始目标区域 (x_start, y_start, width, height)
|
|
orig_x_start, orig_y_start, orig_w, orig_h = target_areas[i]
|
|
|
|
# 📢 应用边距：实际用于图像和文本的区域
|
|
# 新的起始点：向内移动 MARGIN
|
|
x_start = orig_x_start + MARGIN
|
|
y_start = orig_y_start + MARGIN
|
|
|
|
# 新的宽高：减去两倍的 MARGIN (左右/上下)
|
|
target_w = orig_w - 2 * MARGIN
|
|
target_h = orig_h - 2 * MARGIN
|
|
|
|
# --- 图像缩放与居中 ---
|
|
|
|
# Calculate new size while maintaining aspect ratio
|
|
original_w, original_h = img.size
|
|
|
|
# Calculate the ratio needed to fit within the *带边距的* 目标区域
|
|
ratio_w = target_w / original_w
|
|
ratio_h = target_h / original_h
|
|
|
|
# Use the *smaller* of the two ratios to ensure the image fits entirely
|
|
resize_ratio = min(ratio_w, ratio_h)
|
|
|
|
# Calculate the new dimensions
|
|
new_w = int(original_w * resize_ratio)
|
|
new_h = int(original_h * resize_ratio)
|
|
|
|
# Resize the image.
|
|
resized_img = img.resize((new_w, new_h), Image.Resampling.LANCZOS)
|
|
|
|
# Calculate the paste position to center the resized image within its target area
|
|
# Center X: (Target Width - New Width) / 2 + X Start (带边距的 X_start)
|
|
paste_x = (target_w - new_w) // 2 + x_start
|
|
|
|
# 预留文本高度 ( TEXT_RESERVE_HEIGHT )
|
|
TEXT_RESERVE_HEIGHT = 30
|
|
|
|
# Center Y: (Target Height - New Height - 预留文本高度) / 2 + Y Start (带边距的 Y_start)
|
|
paste_y = (target_h - new_h - TEXT_RESERVE_HEIGHT) // 2 + y_start
|
|
# 确保图片顶部不超出目标区域的 Y_start
|
|
paste_y = max(paste_y, y_start)
|
|
|
|
# Paste the resized image onto the canvas
|
|
canvas.paste(resized_img, (paste_x, paste_y))
|
|
|
|
# --- 文本居中与定位 ---
|
|
|
|
full_text = f"ID: {item_id}, Category: {category}"
|
|
if add_text:
|
|
try:
|
|
# 推荐使用：计算文本的实际尺寸 (width, height)
|
|
bbox = draw.textbbox((0, 0), full_text, font=font)
|
|
text_w = bbox[2] - bbox[0]
|
|
text_h = bbox[3] - bbox[1]
|
|
except AttributeError:
|
|
# 兼容旧版本 Pillow
|
|
text_w, text_h = draw.textsize(full_text, font=font)
|
|
|
|
# 计算 X 轴起始位置：使其在 *带边距的目标区域* (target_w) 中居中
|
|
text_x_center = x_start + target_w // 2
|
|
text_x_start = text_x_center - text_w // 2
|
|
|
|
# 计算 Y 轴起始位置：将其放在 *带边距的目标区域* 的底部
|
|
# (带边距的起始Y + 带边距的高度 - 文本行的高度)
|
|
# 📢 在带边距的目标区域底部再减去 5 像素作为与底部的边距
|
|
text_y_start = y_start + target_h - text_h
|
|
|
|
draw.text((text_x_start, text_y_start),
|
|
full_text,
|
|
fill='black',
|
|
font=font)
|
|
|
|
# Save as a high-quality JPG (quality=90 is a good balance)
|
|
# canvas.save(output_path, 'JPEG', quality=90)
|
|
|
|
return canvas
|
|
|
|
# def merge_images_to_square(outfit_items: List[Dict[str, str]], max_len=9, add_text=True):
|
|
# """
|
|
# Loads up to 4 images from the given paths, resizes them while maintaining
|
|
# aspect ratio, and merges them onto a 1024x1024 white background JPG.
|
|
#
|
|
# The layout depends on the number of images:
|
|
# 1: Center the single image on the 1024x1024 canvas.
|
|
# 2: Place side-by-side, each scaled to fit a 512x1024 half.
|
|
# 3: Place in top-left (512x512), top-right (512x512), and bottom-left (512x512).
|
|
# 4: Place in all four 512x512 quadrants.
|
|
#
|
|
# Args:
|
|
# outfit_items: A list of item metadata (max length 9).
|
|
#
|
|
# Returns:
|
|
# The file path of the temporary merged JPG image.
|
|
# """
|
|
#
|
|
# # Define the final canvas size
|
|
# CANVAS_SIZE = 1024
|
|
#
|
|
# # 1. Create the final white canvas
|
|
# # Using 'RGB' mode for JPG output
|
|
# canvas = Image.new('RGB', (CANVAS_SIZE, CANVAS_SIZE), 'white')
|
|
# draw = ImageDraw.Draw(canvas)
|
|
# font = ImageFont.load_default()
|
|
#
|
|
# # 2. Define the quadrants/target areas (x, y, w, h)
|
|
# # The positions are based on a 512x512 quadrant size
|
|
# quadrants = {
|
|
# 1: [(0, 0, CANVAS_SIZE, CANVAS_SIZE)], # Single full-size placement
|
|
# 2: [(0, 0, 512, CANVAS_SIZE), (512, 0, 512, CANVAS_SIZE)], # Left, Right
|
|
# 3: [(0, 0, 512, 512), (512, 0, 512, 512), (0, 512, 512, 512)], # Top-Left, Top-Right, Bottom-Left
|
|
# 4: [(0, 0, 512, 512), (512, 0, 512, 512), (0, 512, 512, 512), (512, 512, 512, 512)], # All Four
|
|
# 5: ALL_9_CELLS[:5], # 布局前5个单元格 (1-5)
|
|
# 6: ALL_9_CELLS[:6], # 布局前6个单元格 (1-6)
|
|
# 7: ALL_9_CELLS[:7], # 布局前7个单元格 (1-7)
|
|
# 8: ALL_9_CELLS[:8], # 布局前8个单元格 (1-8)
|
|
# 9: ALL_9_CELLS[:9] # 布局全部9个单元格 (1-9)
|
|
# }
|
|
#
|
|
# # 3. Load and Filter Images
|
|
# valid_images = []
|
|
# image_paths = [item['image_path'] for item in outfit_items]
|
|
# for path in image_paths:
|
|
# try:
|
|
# # We use Image.open() and convert to 'RGB' to handle potential transparency (RGBA)
|
|
# # and ensure compatibility with the final 'RGB' canvas and JPG output.
|
|
# img = oss_get_image(oss_client=minio_client, path=f"{MINIO_LC_DATA_PATH}/{path}", data_type="PIL").convert('RGB')
|
|
# # img = Image.open(path).convert('RGB')
|
|
# valid_images.append(img)
|
|
# except Exception as e:
|
|
# logger.error(f"Error loading image {path}. Skipping: {e}")
|
|
#
|
|
# num_images = len(valid_images)
|
|
#
|
|
# if num_images == 0:
|
|
# raise ValueError("No valid images were loaded.")
|
|
#
|
|
# if num_images > max_len:
|
|
# raise ValueError(f"Valid item number {num_images} exceed max limit {max_len}")
|
|
#
|
|
# # Get the correct list of target areas based on the number of valid images
|
|
# target_areas = quadrants.get(num_images, [])
|
|
#
|
|
# # 4. Resize and Paste
|
|
# for i, (img, item) in enumerate(zip(valid_images, outfit_items)):
|
|
# item_id = item['item_id']
|
|
# category = item['category']
|
|
# if i >= len(target_areas):
|
|
# # This should not happen if num_images <= 4
|
|
# break
|
|
#
|
|
# # Target area dimensions (x_start, y_start, width, height)
|
|
# x_start, y_start, target_w, target_h = target_areas[i]
|
|
#
|
|
# # Calculate new size while maintaining aspect ratio
|
|
# original_w, original_h = img.size
|
|
#
|
|
# # Calculate the ratio needed to fit within the target area
|
|
# ratio_w = target_w / original_w
|
|
# ratio_h = target_h / original_h
|
|
#
|
|
# # Use the *smaller* of the two ratios to ensure the image fits entirely
|
|
# resize_ratio = min(ratio_w, ratio_h)
|
|
#
|
|
# # Calculate the new dimensions
|
|
# new_w = int(original_w * resize_ratio)
|
|
# new_h = int(original_h * resize_ratio)
|
|
#
|
|
# # Resize the image. Image.Resampling.LANCZOS provides high-quality scaling.
|
|
# # Pillow documentation recommends ANTIALIAS or BICUBIC for downscaling,
|
|
# # but LANCZOS is a good general high-quality filter.
|
|
# # Note: In Pillow versions > 9.0.0, Image.LANCZOS is now Image.Resampling.LANCZOS
|
|
# resized_img = img.resize((new_w, new_h), Image.Resampling.LANCZOS)
|
|
#
|
|
# # Calculate the paste position to center the resized image within its target area
|
|
# # Center X: (Target Width - New Width) / 2 + X Start
|
|
# paste_x = (target_w - new_w) // 2 + x_start
|
|
# # Center Y: (Target Height - New Height) / 2 + Y Start
|
|
# # paste_y = (target_h - new_h) // 2 + y_start
|
|
#
|
|
# TEXT_RESERVE_HEIGHT = 30
|
|
# paste_y = (target_h - new_h - TEXT_RESERVE_HEIGHT) // 2 + y_start
|
|
# paste_y = max(paste_y, y_start)
|
|
#
|
|
# # Paste the resized image onto the canvas
|
|
# canvas.paste(resized_img, (paste_x, paste_y))
|
|
#
|
|
# full_text = f"ID: {item_id}, Category: {category}"
|
|
# try:
|
|
# # 推荐使用：计算文本的实际尺寸 (width, height)
|
|
# bbox = draw.textbbox((0, 0), full_text, font=font)
|
|
# text_w = bbox[2] - bbox[0]
|
|
# text_h = bbox[3] - bbox[1]
|
|
# except AttributeError:
|
|
# # 兼容旧版本 Pillow
|
|
# text_w, text_h = draw.textsize(full_text, font=font)
|
|
#
|
|
# # 计算 X 轴起始位置：使其在目标区域 (target_w) 中居中
|
|
# text_x_center = x_start + target_w // 2
|
|
# text_x_start = text_x_center - text_w // 2
|
|
#
|
|
# # 计算 Y 轴起始位置：将其放在目标区域的底部
|
|
# # (目标区域的起始Y + 目标区域的高度 - 文本行的高度)
|
|
# text_y_start = y_start + target_h - text_h - 5 # 减去 5 像素作为边距
|
|
#
|
|
# # 3. 绘制合并后的文本
|
|
# if add_text:
|
|
# draw.text((text_x_start, text_y_start),
|
|
# full_text,
|
|
# fill='black',
|
|
# font=font)
|
|
#
|
|
# # Save as a high-quality JPG (quality=90 is a good balance)
|
|
# # canvas.save(output_path, 'JPEG', quality=90)
|
|
#
|
|
# return canvas
|