File size: 2,269 Bytes
287c9ca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
# core/visual_engine.py
from PIL import Image, ImageDraw, ImageFont
from moviepy.editor import ImageClip, concatenate_videoclips
import os
class VisualEngine:
def __init__(self, output_dir="generated_media"):
self.output_dir = output_dir
os.makedirs(self.output_dir, exist_ok=True)
try:
# Try to load a common font, adjust path or font name if needed
self.font = ImageFont.truetype("arial.ttf", 24)
except IOError:
self.font = ImageFont.load_default() # Fallback
def create_placeholder_image(self, text_description, filename, size=(1280, 720)):
img = Image.new('RGB', size, color = (73, 109, 137)) # Blueish background
d = ImageDraw.Draw(img)
# Simple text wrapping
lines = []
words = text_description.split()
current_line = ""
for word in words:
if d.textlength(current_line + word + " ", font=self.font) <= size[0] - 40: # 20px padding
current_line += word + " "
else:
lines.append(current_line)
current_line = word + " "
lines.append(current_line)
y_text = (size[1] - len(lines) * (self.font.size + 5)) / 2 # Center text block vertically
for line in lines:
text_width = d.textlength(line, font=self.font)
d.text((size[0]-text_width)/2, y_text, line, fill=(255,255,0), font=self.font)
y_text += self.font.size + 5 # Line spacing
filepath = os.path.join(self.output_dir, filename)
img.save(filepath)
return filepath
def create_video_from_images(self, image_paths, output_filename="final_video.mp4", fps=1, duration_per_image=3):
if not image_paths:
print("No images provided to create video.")
return None
clips = [ImageClip(m).set_duration(duration_per_image) for m in image_paths]
if not clips:
print("Could not create ImageClips.")
return None
video_clip = concatenate_videoclips(clips, method="compose")
output_path = os.path.join(self.output_dir, output_filename)
video_clip.write_videofile(output_path, fps=fps)
return output_path |