import os

# Simple folder_paths module to replace ComfyUI's folder_paths
current_dir = os.path.dirname(os.path.abspath(__file__))
models_dir = os.path.join(current_dir, "models")

# Model folder mappings
model_folder_paths = {}

def add_model_folder_path(name, path):
    """Add a model folder path."""
    model_folder_paths[name] = path
    os.makedirs(path, exist_ok=True)

def get_full_path(dirname, filename):
    """Get the full path for a model file."""
    if dirname in model_folder_paths:
        return os.path.join(model_folder_paths[dirname], filename)
    return os.path.join(models_dir, dirname, filename)

# Initialize default paths
os.makedirs(models_dir, exist_ok=True)