comfy_integration/nodes.py

import asyncio
import execution
import server
from nodes import init_extra_nodes, NODE_CLASS_MAPPINGS

def import_custom_nodes() -> None:
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    server_instance = server.PromptServer(loop)
    execution.PromptQueue(server_instance)
    
    loop.run_until_complete(init_extra_nodes())

import_custom_nodes()

print("✅ ComfyUI custom nodes and class mappings are ready.")