example_usage.py

"""
Example usage of the Smart Task Prioritizer API.

This file demonstrates how to use the prioritizer programmatically
for integration into other applications.
"""

import os
import sys
from datetime import datetime, timedelta
from pathlib import Path

# Add src to Python path for imports
sys.path.insert(0, str(Path(__file__).parent / 'src'))

from src.models.task import Task, TaskList
from src.ai.prioritizer import PrioritizationEngine
from src.utils.config import load_config


def create_sample_tasks():
    """Create a sample task list for demonstration."""
    tasks = [
        Task(
            title="Urgent client presentation",
            deadline=datetime.now() + timedelta(days=1),
            importance=5,
            effort_minutes=120
        ),
        Task(
            title="Code review for team",
            deadline=datetime.now() + timedelta(days=2),
            importance=4,
            effort_minutes=45
        ),
        Task(  
            title="Update documentation",
            deadline=datetime.now() + timedelta(days=7),
            importance=3,
            effort_minutes=90
        ),
        Task(
            title="Quick bug fix",
            deadline=datetime.now() + timedelta(days=1),
            importance=3,
            effort_minutes=15
        )
    ]
    
    return TaskList(tasks=tasks, created_at=datetime.now())


def demonstrate_prioritization():
    """Demonstrate the AI prioritization process."""
    print("🤖 Smart Task Prioritizer API Demo")
    print("=" * 40)
    
    # Load configuration
    load_config()
    
    # Create sample tasks
    task_list = create_sample_tasks()
    print(f"📝 Created {task_list.task_count()} sample tasks")
    
    # Initialize AI engine
    try:
        engine = PrioritizationEngine()
        print("🧠 AI engine initialized successfully")
    except ValueError as e:
        print(f"⚠️  {e}")
        print("Using fallback prioritization...")
        # For demo purposes, we'll continue with a mock engine
        return
    
    # Perform prioritization
    print("🔄 Analyzing tasks...")
    prioritized_tasks = engine.prioritize_tasks(task_list.tasks)
    
    # Display results
    print("\n🎯 Prioritized Results:")
    print("-" * 30)
    
    for pt in prioritized_tasks:
        print(f"{pt.rank}. {pt.task.title}")
        print(f"   Priority Score: {pt.priority_score:.1f}")
        print(f"   Rationale: {pt.rationale}")
        print()
    
    return prioritized_tasks


def export_to_json(prioritized_tasks, filename="prioritized_tasks.json"):
    """Export prioritized tasks to JSON format."""
    import json
    
    # Convert to JSON-serializable format
    export_data = []
    for pt in prioritized_tasks:
        export_data.append({
            "rank": pt.rank,
            "title": pt.task.title,
            "deadline": pt.task.deadline.isoformat(),
            "importance": pt.task.importance,
            "effort_minutes": pt.task.effort_minutes,
            "priority_score": pt.priority_score,
            "rationale": pt.rationale
        })
    
    with open(filename, 'w') as f:
        json.dump(export_data, f, indent=2)
    
    print(f"📄 Results exported to {filename}")


if __name__ == "__main__":
    # Run the demonstration
    results = demonstrate_prioritization()
    
    if results:
        # Export results
        export_to_json(results)
        
        print("\n✅ Demo completed successfully!")
        print("Check the exported JSON file for integration examples.")
    else:
        print("\n⚠️  Demo requires GEMINI_API_KEY configuration.")
        print("Please run './setup.sh' and configure your .env file.")