video_effects_examples.py

#!/usr/bin/env python3
"""
Video Effects Examples

This script demonstrates the comprehensive video effects system that can add
visual effects during video processing using FFmpeg.
"""

import json
import requests
import time
from typing import Dict, Any, List

def test_video_effects_system():
    """Test the video effects system with various effect combinations."""

    print("✨ Video Effects System Test")
    print("=" * 60)
    print()

    # Test scenarios with different effect combinations
    effects_test_scenarios = [
        {
            "name": "Cinematic Ocean Scene",
            "image_prompt": "A dramatic ocean scene with powerful waves at sunset",
            "motion_intensity": "cinematic",
            "effects": ["cinematic", "dramatic", "hdr"],
            "effects_intensity": "strong",
            "duration": 8,
            "description": "Epic ocean with cinematic color grading and HDR effects"
        },
        {
            "name": "Vibrant Forest with Glow",
            "image_prompt": "A magical forest with ancient trees and mystical atmosphere",
            "motion_intensity": "dynamic",
            "effects": ["vibrant", "glow", "warm"],
            "effects_intensity": "moderate",
            "duration": 6,
            "description": "Enchanted forest with vibrant colors and magical glow"
        },
        {
            "name": "Retro City Scene",
            "image_prompt": "A futuristic city street with neon lights and rain",
            "motion_intensity": "dynamic",
            "effects": ["cyberpunk", "neon", "rain"],
            "effects_intensity": "strong",
            "duration": 5,
            "description": "Cyberpunk city with neon effects and rain"
        },
        {
            "name": "Soft Mountain Landscape",
            "image_prompt": "A serene mountain lake reflecting snow-capped peaks",
            "motion_intensity": "subtle",
            "effects": ["soft", "vignette", "film_grain"],
            "effects_intensity": "subtle",
            "duration": 10,
            "description": "Peaceful mountain scene with soft film-like effects"
        },
        {
            "name": "Instagram-Style Garden",
            "image_prompt": "A beautiful garden with blooming flowers and butterflies",
            "motion_intensity": "moderate",
            "effects": ["instagram", "saturation", "brightness"],
            "effects_intensity": "moderate",
            "duration": 5,
            "description": "Social media ready garden with enhanced colors"
        }
    ]

    results = []

    for i, scenario in enumerate(effects_test_scenarios, 1):
        print(f"✨ Test {i}: {scenario['name']}")
        print(f"   Scene: {scenario['image_prompt']}")
        print(f"   Motion: {scenario['motion_intensity']}")
        print(f"   Effects: {scenario['effects']}")
        print(f"   Intensity: {scenario['effects_intensity']}")
        print(f"   Description: {scenario['description']}")
        print()

        # Test with preview mode first
        print("   🔍 Testing with preview mode...")
        preview_result = test_effects_preview(scenario)

        if preview_result:
            print("   ✅ Preview generated successfully")

            # Show what effects would be applied
            if 'effects_info' in preview_result:
                effects_info = preview_result['effects_info']
                print(f"   🎨 Effects to apply: {effects_info.get('effects', [])}")
                print(f"   ⚡ Effects intensity: {effects_info.get('intensity', 'N/A')}")
        else:
            print("   ❌ Preview failed")

        results.append({
            "scenario": scenario,
            "preview_result": preview_result
        })

        print()
        print("-" * 60)
        print()

    # Save results
    timestamp = int(time.time())
    results_file = f"video_effects_test_results_{timestamp}.json"

    with open(results_file, 'w') as f:
        json.dump(results, f, indent=2, ensure_ascii=False)

    print(f"📄 Results saved to: {results_file}")

    return results

def test_effects_preview(scenario: Dict[str, Any]) -> Dict[Any, Any]:
    """Test the effects preview for a specific scenario."""
    try:
        payload = {
            "image_prompt": scenario["image_prompt"],
            "motion_intensity": scenario["motion_intensity"],
            "duration": scenario["duration"],
            "effects": scenario["effects"],
            "effects_intensity": scenario["effects_intensity"],
            "preview_only": True
        }

        response = requests.post("http://localhost:8000/generate-video-from-prompt", json=payload)

        if response.status_code == 200:
            result = response.json()

            # Add effects info to result for testing
            if 'effects' in scenario:
                result['effects_info'] = {
                    'effects': scenario['effects'],
                    'intensity': scenario['effects_intensity']
                }

            return result
        else:
            print(f"   ❌ API request failed: {response.status_code}")
            return None

    except Exception as e:
        print(f"   ❌ Request failed: {e}")
        return None

