subprocess1 / video.py
sreepathi-ravikumar's picture
Update video.py
dc0af05 verified
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)