| 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) |