def demonstrate_effect_categories():
    """Demonstrate different categories of video effects."""

    print("\n🎨 Video Effects Categories")
    print("=" * 60)

    effect_categories = {
        "Color & Brightness": [
            "brightness", "contrast", "saturation", "vibrant",
            "warm", "cool", "vintage", "sepia"
        ],
        "Artistic": [
            "oil_painting", "sketch", "emboss", "sharpen",
            "glow", "neon"
        ],
        "Color Grading": [
            "cinematic", "dramatic", "soft", "hdr",
            "film_grain", "film_look"
        ],
        "Blur & Focus": [
            "blur", "gaussian_blur", "motion_blur",
            "radial_blur", "depth_blur"
        ],
        "Lighting": [
            "vignette", "lens_flare", "light_leak", "god_rays"
        ],
        "Creative": [
            "kaleidoscope", "mirror", "pixelate", "retro", "cyberpunk"
        ],
        "Weather": [
            "rain", "snow", "fog", "sunset"
        ],
        "Professional": [
            "broadcast", "instagram", "tiktok"
        ],
        "Time Effects": [
            "slow_motion", "fast_motion", "zoom_in", "zoom_out"
        ]
    }

    for category, effects in effect_categories.items():
        print(f"\n🎬 {category}:")
        for effect in effects:
            print(f"   • {effect}")

    print(f"\n📊 Total Effects Available: {sum(len(effects) for effects in effect_categories.values())}")

def test_effects_intensity_comparison():
    """Compare the same effects at different intensity levels."""

    print("\n⚡ Effects Intensity Comparison")
    print("=" * 60)

    base_scenario = {
        "image_prompt": "A beautiful sunset over a calm lake with mountains",
        "motion_intensity": "moderate",
        "effects": ["cinematic", "vibrant", "warm"],
        "duration": 5
    }

    intensities = ["subtle", "moderate", "strong"]

    print(f"Base Scene: {base_scenario['image_prompt']}")
    print(f"Effects: {base_scenario['effects']}")
    print()

    for intensity in intensities:
        print(f"🎛️ Testing {intensity.upper()} Intensity:")

        test_payload = {
            **base_scenario,
            "effects_intensity": intensity,
            "preview_only": True
        }

        try:
            response = requests.post("http://localhost:8000/generate-video-from-prompt", json=test_payload)

            if response.status_code == 200:
                result = response.json()
                print(f"   Effects: {test_payload['effects']}")
                print(f"   Intensity: {intensity}")
                print("   ✅ Success")
            else:
                print(f"   ❌ Failed: {response.status_code}")

        except Exception as e:
            print(f"   ❌ Error: {e}")

        print()

def create_effects_combination_examples():
    """Create examples of effective effect combinations."""

    print("\n🌟 Recommended Effects Combinations")
    print("=" * 60)

    combinations = [
        {
            "name": "Professional Cinematic",
            "effects": ["cinematic", "dramatic", "film_grain"],
            "use_case": "High-end professional videos",
            "best_intensity": "strong"
        },
        {
            "name": "Social Media Ready",
            "effects": ["instagram", "saturation", "brightness"],
            "use_case": "Instagram, TikTok content",
            "best_intensity": "moderate"
        },
        {
            "name": "Dreamy Artistic",
            "effects": ["soft", "glow", "vignette"],
            "use_case": "Artistic, romantic content",
            "best_intensity": "moderate"
        },
        {
            "name": "Cyberpunk Future",
            "effects": ["cyberpunk", "neon", "contrast"],
            "use_case": "Sci-fi, futuristic themes",
            "best_intensity": "strong"
        },
        {
            "name": "Vintage Film",
            "effects": ["vintage", "film_grain", "warm"],
            "use_case": "Retro, nostalgic content",
            "best_intensity": "moderate"
        },
        {
            "name": "Nature Documentary",
            "effects": ["hdr", "sharpen", "vibrant"],
            "use_case": "Wildlife, nature videos",
            "best_intensity": "strong"
        },
        {
            "name": "Meditation & Wellness",
            "effects": ["soft", "warm", "vignette"],
            "use_case": "Relaxation, wellness content",
            "best_intensity": "subtle"
        },
        {
            "name": "Action & Energy",
            "effects": ["dramatic", "contrast", "saturation"],
            "use_case": "Sports, action content",
            "best_intensity": "strong"
        }
    ]

    for combo in combinations:
        print(f"🎨 {combo['name']}:")
        print(f"   Effects: {', '.join(combo['effects'])}")
        print(f"   Use Case: {combo['use_case']}")
        print(f"   Best Intensity: {combo['best_intensity']}")
        print()

def test_api_health():
    """Check if the API server is running and healthy."""
    try:
        response = requests.get("http://localhost:8000/health")
        if response.status_code == 200:
            print("✅ API server is healthy")
            return True
        else:
            print(f"⚠️ API server returned status: {response.status_code}")
            return False
    except requests.exceptions.ConnectionError:
        print("❌ Cannot connect to API server at http://localhost:8000")
        print("   Please start the server with: python run_api.py")
        return False
    except Exception as e:
        print(f"❌ API health check failed: {e}")
        return False

