Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import os | |
| import warnings | |
| warnings.filterwarnings('ignore') | |
| from src.pipeline import EnhancedAdPipeline | |
| from config.settings import UI_CONFIG | |
| def create_interface(): | |
| pipeline = EnhancedAdPipeline() | |
| def process_videos_ui(video1, video2, target_duration, use_ai, gemini_key, progress=gr.Progress()): | |
| if video1 is None: | |
| return None, "❌ Please upload at least the first video" | |
| try: | |
| if gemini_key and gemini_key.strip(): | |
| pipeline.set_gemini_key(gemini_key.strip()) | |
| progress(0.1, desc="🔍 Analyzing videos...") | |
| result = pipeline.process_videos(video1, video2, target_duration, use_ai, progress) | |
| if result['success']: | |
| progress(1.0, desc="🎉 Complete!") | |
| status = f"✅ Video processing successful!\n" | |
| status += f"📊 Original duration: {result.get('original_duration', 0):.1f}s\n" | |
| if result.get('second_video'): | |
| status += f"📊 Second video included\n" | |
| if result.get('ai_generated'): | |
| status += "🤖 AI-generated extension added\n" | |
| status += f"🎯 Final duration: ~{target_duration}s" | |
| return result['output_video'], status | |
| else: | |
| return None, f"❌ Error: {result.get('error', 'Unknown error')}" | |
| except Exception as e: | |
| return None, f"❌ Error: {str(e)}" | |
| with gr.Blocks(**UI_CONFIG) as app: | |
| gr.Markdown(""" | |
| 🎬 Enhanced AI Ad Completion Tool | |
| **Transform your videos into professional commercials using AI!** | |
| ✨ **Features:** Dual video upload • AI video generation • Gemini-powered prompts • Professional processing | |
| """) | |
| with gr.Row(): | |
| with gr.Column(scale=1): | |
| gr.Markdown("### 📹 Video Inputs") | |
| video1_input = gr.Video( | |
| label="🎬 Primary Video (Required)", | |
| format="mp4" | |
| ) | |
| video2_input = gr.Video( | |
| label="🎬 Secondary Video (Optional)", | |
| format="mp4" | |
| ) | |
| gr.Markdown("### ⚙️ Settings") | |
| duration_slider = gr.Slider( | |
| minimum=10, | |
| maximum=45, | |
| value=20, | |
| step=1, | |
| label="🎯 Target Duration (seconds)" | |
| ) | |
| ai_toggle = gr.Checkbox( | |
| value=True, | |
| label="🤖 Use AI Video Generation", | |
| info="Generate new content using AI (requires GPU)" | |
| ) | |
| gemini_key_input = gr.Textbox( | |
| label="🔑 Gemini API Key (Optional)", | |
| placeholder="Enter your Gemini API key for better prompts", | |
| type="password" | |
| ) | |
| process_btn = gr.Button( | |
| "🚀 Create Enhanced Ad", | |
| variant="primary", | |
| size="lg" | |
| ) | |
| gr.Markdown(""" | |
| ### 📋 How to Use: | |
| 1. **Upload primary video** (your main content) | |
| 2. **Optional**: Upload second video to combine | |
| 3. **Set target duration** for final output | |
| 4. **Optional**: Add Gemini API key for better prompts | |
| 5. **Enable AI generation** for new content | |
| 6. **Click "Create Enhanced Ad"** | |
| **Get Gemini API Key:** [Google AI Studio](https://makersuite.google.com/app/apikey) (Free) | |
| """) | |
| with gr.Column(scale=2): | |
| gr.Markdown("### 🎉 Output") | |
| video_output = gr.Video( | |
| label="Enhanced Ad Video", | |
| format="mp4" | |
| ) | |
| status_output = gr.Textbox( | |
| label="📋 Processing Status", | |
| interactive=False, | |
| lines=6 | |
| ) | |
| gr.Markdown(""" | |
| ### 🎨 What This Tool Does: | |
| **🔍 Video Analysis:** | |
| - Extracts key frames and visual elements | |
| - Transcribes audio content using Whisper | |
| - Analyzes color palette and scene classification | |
| **🤖 AI Enhancement:** | |
| - Uses Gemini API for intelligent prompt generation | |
| - Generates new video content with Stable Video Diffusion | |
| - Applies professional color grading and effects | |
| **🎬 Smart Processing:** | |
| - Combines multiple videos with smooth transitions | |
| - Extends to target duration intelligently | |
| - Preserves audio quality throughout | |
| """) | |
| process_btn.click( | |
| fn=process_videos_ui, | |
| inputs=[video1_input, video2_input, duration_slider, ai_toggle, gemini_key_input], | |
| outputs=[video_output, status_output] | |
| ) | |
| return app | |
| if __name__ == "__main__": | |
| app = create_interface() | |
| app.launch(server_name="0.0.0.0", server_port=7860, share=True) | |