main.py

"""
Mood Orchestrator Agent - Main Application
"""
import asyncio
from google.adk import Runner
from google.adk.sessions import InMemorySessionService
from google.genai import types
from agent import root_agent


def print_banner():
    """Print welcome banner."""
    print("\n" + "="*60)
    print("🎵  MOOD ORCHESTRATOR AGENT  🎵")
    print("="*60)
    print("Welcome! I'll capture your voice, detect your mood,")
    print("generate personalized music, and sync your smart lights!")
    print("="*60 + "\n")


async def run_agent():
    """Run the mood orchestrator agent."""
    print_banner()
    
    # Create runner with in-memory session service
    runner = Runner(
        app_name="mood_orchestrator",
        agent=root_agent,
        session_service=InMemorySessionService()
    )
    
    # User and session IDs
    user_id = "user_001"
    session_id = "session_001"
    
    print("🎤 Ready to start! The agent will guide you through the process.\n")
    print("Type 'start' to begin, or 'quit' to exit.\n")
    
    while True:
        user_input = input("You: ").strip()
        
        if user_input.lower() in ['quit', 'exit', 'q']:
            print("\n👋 Goodbye! Thanks for using Mood Orchestrator!\n")
            break
        
        if not user_input:
            continue
        
        # Create message content
        content = types.Content(
            role='user',
            parts=[types.Part(text=user_input)]
        )
        
        print("\n🤖 Agent:")
        print("-" * 60)
        
        # Run agent and stream events
        try:
            async for event in runner.run_async(
                user_id=user_id,
                session_id=session_id,
                new_message=content
            ):
                # Handle different event types
                if event.content and event.content.parts:
                    for part in event.content.parts:
                        if part.text:
                            # Print agent responses
                            if event.author == root_agent.name:
                                print(f"{part.text}")
                
                # Show tool execution
                if hasattr(event, 'tool_name') and event.tool_name:
                    print(f"\n🔧 Executing tool: {event.tool_name}")
                
        except Exception as e:
            print(f"\n❌ Error: {str(e)}")
            print("Please try again or type 'quit' to exit.")
        
        print("-" * 60)
        print()


def main():
    """Main entry point."""
    try:
        asyncio.run(run_agent())
    except KeyboardInterrupt:
        print("\n\n👋 Interrupted. Goodbye!\n")
    except Exception as e:
        print(f"\n❌ Fatal error: {str(e)}\n")


if __name__ == "__main__":
    main()