zfff

video_generator.py

@@ -2,31 +2,41 @@ from PIL import Image, ImageDraw, ImageFont
 import cv2
 import numpy as np
 
+import textwrap
+
 def generate_lyric_frame(lyric, image_size):
     # If image_size is None, provide a default value
     if image_size is None:
         image_size = (800, 600) # Example default size, you can change this
 
     # Ensure image_size is a tuple
-    if not isinstance(image_size, tuple):
-        image_size = tuple(image_size)
+    image_size = tuple(image_size)
 
     image = Image.new('RGB', image_size, color='black')
     draw = ImageDraw.Draw(image)
 
-    # Define font and text alignment
+    # Use PIL's default font
     font = ImageFont.load_default()
-    text_width = draw.textlength(lyric, font=font) # Getting text width
 
-    # Getting text height, with a check for empty text
-    text_bbox = font.getmask(lyric).getbbox()
-    text_height = text_bbox[3] if text_bbox else 0
+    # Wrap text
+    max_width = image_size[0] - 40  # Keeping 20 pixels margin on each side
+    wrapped_text = textwrap.fill(lyric, width=40)
+    lines = wrapped_text.split('\n')
 
+    # Calculate text width and height
+    text_width = max(draw.textlength(line, font=font) for line in lines)  # Corrected line here
+    text_height = sum([font.getmask(line).getbbox()[3] for line in lines if font.getmask(line).getbbox()]) # Corrected line here
+    
+    # Calculate starting coordinates to center the text block
     text_x = (image_size[0] - text_width) / 2
     text_y = (image_size[1] - text_height) / 2
 
-    # Draw the text on the image
-    draw.text((text_x, text_y), lyric, font=font, fill='white')
+    # Draw the wrapped text on the image
+    y = text_y
+    for line in lines:
+        draw.text((text_x, y), line, font=font, fill='white')
+        y += font.getmask(line).getbbox()[3] if font.getmask(line).getbbox() else 0  # Corrected line here
+
     return image
 
 def generate_video(lyrics_with_timing, image_size=(800, 600), frame_rate=30):