# core/visual_engine.py from PIL import Image, ImageDraw, ImageFont from moviepy.editor import (ImageClip, concatenate_videoclips, TextClip, CompositeVideoClip) import moviepy.video.fx.all as vfx import numpy as np import os import openai import requests import io import time # For adding slight delay if API rate limits are hit class VisualEngine: def __init__(self, output_dir="temp_cinegen_media"): self.output_dir = output_dir os.makedirs(self.output_dir, exist_ok=True) self.font_filename = "arial.ttf" self.font_path_in_container = f"/usr/local/share/fonts/truetype/mycustomfonts/{self.font_filename}" self.font_size_pil = 20 # Slightly smaller for placeholder text to fit more self.video_overlay_font_size = 32 self.video_overlay_font_color = 'white' self.video_overlay_font = 'Arial-Bold' # Try specific variant; ensure ImageMagick can find it or use full path try: self.font = ImageFont.truetype(self.font_path_in_container, self.font_size_pil) print(f"Successfully loaded font: {self.font_path_in_container} for placeholders.") except IOError: print(f"Warning: Could not load font from '{self.font_path_in_container}'. Placeholders will use default font.") self.font = ImageFont.load_default() self.font_size_pil = 10 # Default font size estimate self.openai_api_key = None self.USE_AI_IMAGE_GENERATION = False self.dalle_model = "dall-e-3" # DALL-E 3 standard size for highest quality generally. Other options: "1792x1024", "1024x1792" self.image_size_dalle3 = "1792x1024" # Landscape, good for cinematic self.video_frame_size = (1280, 720) # 16:9 aspect ratio for video output def set_openai_api_key(self, api_key): if api_key: self.openai_api_key = api_key self.USE_AI_IMAGE_GENERATION = True print(f"OpenAI API key set. AI Image Generation Enabled with {self.dalle_model}.") else: self.USE_AI_IMAGE_GENERATION = False print("OpenAI API key not provided. AI Image Generation Disabled. Using placeholders.") def _get_text_dimensions(self, text_content, font_obj): # Remains the same if not text_content: return 0, self.font_size_pil try: if hasattr(font_obj, 'getbbox'): bbox = font_obj.getbbox(text_content); width = bbox[2] - bbox[0]; height = bbox[3] - bbox[1] return width, height if height > 0 else self.font_size_pil elif hasattr(font_obj, 'getsize'): width, height = font_obj.getsize(text_content) return width, height if height > 0 else self.font_size_pil else: return int(len(text_content) * self.font_size_pil*0.6), int(self.font_size_pil*1.2 if self.font_size_pil*1.2 > 0 else self.font_size_pil) except Exception: return int(len(text_content) * self.font_size_pil*0.6), int(self.font_size_pil*1.2) def _create_placeholder_image_content(self, text_description, filename, size=(1280, 720)): # Default to video_frame_size img = Image.new('RGB', size, color=(20, 20, 40)) # Darker draw = ImageDraw.Draw(img) padding = 25 max_text_width = size[0] - (2 * padding) lines = [] if not text_description: text_description = "(Placeholder: No prompt provided)" # Simplified text wrapping for placeholder words = text_description.split() current_line = "" for word in words: test_line = current_line + word + " " if self._get_text_dimensions(test_line, self.font)[0] <= max_text_width: current_line = test_line else: if current_line: lines.append(current_line.strip()) current_line = word + " " if current_line: lines.append(current_line.strip()) if not lines: lines.append("(Text too long or unrenderable for placeholder)") _, single_line_height = self._get_text_dimensions("Ay", self.font) if single_line_height == 0: single_line_height = self.font_size_pil + 2 num_lines_to_display = min(len(lines), (size[1] - 2 * padding) // (single_line_height + 2)) # Max lines based on height y_text = padding + (size[1] - 2*padding - num_lines_to_display * (single_line_height + 2)) / 2.0 for i in range(num_lines_to_display): line = lines[i] line_width, _ = self._get_text_dimensions(line, self.font) x_text = (size[0] - line_width) / 2.0 draw.text((x_text, y_text), line, font=self.font, fill=(200, 200, 180)) y_text += single_line_height + 2 # Line spacing if i == 6 and num_lines_to_display > 7: # Show ellipsis if more text draw.text((x_text, y_text), "...", font=self.font, fill=(200, 200, 180)) break filepath = os.path.join(self.output_dir, filename) try: img.save(filepath); return filepath except Exception as e: print(f"Error saving placeholder: {e}"); return None def generate_image_visual(self, image_prompt_text, scene_identifier_filename): filepath = os.path.join(self.output_dir, scene_identifier_filename) if self.USE_AI_IMAGE_GENERATION and self.openai_api_key: max_retries = 2 for attempt in range(max_retries): try: print(f"Attempt {attempt+1}: DALL-E ({self.dalle_model}) for: {image_prompt_text[:120]}...") client = openai.OpenAI(api_key=self.openai_api_key, timeout=60.0) # Timeout for client response = client.images.generate( model=self.dalle_model, prompt=image_prompt_text, n=1, size=self.image_size_dalle3, quality="hd", # Use "hd" for DALL-E 3 for better detail, "standard" for faster/cheaper response_format="url", style="vivid" # "vivid" or "natural" for DALL-E 3 ) image_url = response.data[0].url revised_prompt = getattr(response.data[0], 'revised_prompt', None) if revised_prompt: print(f"DALL-E 3 revised_prompt: {revised_prompt[:100]}...") image_response = requests.get(image_url, timeout=90) # Increased download timeout image_response.raise_for_status() img_data = Image.open(io.BytesIO(image_response.content)) if img_data.mode != 'RGB': img_data = img_data.convert('RGB') img_data.save(filepath) print(f"AI Image (DALL-E) saved: {filepath}") return filepath except openai.RateLimitError as e: print(f"OpenAI Rate Limit Error: {e}. Retrying after delay...") if attempt < max_retries - 1: time.sleep(5 * (attempt + 1)); continue else: print("Max retries reached for RateLimitError."); break except openai.APIError as e: print(f"OpenAI API Error: {e}"); break except requests.exceptions.RequestException as e: print(f"Requests Error (DALL-E image download): {e}"); break except Exception as e: print(f"Generic error (DALL-E image gen): {e}"); break print("DALL-E generation failed after retries. Falling back to placeholder.") return self._create_placeholder_image_content( f"[AI Gen Failed] Prompt: {image_prompt_text[:100]}...", scene_identifier_filename, size=self.video_frame_size ) else: return self._create_placeholder_image_content( image_prompt_text, scene_identifier_filename, size=self.video_frame_size ) def create_video_from_images(self, image_data_list, output_filename="final_video.mp4", fps=24, duration_per_image=4): if not image_data_list: return None print(f"Creating video from {len(image_data_list)} image sets.") processed_clips = [] for i, data in enumerate(image_data_list): img_path, scene_num, key_action = data.get('path'), data.get('scene_num', i+1), data.get('key_action', '') if not (img_path and os.path.exists(img_path)): print(f"Image not found: {img_path}. Skipping."); continue try: pil_img_orig = Image.open(img_path) if pil_img_orig.mode != 'RGB': pil_img_orig = pil_img_orig.convert('RGB') # Resize and letterbox/pillarbox to video_frame_size img_for_frame = pil_img_orig.copy() img_for_frame.thumbnail(self.video_frame_size, Image.Resampling.LANCZOS) canvas = Image.new('RGB', self.video_frame_size, (0,0,0)) x_offset = (self.video_frame_size[0] - img_for_frame.width) // 2 y_offset = (self.video_frame_size[1] - img_for_frame.height) // 2 canvas.paste(img_for_frame, (x_offset, y_offset)) frame_np = np.array(canvas) img_clip = ImageClip(frame_np).set_duration(duration_per_image) # Ken Burns: zoom from 100% to 110% img_clip = img_clip.fx(vfx.resize, lambda t: 1 + 0.1 * (t / duration_per_image)) img_clip = img_clip.set_position('center') # Text Overlay if key_action: overlay_text = f"Scene {scene_num}\n{key_action}" txt_clip = TextClip(overlay_text, fontsize=self.video_overlay_font_size, color=self.video_overlay_font_color, font=self.video_overlay_font, bg_color='rgba(0,0,0,0.7)', method='caption', align='West', size=(self.video_frame_size[0]*0.85, None), kerning=-1, stroke_color='black', stroke_width=0.5 ).set_duration(duration_per_image - 1.0).set_start(0.5) # Show for duration-1s, slight delay txt_clip = txt_clip.set_position(('center', 0.88), relative=True) final_scene_clip = CompositeVideoClip([img_clip, txt_clip], size=self.video_frame_size) else: final_scene_clip = img_clip processed_clips.append(final_scene_clip) except Exception as e: print(f"Error processing clip for {img_path}: {e}. Skipping.") if not processed_clips: print("No clips processed for video."); return None transition_duration = 0.75 # Crossfade duration final_video = concatenate_videoclips(processed_clips, padding=-transition_duration, method="compose") if final_video.duration > transition_duration*2: # Ensure enough duration for fades final_video = final_video.fx(vfx.fadein, transition_duration).fx(vfx.fadeout, transition_duration) output_path = os.path.join(self.output_dir, output_filename) try: final_video.write_videofile(output_path, fps=fps, codec='libx264', preset='medium', audio_codec='aac', temp_audiofile=os.path.join(self.output_dir, f'temp-audio-{os.urandom(4).hex()}.m4a'), remove_temp=True, threads=os.cpu_count() or 2, logger='bar') print(f"Video created: {output_path}"); return output_path except Exception as e: print(f"Error writing video file: {e}"); return None finally: for clip in processed_clips: clip.close() if hasattr(final_video, 'close'): final_video.close()