from PIL import Image, ImageDraw, ImageFont
import os

IMAGE_DIR = "/tmp/images"
os.makedirs(IMAGE_DIR, exist_ok=True)

def get_text_size(text, font):
    bbox = font.getbbox(text)
    width = bbox[2] - bbox[0]
    height = bbox[3] - bbox[1]
    return width, height

def create_text_image(text,id,image_olst, image_size=(1280, 720), bg_color="white", text_color="black"):
    image = Image.new("RGB", image_size, color=bg_color)
    draw = ImageDraw.Draw(image)

    font_path = "/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf"
    bold_font_path = "/usr/share/fonts/truetype/liberation/LiberationSans-Bold.ttf"
    max_font_size = 55
    min_font_size = 10

    best_font = None
    best_lines = []

    for font_size in range(max_font_size, min_font_size, -2):
        font = ImageFont.truetype(font_path, font_size)
        words = text.split()
        lines = []
        current_line = ""

        lines_raw = text.splitlines()
        lines = []

        for raw_line in lines_raw:
            words = raw_line.split()
            current_line = ""

            left_padding = 80
            right_padding = 80
            max_text_width = image_size[0] - left_padding - right_padding

            for word in words:
                test_line = f"{current_line} {word}" if current_line else word
                w, _ = get_text_size(test_line, font)
                if w <= max_text_width:
                    current_line = test_line
                else:
                    lines.append(current_line)
                    current_line = word
            if current_line:
                lines.append(current_line)

        total_height = sum(get_text_size(line, font)[1] + 23 for line in lines)
        if total_height <= image_size[1] * 0.95:
            best_font = font
            best_lines = lines
            break

    heading_color = "#0033cc"  # Deep attractive blue
    normal_color = text_color  # Usually black
    
    y = (image_size[1] - total_height) // 2
    for line in best_lines:
        is_heading =  line.strip().startswith("#") and (line.strip().endswith("?") or line.strip().endswith(":"))
        font_to_use = ImageFont.truetype(bold_font_path, best_font.size) if is_heading else best_font
        color_to_use = heading_color if is_heading else normal_color
    
        w, h = get_text_size(line, font_to_use)
        x = 80  # Left padding
        draw.text((x, y), line, font=font_to_use, fill=color_to_use)
        y += h + 23
    image_name="slide"+str(id)+".png"
    image_olst.append(image_name)
    image_path=os.path.join(IMAGE_DIR,image_name)
    image.save(image_path)