o Ch@s~dZddlZddlmZmZddlmZmZddlm Z ddl Z ddl m Z ddl Z e je jde eZGdd d ZdS) u PROFESSIONAL VIDEO GENERATOR (LOCAL & OPEN SOURCE) ==================================================== Features: ✅ 100% FREE & Open Source (runs on your hardware) ✅ State-of-the-art Stable Video Diffusion model ✅ High-quality, coherent video generation ✅ Full privacy and control, no external APIs ✅ Optimized for consumer-grade GPUs (>= 12GB VRAM recommended) ✅ Production-ready local inference engine Author: Professional AI Solutions Architect Date: 2024 N)StableVideoDiffusionPipelineStableDiffusionXLPipeline) load_imageexport_to_video)Image)Path)levelc @s4eZdZdZddZd dedededefd d Zd S) ProfessionalVideoGeneratorz Video generation using a local, self-hosted Stable Video Diffusion model. This approach provides maximum control, privacy, and eliminates API dependencies. c Cstjs tdd|_td|_|jjddd|_t dzDt j dtj d d |_|jtj d tj d d |_|j|j|_t d t d tjdt dt d|jdWdStyy}z t d|d}~ww)z Initializes the video generation pipelines. This is a heavy operation and should only be done once at application startup. It will download models from Hugging Face on the first run. uF❌ CUDA is not available. A GPU is required for this video generator.cudagenerated_videosT)exist_okrz@ M odel Loading: This may take a few minutes on the first run...z-stabilityai/stable-video-diffusion-img2vid-xtfp16) torch_dtypevariantzstabilityai/sdxl-turbou6✅ ProfessionalVideoGenerator initialized (Local SVD)z GPU: zR Models: SDXL-Turbo (for initial image) + Stable Video Diffusion (for animation)z Output directory: /u{❌ Failed to load models. Ensure you have a stable internet connection for the first download and sufficient VRAM. Error: N)torchr is_available RuntimeErrordevicer output_dirmkdirgeneration_countloggerinforfrom_pretrainedfloat16svd_pipeenable_model_cpu_offloadr txt2img_pipetoget_device_name Exceptionerror)selfer%F/var/www/eduai.edurigo.com/image_generation/testing/video_generator.py__init__!s:      z#ProfessionalVideoGenerator.__init__16:9promptstyle aspect_ratioreturnc Cs|jd7_t}td|jtd|dddtd|ztd|d |}d \}}|j|d d ||d jd}td|d } |j| dddddjd} tdt td} d| d} |j | } t | t | ddt|}| jd}tdtd|ddtd |d!d"td#| d$t | | |||d%WSty}z%ddl}d&t |}td'|t|d(|d)WYd}~Sd}~ww)*aK Generates a video from a text prompt. Workflow: 1. Generate a high-quality initial image from the prompt using SDXL Turbo. 2. Animate the image into a video using Stable Video Diffusion. Args: prompt (str): Text description for the video. style (str): Visual style (e.g., 'realistic', 'cinematic', 'animated'). aspect_ratio (str): The desired aspect ratio. Note: SVD is optimized for 16:9. Returns: dict: Result with video path and metadata, or an error message. u 🎬 Generating Video #z Prompt: NPz...z Style: u' ⏳ 1/3: Generating initial image...z, )ii@g)r)num_inference_stepsguidance_scalewidthheightru( ⏳ 2/3: Animating frames with SVD...g{Gz?)decode_chunk_sizemotion_bucket_idnoise_aug_strength num_framesu# ⏳ 3/3: Exporting video file...ivideo_z.mp4)fpsiu ✅ SUCCESSu ⏱️ Time: z.1fsu 📊 Size: z.2fz MBu 💾 Saved: T)success video_pathvideo_filenamer)r*generation_timezVideo generation failed: u ❌ F)r?r")rtimerrrimagesresizerframesintrrstrstatst_sizer! tracebackr" format_exc)r#r)r*r+ start_time full_prompt image_width image_height initial_imageimagerF timestamprAr@rB video_size_mbr$rK error_msgr%r%r&generate_videoOsn        z)ProfessionalVideoGenerator.generate_videoN)r()__name__ __module__ __qualname____doc__r'rHdictrVr%r%r%r&r s .r )rZr diffusersrrdiffusers.utilsrrPILrrCpathlibrlogging basicConfigINFO getLoggerrWrr r%r%r%r&s