app/core/router.py

# app/core/router.py

def route_agent(task: str):
    t = task.lower()

    if "image" in t or "hair" in t:
        return "vision_agent"

    if "fashion" in t or "style" in t:
        return "fashion_agent"

    if "voice" in t or "tts" in t:
        return "voice_agent"

    if "code" in t:
        return "chat_agent_code"

    return "chat_agent"