task_loader.py

import json
import os

def load_all_tasks() -> tuple:
    tasks = {"EASY": [], "MEDIUM": [], "HARD": []}
    tests = {}
    
    # Find tasks dir relative to this file, not cwd
    base_dir = os.path.dirname(os.path.abspath(__file__))  
    tasks_dir = os.path.join(base_dir, "tasks")
    
    for task_folder in os.listdir(tasks_dir):
        task_path = os.path.join(tasks_dir, task_folder)
        if not os.path.isdir(task_path):
            continue
        with open(os.path.join(task_path, "task.json")) as f:
            task = json.load(f)
        with open(os.path.join(task_path, "tests.json")) as f:
            test_data = json.load(f)
        difficulty = task["difficulty"]
        tasks[difficulty].append(task)
        tests[task["id"]] = {
            "baseline_time": task["baseline_time_ms"],
            "baseline_mem": task["baseline_mem_mib"],
            "cases": test_data["cases"]
        }
    return tasks, tests

TASKS, TESTS = load_all_tasks()