from .models import Email

def get_initial_emails(task_level: str) -> list[Email]:
    if task_level == "EASY":
        return [
            Email(id="e1", subject="Server outage", body="The production server is down. Urgent!"),
            Email(id="e2", subject="Weekly Newsletter", body="Here are the top stories of the week."),
        ]
    elif task_level == "MEDIUM":
        return [
            Email(id="e1", subject="Project sync", body="Let's schedule a meeting tomorrow at 10 AM to discuss the project."),
            Email(id="e2", subject="Bug report", body="Found a bug in the login flow. Please fix it."),
        ]
    elif task_level == "HARD":
        return [
            Email(id="e1", subject="Client escalation", body="CLIENT ANGRY! Needs meeting today at 2 PM or 3 PM!"),
            Email(id="e2", subject="Urgent bug", body="Payment gateway is failing. Critical!"),
            Email(id="e3", subject="Lunch?", body="Want to grab lunch today at 12 PM?"),
        ]
    return []