"""
Configuration utilities for loading and managing settings.
"""
import yaml
import os
from typing import Dict


def load_config(config_path: str = "config.yaml") -> Dict:
    """Load configuration from YAML file."""
    if not os.path.exists(config_path):
        raise FileNotFoundError(f"Config file not found: {config_path}")

    with open(config_path, "r") as f:
        config = yaml.safe_load(f)

    return config


def create_directories(config: Dict):
    """Create necessary directories for checkpoints and logs."""
    checkpoint_dir = config["training"]["checkpoint_dir"]
    log_dir = config["training"]["log_dir"]

    os.makedirs(checkpoint_dir, exist_ok=True)
    os.makedirs(log_dir, exist_ok=True)