RedditVideoMakerBot/utils/imagenarator.py

from PIL import Image, ImageDraw,ImageFont 
import textwrap
import re

from rich.progress import track
def draw_multiple_line_text(image, text, font, text_color,padding):
    '''
    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=50)
    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


def imagemaker( theme,reddit_obj,
                txtclr,
                padding=5
                ):
    texts=reddit_obj['thread_post']
    id = re.sub(r"[^\w\s-]", "", reddit_obj["thread_id"])
    font=ImageFont.truetype("arial.ttf", 20)
    size=(500,176) 
    textcolor=txtclr

    for idx,text in track(enumerate(texts),'Rendering Imaging..'):
        image =Image.new('RGBA',size,theme)
        draw = ImageDraw.Draw(image)
        if len(text)>50:
            draw_multiple_line_text(image, text,font, textcolor,padding)
        else:
            image =Image.new('RGBA',size,theme)
            draw = ImageDraw.Draw(image)
            Fontperm= font.getsize(text)
            draw.text((((image.size[0]-Fontperm[0])/2),((image.size[1]-Fontperm[1])/2)),font=font,text=text,align='center')
        image.save(f'assets/temp/{id}/png/img{idx}.png')