tasks.py

TASK_DEFINITION = {
    1: {
        "name": "Single Item Retrieval",
        "level": 1,
        "targets": [[5, 5]],
        "max_steps": 50,
        "description": "Navigate to [5,5], pick the item, and return to the Dock at [0,0]."
    },
    2: {
        "name": "Multi-Order Sequential",
        "level": 2,
        "targets": [[8, 2], [2, 8]],
        "max_steps": 150,
        "description": "Collect two items in order and return each safely to the Dock."
    },
    3: {
        "name": "Dynamic Warehouse Master",
        "level": 3,
        "targets": [[9, 9], [5, 1], [1, 5]],
        "max_steps": 300,
        "description": "Avoid moving forklifts while fulfilling a triple-item order."
    }
}

def get_task(task_id):
    return TASK_DEFINITION.get(task_id, TASK_DEFINITION[1])