PDF-Extract-Kit/pdf_extract_kit/registry/registry.py

class Registry:
    def __init__(self):
        self._registry = {}

    def register(self, name):
        def decorator(item):
            if name in self._registry:
                raise ValueError(f"Item {name} already registered.")
            self._registry[name] = item
            return item
        return decorator

    def get(self, name):
        if name not in self._registry:
            raise ValueError(f"Item {name} not found in registry.")
        return self._registry[name]

    def list_items(self):
        return list(self._registry.keys())

# Create global registries for tasks and models
TASK_REGISTRY = Registry()
MODEL_REGISTRY = Registry()