__init__.py

import server as comfy_server
from .router import upload_tensor_handler

from .nodes import (
    BridgeSenderLatent,
    BridgeReceiverLatent,
    BridgeSenderImage,
    BridgeReceiverImage,
    BridgeSenderConditioning,
    BridgeReceiverConditioning,
    BridgeExecutionSync
)

@comfy_server.PromptServer.instance.routes.post("/tensor_bridge/upload")
async def route_upload_tensor(request):
    return await upload_tensor_handler(request)

NODE_CLASS_MAPPINGS = {
    "BridgeSenderLatent": BridgeSenderLatent,
    "BridgeReceiverLatent": BridgeReceiverLatent,
    "BridgeSenderImage": BridgeSenderImage,
    "BridgeReceiverImage": BridgeReceiverImage,
    "BridgeSenderConditioning": BridgeSenderConditioning,
    "BridgeReceiverConditioning": BridgeReceiverConditioning,
    "BridgeExecutionSync": BridgeExecutionSync
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "BridgeSenderLatent": "📡 Bridge: Send Latent",
    "BridgeReceiverLatent": "📡 Bridge: Receive Latent",
    "BridgeSenderImage": "📡 Bridge: Send Image",
    "BridgeReceiverImage": "📡 Bridge: Receive Image",
    "BridgeSenderConditioning": "📡 Bridge: Send Conditioning",
    "BridgeReceiverConditioning": "📡 Bridge: Receive Conditioning",
    "BridgeExecutionSync": "⏱️ Bridge: Sync (Wait for all)"
}

print("[TensorBridge] ComfyUI-Tensor-Bridge Loaded: Nodes registered")