|
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" |
|
normal_color = text_color |
|
|
|
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 |
|
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) |