start_task_master.py

#!/usr/bin/env python3
"""
Task Master Quick Start
Launch Task Master as the central AI project manager for Julep agents
"""

import asyncio
import sys
import os
from pathlib import Path

# Add src to path
sys.path.append(str(Path(__file__).parent / "src"))

# Import Task Master components
from task_master_julep_bridge import (
    TaskMasterCommandCenter,
    TaskMasterJulepOrchestrator,
    demonstrate_task_master_julep_integration
)

async def interactive_task_master():
    """Interactive Task Master session"""
    
    print("\n" * 2)
    print("╔" + "═" * 68 + "╗")
    print("║" + " " * 20 + "TASK MASTER ACTIVATED" + " " * 27 + "║")
    print("║" + " " * 10 + "Central AI Project Manager for Julep Agents" + " " * 14 + "║")
    print("╚" + "═" * 68 + "╝")
    
    # Initialize command center
    command_center = TaskMasterCommandCenter()
    
    print("\n🎯 Task Master is ready to orchestrate your agents!")
    print("=" * 70)
    
    print("\n📋 Available Commands:")
    print("  1. Create task         - Create a new task")
    print("  2. Research           - Perform deep research")
    print("  3. Scrape             - Web scraping task")
    print("  4. Analyze            - Analyze data/text")
    print("  5. Workflow           - Create complex workflow")
    print("  6. Demo               - Run full demonstration")
    print("  7. Exit               - Exit Task Master")
    
    while True:
        print("\n" + "-" * 70)
        choice = input("\n🎯 Enter command (1-7) or describe your task: ").strip()
        
        if choice == "7" or choice.lower() in ["exit", "quit"]:
            print("\n👋 Task Master shutting down. Goodbye!")
            break
            
        elif choice == "6" or choice.lower() == "demo":
            await demonstrate_task_master_julep_integration()
            
        elif choice == "1":
            title = input("📝 Task title: ")
            description = input("📄 Task description: ")
            priority = input("⚡ Priority (low/medium/high/critical) [medium]: ") or "medium"
            
            task = await command_center.orchestrator.create_task(
                title=title,
                description=description,
                priority=priority
            )
            
            print(f"\n✅ Task created: {task.id}")
            print(f"🔧 Required tools: {', '.join(task.required_tools)}")
            
            execute = input("\n▶️ Execute task now? (y/n): ")
            if execute.lower() == 'y':
                result = await command_center.router.route_and_execute(task)
                print(f"\n📊 Result: {task.status}")
                
        elif choice == "2":
            topic = input("🔍 Research topic: ")
            result = await command_center.process_user_request(
                f"Research {topic} using multiple sources including academic papers"
            )
            print(f"\n✅ Research completed: {result['status']}")
            
        elif choice == "3":
            url = input("🌐 URL to scrape (or description): ")
            result = await command_center.process_user_request(
                f"Scrape data from {url} using advanced crawling techniques"
            )
            print(f"\n✅ Scraping completed: {result['status']}")
            
        elif choice == "4":
            text = input("📊 Text/data to analyze: ")
            analysis_type = input("🧠 Analysis type (sentiment/entities/patterns) [sentiment]: ") or "sentiment"
            
            result = await command_center.process_user_request(
                f"Analyze the following for {analysis_type}: {text}"
            )
            print(f"\n✅ Analysis completed: {result['status']}")
            
        elif choice == "5":
            project_name = input("📁 Project name: ")
            num_requirements = int(input("📋 Number of requirements: "))
            
            requirements = []
            for i in range(1, num_requirements + 1):
                req = input(f"  Requirement {i}: ")
                requirements.append(req)
            
            workflow_id = await command_center.create_project_workflow(
                project_name,
                requirements
            )
            print(f"\n✅ Workflow created: {workflow_id}")
            
        else:
            # Treat as free-form task description
            if choice:
                print(f"\n🤔 Processing: {choice}")
                result = await command_center.process_user_request(choice)
                print(f"\n✅ Task completed: {result['status']}")
            else:
                print("❌ Invalid input. Please try again.")
    
    print("\n" + "=" * 70)
    print("Task Master session ended.")

async def main():
    """Main entry point"""
    
    # Check for command line arguments
    if len(sys.argv) > 1:
        if sys.argv[1] == "demo":
            await demonstrate_task_master_julep_integration()
        elif sys.argv[1] == "help":
            print("Usage: python start_task_master.py [demo|interactive]")
            print("  demo        - Run demonstration")
            print("  interactive - Start interactive session (default)")
        else:
            # Process as task request
            request = " ".join(sys.argv[1:])
            command_center = TaskMasterCommandCenter()
            result = await command_center.process_user_request(request)
            print(f"\n✅ Task completed: {result['status']}")
    else:
        # Interactive mode
        await interactive_task_master()

if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\n\n👋 Task Master interrupted. Goodbye!")
    except Exception as e:
        print(f"\n❌ Error: {e}")
        sys.exit(1)