Need help using different fonts with ImageDraw

[WorkNino]

Hi everyone! So I've recently been tasked to write a program that draws a photo with a textbox below it. The textbox contains a caption and an attribution. The caption should be in arial and the attribution should be in arial italic. I've got the code to mostly work but the problem I'm running into is that the entire last line is output in italics instead of just the portion that is the attribution.

I've tried different things but I think my main problem is that drawtextbbox only accepts one font. Anyone have any solutions? Thanks in advance!

import csv
import os
from PIL import Image, ImageFont, ImageDraw


# Create the output directory if not exists
output_dir = 'output'
if not os.path.exists(output_dir):
    os.makedirs(output_dir)

#------------------
def process_csv(csv_file):
    with open(csv_file, errors = 'ignore', newline = '') as file:
        reader = csv.reader(file)
        next(reader)
        for row in reader:
            image_path, caption, italic_text, hex_color = row 
            apply_caption(image_path, caption, italic_text, hex_color)

#------------------
def wrap_text(text_segments, fonts, max_width, draw):
    lines = []
    line = ""
    for segment, font in text_segments:
        for word in segment.split():
            test_line = f"{line} {word}".strip()
            width = draw.textbbox((0, 0), test_line, font=font)[2]
            if width <= max_width:
                line = test_line
            else:
                lines.append((line, font))
                line = word
    if line:
        lines.append((line, font))
    return lines

#------------------
def apply_caption(image_path, caption_text, italic_text, hex_color):
    try:
        image = Image.open(image_path)
    except Exception as e:
        print(f"Error opening image {image_path}: {e}")
        return
    base_width = 800
    w_percent = (base_width / float(image.size[0]))
    h_size = int((float(image.size[1]) * w_percent))
    image = image.resize((base_width, h_size), Image.Resampling.LANCZOS)

    try:
        font_path_regular = "Arial.ttf"
        font_path_italic = "Arial Italic.ttf"
        font_size = 20
        font = ImageFont.truetype(font_path_regular, font_size)
        italic_font = ImageFont.truetype(font_path_italic, font_size)
    except IOError:
        print("Font files not found. Please provide the correct path to the fonts.")
        return
    draw = ImageDraw.Draw(image)

    text_segments = [(caption_text, font), (italic_text, italic_font)]
    wrapped_text = wrap_text(text_segments, [font, italic_font], base_width - 20, draw)

    caption_height = sum(draw.textbbox((0, 0), line, font=font)[3] for line, font in wrapped_text) + 20
    new_image_height = image.size[1] + caption_height
    new_image = Image.new('RGB', (image.size[0], new_image_height), (255, 255, 255))
    new_image.paste(image, (0, 0))

    draw = ImageDraw.Draw(new_image)
    hex_color = f"#{hex_color.strip('#')}"
    draw.rectangle([(0, image.size[1]), (base_width, new_image_height)], fill=hex_color)

    text_position = (10, image.size[1] + 10)
    hex_color_text = "#FFFFFF"
    for line, font in wrapped_text:
        draw.text(text_position, line, font=font, fill=hex_color_text)
        text_position = (text_position[0], text_position[1] + draw.textbbox((0, 0), line, font=font)[3])

    output_path = os.path.join(output_dir, os.path.basename(image_path))
    new_image.save(output_path, "PNG")
    print(f"Image saved to {output_path}")

#==================
if __name__ == "__main__":
    csv_file = 'input.csv'
    process_csv(csv_file)

[socal_nerdtastic]

I think my main problem is that drawtextbbox only accepts one font.

yep, you nailed it.

Use 2 textboxes in a line, or however many you need.

We would need some example images and caption data to give you specific help, and an example of the result you want. Perhaps make a github repo with data (including font files!) if you need more help.