task_classifier.py

def classify_prompt(prompt: str) -> str:
    """
    Classify the prompt into one of three task types that require
    different confidence interpretation strategies.

    Returns: "creative" | "analytical" | "factual"
    """
    p = prompt.lower()

    creative_words = [
        "story", "poem", "imagine", "invent", "create", "write a", "generate a",
        "fiction", "narrative", "fantasy", "describe a scene", "compose"
    ]
    analytical_words = [
        "explain", "why", "how", "should", "what causes", "what is the reason",
        "compare", "analyze", "analyse", "difference between", "pros and cons",
        "evaluate", "argue", "opinion", "think about"
    ]

    if any(word in p for word in creative_words):
        return "creative"
    elif any(word in p for word in analytical_words):
        return "analytical"
    else:
        return "factual"