example_usage.py

#!/usr/bin/env python3
"""
Example usage of AI Video Generation System.
This script demonstrates how to use the system programmatically.
"""

import os
import sys
from pathlib import Path

# Add src directory to path
sys.path.append(os.path.join(os.path.dirname(__file__), 'src'))

def create_sample_script():
    """Create a sample script for testing."""
    script_content = """
    FADE IN:
    
    INT. COFFEE SHOP - MORNING
    
    The bustling coffee shop is filled with morning commuters. The aroma of freshly brewed coffee fills the air.
    
    SARAH, a determined young woman in her late 20s, sits at a corner table with her laptop open. She's focused and typing rapidly.
    
    SARAH
    (muttering to herself)
    This presentation has to be perfect. My entire career depends on it.
    
    A barista approaches with her order.
    
    BARISTA
    (cheerful)
    One large cappuccino for Sarah!
    
    SARAH
    (looking up, grateful)
    Thank you so much. You're a lifesaver.
    
    She takes a sip and returns to her work, more confident now.
    
    CUT TO:
    
    INT. OFFICE CONFERENCE ROOM - LATER
    
    SARAH stands at the front of a conference room, presenting to a panel of executives.
    
    SARAH
    (confident, professional)
    And that's why I believe this strategy will increase our market share by 25% in the next quarter.
    
    The executives nod approvingly.
    
    FADE OUT.
    """
    
    script_path = "sample_script.txt"
    with open(script_path, 'w') as f:
        f.write(script_content)
    
    return script_path

def main():
    """Run example usage."""
    print("AI Video Generation System - Example Usage")
    print("=" * 50)
    
    try:
        # Import modules
        from config import config
        from database import db_manager
        from script_parser import ScriptParser
        from character_generator import CharacterGenerator
        from storyboard_analyzer import StoryboardAnalyzer
        from video_generator import VideoGenerator
        
        print("✓ All modules imported successfully")
        
        # Create sample script
        print("\n1. Creating sample script...")
        script_path = create_sample_script()
        print(f"✓ Created sample script: {script_path}")
        
        # Create project
        print("\n2. Creating project...")
        project_id = db_manager.create_project("Sample Movie", script_path)
        print(f"✓ Created project with ID: {project_id}")
        
        # Parse script
        print("\n3. Parsing script...")
        parser = ScriptParser()
        characters, scenes = parser.parse_script(script_path)
        print(f"✓ Found {len(characters)} characters and {len(scenes)} scenes")
        
        # Display characters
        print("\nCharacters found:")
        for char in characters:
            print(f"  - {char.name}: {char.description[:50]}... (importance: {char.importance_score:.2f})")
        
        # Save characters to database
        print("\n4. Saving characters to database...")
        for char in characters:
            char_id = db_manager.create_character(
                project_id=project_id,
                name=char.name,
                description=char.description,
                traits=char.traits,
                importance_score=char.importance_score
            )
            print(f"✓ Saved character: {char.name} (ID: {char_id})")
        
        # Save scenes to database
        print("\n5. Saving scenes to database...")
        for scene in scenes:
            scene_id = db_manager.create_scene(
                project_id=project_id,
                scene_number=scene.scene_number,
                header=scene.header,
                description=scene.description,
                dialogue=str(scene.dialogue),
                action=str(scene.action),
                camera_notes=str(scene.camera_notes),
                emotions=scene.emotions
            )
            print(f"✓ Saved scene {scene.scene_number}: {scene.header}")
        
        # Get project summary
        print("\n6. Project summary:")
        summary = db_manager.get_project_summary(project_id)
        print(f"  - Characters: {summary['character_count']}")
        print(f"  - Scenes: {summary['scene_count']}")
        print(f"  - Videos: {summary['video_count']}")
        
        # Note about character generation and video generation
        print("\n7. Next steps:")
        print("  - Run: python src/main.py generate-references --project-id", project_id)
        print("  - Run: python src/main.py analyze-storyboard --project-id", project_id, "--storyboard storyboard.png")
        print("  - Run: python src/main.py generate-video --project-id", project_id)
        
        print("\n✓ Example completed successfully!")
        
    except Exception as e:
        print(f"\n❌ Error: {e}")
        import traceback
        traceback.print_exc()
        return 1
    
    finally:
        # Clean up
        if 'script_path' in locals() and os.path.exists(script_path):
            os.unlink(script_path)
            print(f"\n✓ Cleaned up sample script")
    
    return 0

if __name__ == "__main__":
    sys.exit(main())