RedditVideoMakerBot/utils/imagenarator.py

import re
import textwrap
import os

from PIL import Image, ImageDraw, ImageFont
from rich.progress import track
from TTS.engine_wrapper import process_text

def draw_multiple_line_text(image, text, font, text_color, padding, wrap=50):
    """
    Draw multiline text over given image
    """
    draw = ImageDraw.Draw(image)
    Fontperm = font.getsize(text)
    image_width, image_height = image.size
    lines = textwrap.wrap(text, width=wrap)
    y = (image_height / 2) - (
        ((Fontperm[1] + (len(lines) * padding) / len(lines)) * len(lines)) / 2
    )
    for line in lines:
        line_width, line_height = font.getsize(line)
        draw.text(((image_width - line_width) / 2, y), line, font=font, fill=text_color)
        y += line_height + padding


# theme=bgcolor,reddit_obj=reddit_object,txtclr=txtcolor
def imagemaker(theme, reddit_obj: dict, txtclr, padding=5):
    """
    Render Images for video
    """
    title = process_text(reddit_obj["thread_title"], False) #TODO if second argument cause any error
    texts = reddit_obj["thread_post"]
    id = re.sub(r"[^\w\s-]", "", reddit_obj["thread_id"])

    tfont = ImageFont.truetype(os.path.join("fonts", "Roboto-Bold.ttf"), 27)  # for title
    font = ImageFont.truetype(
        os.path.join("fonts", "Roboto-Regular.ttf"), 20
    )  # for despcription|comments
    size = (500, 176)

    image = Image.new("RGBA", size, theme)
    draw = ImageDraw.Draw(image)

    # for title
    if len(title) > 40:
        draw_multiple_line_text(image, title, tfont, txtclr, padding, wrap=30)
    else:

        Fontperm = tfont.getsize(title)
        draw.text(
            ((image.size[0] - Fontperm[0]) / 2, (image.size[1] - Fontperm[1]) / 2),
            font=tfont,
            text=title,
        )  # (image.size[1]/2)-(Fontperm[1]/2)

    image.save(f"assets/temp/{id}/png/title.png")

    # for comment|description

    for idx, text in track(enumerate(texts), "Rendering Image"):#, total=len(texts)):

        image = Image.new("RGBA", size, theme)
        draw = ImageDraw.Draw(image)
        text = process_text(text,False)
        if len(text) > 50:
            draw_multiple_line_text(image, text, font, txtclr, padding)

        else:

            Fontperm = font.getsize(text)
            draw.text(
                ((image.size[0] - Fontperm[0]) / 2, (image.size[1] - Fontperm[1]) / 2),
                font=font,
                text=text,
            )  # (image.size[1]/2)-(Fontperm[1]/2)
        image.save(f"assets/temp/{id}/png/img{idx}.png")