zfff

video_generator.py

@@ -9,37 +9,26 @@ def generate_lyric_frame(lyric, image_size):
         image_size = (800, 600) # Example default size, you can change this
 
     # Ensure image_size is a tuple
-    image_size = tuple(image_size)
+    if not isinstance(image_size, tuple):
+        image_size = tuple(image_size)
 
-    # Create an image that is larger by a factor (e.g. 2)
-    factor = 2
-    large_image_size = (image_size[0] * factor, image_size[1] * factor)
-    image = Image.new('RGB', large_image_size, color='black')
+    image = Image.new('RGB', image_size, color='black')
     draw = ImageDraw.Draw(image)
-    font = ImageFont.load_default()
-
-    # Wrap text
-    max_width = large_image_size[0] - 40 * factor
-    wrapped_text = textwrap.fill(lyric, width=40 * factor)
-    lines = wrapped_text.split('\n')
 
-    # Calculate text width and height
-    text_width = max(draw.textlength(line, font=font) for line in lines)
-    text_height = sum([font.getmask(line).getbbox()[3] for line in lines if font.getmask(line).getbbox()])
+    # Define font and text alignment
+    font = ImageFont.load_default()
+    text_width = draw.textlength(lyric, font=font) # Getting text width
 
-    # Calculate starting coordinates to center the text block
-    text_x = (large_image_size[0] - text_width) / 2
-    text_y = (large_image_size[1] - text_height) / 2
+    # 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
 
-    # 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
+    text_x = (image_size[0] - text_width) / 2
+    text_y = (image_size[1] - text_height) / 2
 
-    # Resize the image back to the original size
-    resized_image = image.resize(image_size)
-    return resized_image
+    # Draw the text on the image
+    draw.text((text_x, text_y), lyric, font=font, fill='white')
+    return image
 
 
 def generate_video(lyrics_with_timing, image_size=(800, 600), frame_rate=30):