audio_video_examples.py

#!/usr/bin/env python3
"""
Audio + Video Generation Examples

This script demonstrates the complete multimedia generation pipeline:
1. Generate image from text prompt (DALL-E)
2. Create video animation from image (Runway)
3. Generate audio narration (OpenAI TTS)

The result is a complete multimedia experience with synchronized content.
"""

import json
import requests
import time
from pathlib import Path

API_BASE_URL = "http://localhost:8000"

def test_prompt_to_multimedia():
    """Test complete prompt-to-multimedia generation."""
    print("🎨➡️🎬🎙️ Complete Multimedia Generation Test")
    print("=" * 60)
    print()

    examples = [
        {
            "name": "Peaceful Nature Scene",
            "image_prompt": "A serene mountain lake surrounded by pine trees at golden hour",
            "video_prompt": "Gentle ripples on the water surface with soft morning light filtering through trees",
            "voice_type": "nova",
            "style": "descriptive"
        },
        {
            "name": "Ocean Meditation",
            "image_prompt": "A peaceful beach with gentle waves and sunset colors in the sky",
            "video_prompt": "Soft waves washing onto shore with seabirds flying in the distance",
            "voice_type": "shimmer",
            "style": "meditative"
        },
        {
            "name": "Forest Adventure",
            "image_prompt": "A mystical forest path with sunbeams shining through tall trees",
            "video_prompt": "Leaves gently swaying in the breeze with magical light dancing",
            "voice_type": "fable",
            "style": "poetic"
        },
        {
            "name": "Educational Nature",
            "image_prompt": "A diverse ecosystem with mountains, forests, and a flowing river",
            "video_prompt": "Natural processes showing the interconnection of water, plants, and landscape",
            "voice_type": "onyx",
            "style": "educational"
        }
    ]

    for i, example in enumerate(examples, 1):
        print(f"🎬 Example {i}: {example['name']}")
        print(f"   Image: {example['image_prompt']}")
        print(f"   Video: {example['video_prompt']}")
        print(f"   Voice: {example['voice_type']} ({example['style']} style)")
        print()

        # Test this example
        payload = {
            "image_prompt": example["image_prompt"],
            "video_prompt": example["video_prompt"],
            "ratio": "1280:720",
            "duration": 5,
            "voice_type": example["voice_type"],
            "style": example["style"]
        }

        try:
            print(f"   🚀 Generating multimedia content...")
            start_time = time.time()

            response = requests.post(
                f"{API_BASE_URL}/generate-video-from-prompt",
                json=payload,
                timeout=180  # 3 minutes for complete generation
            )

            processing_time = time.time() - start_time

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

                print(f"   ✅ SUCCESS! ({processing_time:.1f}s total)")
                print(f"   🎬 Video: {result.get('video_url', 'N/A')}")

                if result.get('has_audio', False):
                    print(f"   🎙️ Audio: Generated with {result.get('voice_type')} voice")
                    print(f"   📝 Style: {result.get('narration_style')}")
                    print(f"   📁 File: {result.get('audio_file', 'N/A')}")
                    print("   🎉 COMPLETE MULTIMEDIA PACKAGE!")
                else:
                    print("   📽️ Video only (audio generation skipped)")

                # Save result
                timestamp = time.strftime("%Y%m%d_%H%M%S")
                filename = f"multimedia_result_{example['name'].lower().replace(' ', '_')}_{timestamp}.json"

                with open(filename, 'w') as f:
                    json.dump(result, f, indent=2)

                print(f"   💾 Saved to: {filename}")

            else:
                print(f"   ❌ Failed: HTTP {response.status_code}")
                print(f"   📄 Response: {response.text}")

        except requests.exceptions.Timeout:
            print(f"   ⏰ Timeout after 3 minutes")
        except Exception as e:
            print(f"   ❌ Error: {e}")

        print()

        # Small delay between examples
        if i < len(examples):
            print("   ⏸️ Waiting 10 seconds before next example...")
            time.sleep(10)
            print()

def generate_curl_examples():
    """Generate CURL examples for multimedia generation."""
    print("📋 CURL Examples for Multimedia Generation")
    print("=" * 50)
    print()

    examples = [
        {
            "name": "Descriptive Nature",
            "curl": f'''curl -X POST "{API_BASE_URL}/generate-video-from-prompt" \\
  -H "Content-Type: application/json" \\
  -d '{{
    "image_prompt": "A serene mountain lake surrounded by pine trees at golden hour",
    "video_prompt": "Gentle ripples on the water with soft morning light",
    "ratio": "1280:720",
    "duration": 5,
    "voice_type": "nova",
    "style": "descriptive"
  }}\''''
        },
        {
            "name": "Meditative Ocean",
            "curl": f'''curl -X POST "{API_BASE_URL}/generate-video-from-prompt" \\
  -H "Content-Type: application/json" \\
  -d '{{
    "image_prompt": "A peaceful beach with gentle waves at sunset",
    "video_prompt": "Soft waves washing onto shore with seabirds flying",
    "ratio": "1280:720",
    "duration": 10,
    "voice_type": "shimmer",
    "style": "meditative"
  }}\''''
        },
        {
            "name": "Poetic Forest",
            "curl": f'''curl -X POST "{API_BASE_URL}/generate-video-from-prompt" \\
  -H "Content-Type: application/json" \\
  -d '{{
    "image_prompt": "A mystical forest path with sunbeams through trees",
    "video_prompt": "Leaves swaying with magical light dancing",
    "ratio": "1024:1024",
    "duration": 5,
    "voice_type": "fable",
    "style": "poetic"
  }}\''''
        }
    ]

    for example in examples:
        print(f"# {example['name']}")
        print(example['curl'])
        print()

def main():
    """Main function."""
    print("🎬 AI Multimedia Generation Test Suite")
    print("Create complete video experiences with AI-generated:")
    print("  🎨 Images (DALL-E)")
    print("  🎬 Videos (Runway)")
    print("  🎙️ Audio Narration (OpenAI TTS)")
    print()

    # Test multimedia generation
    test_prompt_to_multimedia()

    # Generate CURL examples
    generate_curl_examples()

    print("🎉 Multimedia generation test complete!")
    print()
    print("💡 Available Voice Types:")
    print("   - alloy: Balanced voice, suitable for most content")
    print("   - echo: Warm, friendly voice")
    print("   - fable: Storytelling voice with character")
    print("   - onyx: Deep, authoritative voice")
    print("   - nova: Bright, energetic voice")
    print("   - shimmer: Soft, gentle voice")
    print()
    print("🎨 Available Styles:")
    print("   - descriptive: Detailed scene description")
    print("   - poetic: Artistic, flowing narration")
    print("   - educational: Informative, explanatory")
    print("   - meditative: Calming, peaceful guidance")

if __name__ == "__main__":
    main()