def demonstrate_effects_api_usage():
    """Show how to use the effects API."""

    print("\n🔧 Effects API Usage Examples")
    print("=" * 60)

    print("\n📝 CURL Examples:")
    print()

    # Basic effects example
    print("🎨 Basic Effects Example:")
    print("```bash")
    print('curl -X POST "http://localhost:8000/generate-video-from-prompt" \\')
    print('  -H "Content-Type: application/json" \\')
    print('  -d \'{')
    print('    "image_prompt": "A beautiful forest scene with sunlight",')
    print('    "motion_intensity": "dynamic",')
    print('    "effects": ["cinematic", "vibrant", "glow"],')
    print('    "effects_intensity": "moderate",')
    print('    "duration": 6')
    print('  }\'')
    print("```")
    print()

    # Professional effects example
    print("🎬 Professional Effects Example:")
    print("```bash")
    print('curl -X POST "http://localhost:8000/generate-video-from-prompt" \\')
    print('  -H "Content-Type: application/json" \\')
    print('  -d \'{')
    print('    "image_prompt": "An epic mountain landscape during golden hour",')
    print('    "motion_intensity": "cinematic",')
    print('    "effects": ["dramatic", "hdr", "film_grain"],')
    print('    "effects_intensity": "strong",')
    print('    "duration": 10,')
    print('    "voice_type": "onyx",')
    print('    "style": "dramatic",')
    print('    "merge_audio": true')
    print('  }\'')
    print("```")
    print()

    # Social media effects example
    print("📱 Social Media Effects Example:")
    print("```bash")
    print('curl -X POST "http://localhost:8000/generate-video-from-prompt" \\')
    print('  -H "Content-Type: application/json" \\')
    print('  -d \'{')
    print('    "image_prompt": "A trendy urban cafe with modern design",')
    print('    "motion_intensity": "dynamic",')
    print('    "effects": ["instagram", "saturation", "brightness"],')
    print('    "effects_intensity": "moderate",')
    print('    "duration": 5,')
    print('    "ratio": "1024:1024"')
    print('  }\'')
    print("```")
    print()

    print("🐍 Python Examples:")
    print()
    print("```python")
    print("import requests")
    print()
    print("# Professional cinematic video with effects")
    print("response = requests.post('http://localhost:8000/generate-video-from-prompt', json={")
    print("    'image_prompt': 'A majestic waterfall in a lush jungle',")
    print("    'motion_intensity': 'cinematic',")
    print("    'effects': ['cinematic', 'dramatic', 'hdr'],")
    print("    'effects_intensity': 'strong',")
    print("    'duration': 8,")
    print("    'merge_audio': True,")
    print("    'voice_type': 'nova',")
    print("    'style': 'dramatic'")
    print("})")
    print()
    print("result = response.json()")
    print("print(f'Video URL: {result[\"video_url\"]}')")
    print("print(f'Effects Applied: {result.get(\"effects_applied\", [])}')")
    print("```")

if __name__ == "__main__":
    print("🚀 Starting Video Effects System Tests")
    print("=" * 60)

    # Check API health
    if not test_api_health():
        exit(1)

    print()

    # Demonstrate effect categories
    demonstrate_effect_categories()

    # Show recommended combinations
    create_effects_combination_examples()

    # Show API usage
    demonstrate_effects_api_usage()

    # Run effects tests
    print()
    results = test_video_effects_system()

    # Test intensity comparison
    test_effects_intensity_comparison()

    print("\n🎉 Video Effects Testing Complete!")
    print("\n✨ Key Effects Features:")
    print("• 40+ professional video effects available")
    print("• 3 intensity levels: subtle, moderate, strong")
    print("• 9 effect categories: Color, Artistic, Creative, etc.")
    print("• Real-time FFmpeg processing during generation")
    print("• Perfect integration with motion and audio systems")

    print("\n🎨 Effect Categories:")
    print("• Color & Brightness: brightness, contrast, saturation, vibrant")
    print("• Artistic: oil_painting, sketch, emboss, glow, neon")
    print("• Professional: cinematic, dramatic, hdr, film_look")
    print("• Creative: kaleidoscope, mirror, pixelate, cyberpunk")
    print("• Weather: rain, snow, fog, sunset")
    print("• Time Effects: slow_motion, fast_motion, zoom effects")

    print("\n🌟 Recommended Combinations:")
    print("• Professional: ['cinematic', 'dramatic', 'film_grain']")
    print("• Social Media: ['instagram', 'saturation', 'brightness']")
    print("• Artistic: ['soft', 'glow', 'vignette']")
    print("• Cyberpunk: ['cyberpunk', 'neon', 'contrast']")
    print("• Vintage: ['vintage', 'film_grain', 'warm']")

    print("\n⚡ Usage Tips:")
    print("• Start with 1-3 effects for best results")
    print("• Match effects intensity to content style")
    print("• Combine with motion_intensity for full impact")
    print("• Use preview_only=true to test combinations")