Sebastian Monroy
778ed5272f
This commit aligns the nilor-nodes with the project's new unified, name-based I/O system, as specified in the workflow override fix plan. This change establishes a stable, human-readable API contract for all workflows, replacing the previous fragile node-ID-based system. Key Changes: - **`MediaStreamInput` & `NilorUserInput`**: Added a static, non-overridable `input_name` string widget. Workflow authors now assign a logical name to each input, which is used by the Brain API to inject data. - **`MediaStreamOutput`**: Added a static `output_name` widget. This provides a stable key for the Brain API to identify and retrieve specific outputs. - **`MediaStreamOutput` (Logic)**: Corrected the completion logic to properly parse the full dictionary of named outputs it receives from the Brain API, ensuring it sends the correct, complete payload upon job completion. These changes are a critical part of the larger refactor to improve the security, scalability, and maintainability of the ComfyUI integration.
30 lines
1.3 KiB
Python
30 lines
1.3 KiB
Python
import threading
|
|
|
|
# --- Nilor-Nodes Custom Node Registration and Startup ---
|
|
# This file is executed when ComfyUI starts and discovers this custom node directory.
|
|
# It's responsible for:
|
|
# 1. Starting background services (like the SQS worker and a FastAPI server).
|
|
# 2. Registering the custom nodes with ComfyUI so they appear in the menu.
|
|
|
|
# --- Background Services ---
|
|
|
|
# 1. Start the SQS Worker Consumer
|
|
from .worker_consumer import consume_jobs
|
|
consumer_thread = threading.Thread(target=consume_jobs, daemon=True)
|
|
consumer_thread.start()
|
|
print("✅ Nilor-Nodes: SQS worker consumer thread started.")
|
|
|
|
# --- Node Registration ---
|
|
from .nilornodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
from .media_stream import NODE_CLASS_MAPPINGS as ms_NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as ms_NODE_DISPLAY_NAME_MAPPINGS
|
|
from .user_input import NODE_CLASS_MAPPINGS as ui_NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as ui_NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
NODE_CLASS_MAPPINGS.update(ms_NODE_CLASS_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ms_NODE_DISPLAY_NAME_MAPPINGS)
|
|
|
|
NODE_CLASS_MAPPINGS.update(ui_NODE_CLASS_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ui_NODE_DISPLAY_NAME_MAPPINGS)
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
|
print("✅ Nilor-Nodes: All custom nodes registered.")
|