src/front_card_image_historic_site.py

from PIL import Image, ImageDraw, ImageFont
from io import BytesIO

# Card image information

# Historic Site Card files
card_frame_path_dict = {
    '橙': 'data/cards/史跡カードフレーム(橙).png',
    '白': 'data/cards/史跡カードフレーム(白).png',
    '紫': 'data/cards/史跡カードフレーム(紫).png',
    '緑': 'data/cards/史跡カードフレーム(緑).png',
    '茶': 'data/cards/史跡カードフレーム(茶).png',
    '赤': 'data/cards/史跡カードフレーム(赤).png',
    '青': 'data/cards/史跡カードフレーム(青).png',
    '黄': 'data/cards/史跡カードフレーム(黄).png',
    '黒': 'data/cards/史跡カードフレーム(黒).png'
}
card_mark_path_dict = {
    '区指定': 'data/cards/史跡カード指定マーク(区指定).png',
    '国指定': 'data/cards/史跡カード指定マーク(国指定).png',
    '市指定': 'data/cards/史跡カード指定マーク(市指定).png',
    '府指定': 'data/cards/史跡カード指定マーク(府指定).png',
    '村指定': 'data/cards/史跡カード指定マーク(村指定).png',
    '町指定': 'data/cards/史跡カード指定マーク(町指定).png',
    '県指定': 'data/cards/史跡カード指定マーク(県指定).png',
    '道指定': 'data/cards/史跡カード指定マーク(道指定).png',
    '都指定': 'data/cards/史跡カード指定マーク(都指定).png'
}

# Pixel position information for each area of the card image
# Picture
PICTURE_LT_XY = (65, 188)
PICTURE_RB_XY = (802, 925)
PICTURE_SIZE = (PICTURE_RB_XY[0] - PICTURE_LT_XY[0], PICTURE_RB_XY[1] - PICTURE_LT_XY[1])

# Title
# Create some margin
TITLE_LT_XY = (65, 45) 
TITLE_RB_XY = (647, 132) # Position to avoid overlapping with the marker insertion point
TITLE_SIZE = (TITLE_RB_XY[0] - TITLE_LT_XY[0], TITLE_RB_XY[1] - TITLE_LT_XY[1])

# Explanation Section Divider
DESCRIPTION_LINE_L_XY = (52, 1024)
DESCRIPTION_LINE_R_XY = (816, 1024)

# Historic Site Type
HS_TYPE_LT_XY = (56, 972)

# Difficulty of Visit
DIFFICULTY_LT_XY = (444, 972)
DIFFICULTY_LT_XY_EN = (424, 972)

# Main Body of Description
DESCRIPTION_LT_XY = (46, 1024)
DESCRIPTION_RB_XY = (810, 1174)
DESCRIPTION_SIZE = (DESCRIPTION_RB_XY[0] - DESCRIPTION_LT_XY[0], DESCRIPTION_RB_XY[1] - DESCRIPTION_LT_XY[1])

# Font information
# Ming Typeface
font_selif_path = 'data/fonts/SourceHanSerif-Bold.otf'
# Gothic Typeface
font_sanselif_path = 'data/fonts/SourceHanSans-Bold.otf'

