Update core/visual_engine.py

core/visual_engine.py

@@ -0,0 +1,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