ComfyUI-Lora-Manager/py/metadata_collector/__init__.py

import os
import logging

logger = logging.getLogger(__name__)

# Check if running in standalone mode
standalone_mode = os.environ.get("LORA_MANAGER_STANDALONE", "0") == "1" or os.environ.get("HF_HUB_DISABLE_TELEMETRY", "0") == "0"

if not standalone_mode:
    from .metadata_hook import MetadataHook
    from .metadata_registry import MetadataRegistry

    def init():
        # Install hooks to collect metadata during execution
        MetadataHook.install()
        
        # Initialize registry
        registry = MetadataRegistry()
        
        logger.info("ComfyUI Metadata Collector initialized")
        
    def get_metadata(prompt_id=None):
        """Helper function to get metadata from the registry"""
        registry = MetadataRegistry()
        return registry.get_metadata(prompt_id)
else:
    # Standalone mode - provide dummy implementations
    def init():
        logger.info("ComfyUI Metadata Collector disabled in standalone mode")
        
    def get_metadata(prompt_id=None):
        """Dummy implementation for standalone mode"""
        return {}