import yaml

def save_paths_to_yaml(paths, filename="paths.yml"):
    # to facilitate testing for snowplow (do not have to redo the drone part each time)
    serializable_paths = {
        k: [[u, v] for (u, v) in v] for k, v in paths.items()
    }
    with open(f"paths/{filename}", 'w') as f:
        yaml.dump(serializable_paths, f)

def load_paths_from_yaml(graph, filename="paths.yml"):
    with open(filename, 'r') as f:
        raw_paths = yaml.safe_load(f)
    
    # Reconstruit le format avec des tuples (u, v)
    paths = {
        k: [tuple(edge) for edge in v] for k, v in raw_paths.items()
    }

    return paths