example_usage.py

#!/usr/bin/env python3
"""
Example Usage Script for AI Art Generator
Demonstrates various features and usage patterns
"""

import asyncio
import os
from pathlib import Path
from ai_art_generator import AIArtGenerator

async def example_single_generation():
    """Example: Generate a single image"""
    print("🎨 Example 1: Single Image Generation")
    print("=" * 50)
    
    generator = AIArtGenerator()
    await generator.connect()
    
    try:
        result = await generator.generate_single_image(
            prompt="A futuristic stealth jet streaking through a neon-lit cityscape with glowing purple exhaust",
            model="runware:97@2",
            width=1344,
            height=768,
            steps=40,
            cfg_scale=5
        )
        
        if "error" in result:
            print(f"❌ Generation failed: {result['error']}")
        else:
            print(f"✅ Generated successfully!")
            print(f"📁 Local file: {result['local_path']}")
            print(f"🌐 URL: {result['image_url']}")
            print(f"📊 Model: {result['model']}")
            print(f"📏 Size: {result['width']}x{result['height']}")
            
    finally:
        await generator.disconnect()

async def example_batch_generation():
    """Example: Generate multiple images in batch"""
    print("\n🎨 Example 2: Batch Generation")
    print("=" * 50)
    
    generator = AIArtGenerator()
    await generator.connect()
    
    try:
        prompts = [
            "A serene mountain landscape at sunset with golden light",
            "A cyberpunk city street at night with neon lights",
            "A magical forest with glowing mushrooms and fairy lights",
            "A futuristic robot in a high-tech laboratory"
        ]
        
        print(f"📝 Generating {len(prompts)} images...")
        
        results = await generator.generate_batch(
            prompts=prompts,
            model="runware:101@1",
            width=1024,
            height=1024
        )
        
        successful = [r for r in results if "error" not in r]
        failed = [r for r in results if "error" in r]
        
        print(f"✅ Successfully generated {len(successful)} images")
        if failed:
            print(f"❌ Failed to generate {len(failed)} images")
        
        for i, result in enumerate(successful, 1):
            print(f"  {i}. {result['local_path']}")
            
    finally:
        await generator.disconnect()

async def example_template_generation():
    """Example: Generate images using templates"""
    print("\n🎨 Example 3: Template-Based Generation")
    print("=" * 50)
    
    generator = AIArtGenerator()
    await generator.connect()
    
    try:
        # Generate portraits
        print("📸 Generating portrait templates...")
        portrait_results = await generator.generate_from_template(
            template_type="portrait",
            subject="a young woman with blue hair and green eyes",
            style="anime style",
            count=2
        )
        
        successful_portraits = [r for r in portrait_results if "error" not in r]
        print(f"✅ Generated {len(successful_portraits)} portraits")
        
        # Generate landscapes
        print("🏔️ Generating landscape templates...")
        landscape_results = await generator.generate_from_template(
            template_type="landscape",
            subject="majestic mountains",
            location="Swiss Alps",
            style="realistic",
            count=2
        )
        
        successful_landscapes = [r for r in landscape_results if "error" not in r]
        print(f"✅ Generated {len(successful_landscapes)} landscapes")
        
        # Show results
        for i, result in enumerate(successful_portraits + successful_landscapes, 1):
            print(f"  {i}. {result['local_path']}")
            
    finally:
        await generator.disconnect()

async def example_style_variations():
    """Example: Generate style variations"""
    print("\n🎨 Example 4: Style Variations")
    print("=" * 50)
    
    generator = AIArtGenerator()
    await generator.connect()
    
    try:
        base_prompt = "A majestic dragon flying over mountains"
        styles = ["realistic", "anime", "cyberpunk", "artistic"]
        
        print(f"🎭 Generating style variations for: {base_prompt}")
        print(f"🎨 Styles: {', '.join(styles)}")
        
        results = await generator.generate_with_style_variations(
            base_prompt=base_prompt,
            styles=styles,
            model="runware:97@2"
        )
        
        successful = [r for r in results if "error" not in r]
        print(f"✅ Generated {len(successful)} style variations")
        
        for i, (result, style) in enumerate(zip(successful, styles), 1):
            print(f"  {i}. {style}: {result['local_path']}")
            
    finally:
        await generator.disconnect()

async def example_custom_models():
    """Example: Using different models"""
    print("\n🎨 Example 5: Custom Models")
    print("=" * 50)
    
    generator = AIArtGenerator()
    await generator.connect()
    
    try:
        # Test different models
        models_to_test = {
            "Realistic": "runware:97@2",
            "Artistic": "runware:101@1", 
            "Photorealistic": "runware:100@1"
        }
        
        prompt = "A beautiful sunset over a calm ocean"
        
        for model_name, model_id in models_to_test.items():
            print(f"🎨 Testing {model_name} model...")
            
            result = await generator.generate_single_image(
                prompt=prompt,
                model=model_id,
                width=1024,
                height=1024
            )
            
            if "error" not in result:
                print(f"  ✅ {model_name}: {result['local_path']}")
            else:
                print(f"  ❌ {model_name}: {result['error']}")
                
    finally:
        await generator.disconnect()

async def example_advanced_parameters():
    """Example: Using advanced generation parameters"""
    print("\n🎨 Example 6: Advanced Parameters")
    print("=" * 50)
    
    generator = AIArtGenerator()
    await generator.connect()
    
    try:
        prompt = "A highly detailed portrait of a wise old wizard with magical aura"
        
        # High quality settings
        result = await generator.generate_single_image(
            prompt=prompt,
            model="runware:97@2",
            width=1344,
            height=768,
            steps=50,  # More steps for higher quality
            cfg_scale=7,  # Higher CFG for more prompt adherence
            negative_prompt="blurry, low quality, distorted, ugly, bad anatomy"
        )
        
        if "error" not in result:
            print(f"✅ High-quality generation: {result['local_path']}")
            print(f"📊 Parameters used:")
            print(f"  - Steps: {result['steps']}")
            print(f"  - CFG Scale: {result['cfg_scale']}")
            print(f"  - Size: {result['width']}x{result['height']}")
        else:
            print(f"❌ Generation failed: {result['error']}")
            
    finally:
        await generator.disconnect()

def show_available_features():
    """Show available models and templates"""
    print("\n📋 Available Features")
    print("=" * 50)
    
    generator = AIArtGenerator()
    
    # Show models
    print("🎯 Available Models:")
    models = generator.get_available_models()
    for name, model_id in models.items():
        print(f"  {name}: {model_id}")
    
    # Show templates
    print("\n📝 Available Templates:")
    templates = generator.get_prompt_templates()
    for template_type, template_list in templates.items():
        print(f"  {template_type}:")
        for i, template in enumerate(template_list[:2]):  # Show first 2 examples
            print(f"    {i+1}. {template[:60]}...")

async def main():
    """Run all examples"""
    print("🚀 AI Art Generator - Example Usage")
    print("=" * 60)
    
    # Check if API key is set
    if not os.getenv("RUNWARE_API_KEY"):
        print("❌ RUNWARE_API_KEY not found in environment variables")
        print("Please set your API key in the .env file")
        return
    
    # Show available features
    show_available_features()
    
    # Run examples
    try:
        await example_single_generation()
        await example_batch_generation()
        await example_template_generation()
        await example_style_variations()
        await example_custom_models()
        await example_advanced_parameters()
        
        print("\n🎉 All examples completed!")
        print("📁 Check the 'generated_images' folder for your creations")
        
    except Exception as e:
        print(f"❌ Error running examples: {e}")

if __name__ == "__main__":
    asyncio.run(main())