def crop_center(pil_img, crop_width, crop_height):
    img_width, img_height = pil_img.size
    return pil_img.crop(((img_width - crop_width) // 2,
                         (img_height - crop_height) // 2,
                         (img_width + crop_width) // 2,
                         (img_height + crop_height) // 2))
    
def crop_max_square(pil_img):
    return crop_center(pil_img, min(pil_img.size), min(pil_img.size))
    
def create_historic_site_card_image(img_bytearray, option_dict):

    # Loading and Cropping of Picture
    picture_img = Image.open(img_bytearray)
    picture_img = crop_max_square(picture_img)
    picture_img = picture_img.resize(PICTURE_SIZE)

    # Loading of Card Frame
    card_img = Image.open(card_frame_path_dict[option_dict['色']])

    # Adding a Marker
    if option_dict['マーク'] in card_mark_path_dict:
        mark_image = Image.open(card_mark_path_dict[option_dict['マーク']])
        card_img.paste(mark_image, mask=mark_image)

    # Embedding a Picture in the Card
    card_img.paste(picture_img, PICTURE_LT_XY)

    # Preparing for editting the card image
    card_imgdraw = ImageDraw.Draw(card_img)

    # Adding a Border to the Card
    card_imgdraw.line(( (0, 0), (card_img.size[0], 0) ), fill='black', width=15)
    card_imgdraw.line(( (0, 0), (0, card_img.size[1]) ), fill='black', width=15)
    card_imgdraw.line(( (card_img.size[0], 0),  card_img.size), fill='black', width=15)
    card_imgdraw.line(( (0, card_img.size[1]),  card_img.size), fill='black', width=15)

    # Adding a Divider Line to the Explanation Section
    card_imgdraw.line((DESCRIPTION_LINE_L_XY, DESCRIPTION_LINE_R_XY), fill='black', width=3)

    # Embedding the Title
    # (Support for multiple lines if needed)
    title_font_size = 100
    while True:
        title_font = ImageFont.truetype(font_selif_path, title_font_size)
        title_bbox = card_imgdraw.textbbox(TITLE_LT_XY , option_dict['タイトル'], title_font)
        
        if (title_bbox[2] <= TITLE_RB_XY[0] and title_bbox[3] <= TITLE_RB_XY[1]) or title_font_size <= 30:
            break
        title_font_size -= 1

    card_imgdraw.text((TITLE_LT_XY[0], int((TITLE_LT_XY[1] + TITLE_RB_XY[1]) / 2)), option_dict['タイトル'], fill='black', font=title_font, anchor='lm')

    # Embedding the Historic Site Type
    hs_type_display = f'種類：{option_dict["史跡種類"]}' if option_dict["language"] == 'ja' else f'Type: {option_dict["史跡種類"]}'
    hs_typefont = ImageFont.truetype(font_sanselif_path, 40)
    card_imgdraw.text(HS_TYPE_LT_XY, hs_type_display, fill='black', font=hs_typefont, anchor='lt')

    # Embedding the Difficulty of Visit
    difficulty = int(option_dict['訪問難度'])
    difficulty = 1 if difficulty < 1 else 5 if difficulty > 5 else difficulty
    difficulty_display = ('訪問難度:' if option_dict["language"] == 'ja' else 'Difficulty: ') \
          + '☆' * difficulty + '★' * (5 - difficulty)
    difficulty_font = ImageFont.truetype(font_sanselif_path, 40)
    temp_difficulty_lt_xy = DIFFICULTY_LT_XY if option_dict["language"] == 'ja' else DIFFICULTY_LT_XY_EN
    card_imgdraw.text(temp_difficulty_lt_xy, difficulty_display, fill='black', font=difficulty_font, anchor='lt')

    # Embedding the Description Text
    description_font = ImageFont.truetype(font_sanselif_path, 40)

    description_list = []
    description_length = len(option_dict['説明文'])
    temp_start = 0
    for i in range(description_length):
        temp_end = i
        description_line_bbox = card_imgdraw.textbbox((0, 0), option_dict['説明文'][temp_start:temp_end+1], description_font)
        if description_line_bbox[2] > DESCRIPTION_SIZE[0]:
            description_list.append(option_dict['説明文'][temp_start:temp_end])
            temp_start = i

    description_list.append(option_dict['説明文'][temp_start:])
    description_display = '\n'.join(description_list)

    card_imgdraw.text(DESCRIPTION_LT_XY, description_display, fill='black', font=description_font)

    # Outputting Binary Data
    output_img_bytearray = BytesIO()
    card_img.convert('RGB').save(output_img_bytearray, "JPEG", quality=95)
    output_img_bytearray.seek(0) # Seek to the beginning of the image, otherwise it results in empty data

    return output_img_bytearray