agents/planner.py

class PlannerAgent:

    def plan(self, task: str):
        """
        Break complex requests into steps
        """

        task = task.lower()

        steps = []

        if "remove background" in task:
            steps.append("background")

        if "pdf" in task:
            steps.append("pdf")

        if "describe" in task:
            steps.append("analyze")

        if "edit" in task or "change" in task:
            steps.append("edit")

        if "generate" in task or "create" in task:
            steps.append("generate")

        if not steps:
            steps.append("generate")

        return steps