Spaces:
Sleeping
Sleeping
File size: 2,796 Bytes
9451ca9 |
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 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
from .video_analyzer import VideoAnalyzer
from .prompt_generator import PromptGenerator
from .video_generator import VideoGenerator
from .video_processor import VideoProcessor
import os
class EnhancedAdPipeline:
def __init__(self):
self.analyzer = VideoAnalyzer()
self.prompt_gen = PromptGenerator()
self.video_gen = VideoGenerator()
self.processor = VideoProcessor()
def set_gemini_key(self, key):
self.prompt_gen.set_api_key(key)
def process_videos(self, video1_path, video2_path, target_duration, use_ai, progress=None):
if progress:
progress(0.1, desc="Analyzing first video...")
analysis1 = self.analyzer.analyze_video(video1_path)
if not analysis1:
return {'success': False, 'error': 'Failed to analyze first video'}
if progress:
progress(0.3, desc="Merging videos...")
merged_video = self.processor.merge_videos(video1_path, video2_path, target_duration)
if not merged_video:
return {'success': False, 'error': 'Failed to merge videos'}
current_duration = analysis1['duration']
if video2_path and os.path.exists(video2_path):
analysis2 = self.analyzer.analyze_video(video2_path)
if analysis2:
current_duration += analysis2['duration']
final_video = merged_video
ai_generated = False
if use_ai and current_duration < target_duration and analysis1.get('last_frame') is not None:
if progress:
progress(0.5, desc="Generating AI prompt...")
prompt = self.prompt_gen.generate_video_prompt(analysis1, target_duration)
if progress:
progress(0.6, desc="Generating AI video...")
ai_frames = self.video_gen.generate_video(
analysis1['last_frame'],
prompt,
duration=min(3, target_duration - current_duration)
)
if ai_frames:
if progress:
progress(0.8, desc="Adding AI extension...")
extended_video = self.processor.add_ai_extension(merged_video, ai_frames, target_duration)
if extended_video:
final_video = extended_video
ai_generated = True
if progress:
progress(0.95, desc="Finalizing...")
return {
'success': True,
'output_video': final_video,
'original_duration': analysis1['duration'],
'second_video': video2_path is not None,
'ai_generated': ai_generated
}
|