Sly
cfd8e2cfde
feat: migrate ComfyUI nodes from presigned URLs to Brain API storage endpoints ## Core Architecture Changes - Add new BrainApiClient for direct Brain API communication - Replace presigned URL workflow with storage_id/filename pattern - Update MediaStreamInput to use storage_id + filename instead of presigned_download_url - Update MediaStreamOutput to use storage_id + filename instead of presigned_upload_url ## Brain API Client Implementation - Add comprehensive BrainApiClient with authentication and error handling - Implement get_presigned_upload_url() and get_presigned_download_url() methods - Add proper Bearer token authentication for Brain API requests - Include environment variable configuration for Brain API endpoint and credentials ## Media Stream Node Updates - Refactor MediaStreamInput to request presigned URLs from Brain API using storage_id - Update MediaStreamOutput to upload directly to MinIO using Brain API presigned URLs - Simplify node interface by removing presigned URL inputs from workflow templates - Update JavaScript frontend to hide system inputs and manage node UI state ## Infrastructure Improvements - Add .idea/ to .gitignore for IDE file exclusions - Update user_input.py to use new storage_id pattern - Streamline web interface JavaScript for cleaner node management
110 lines
2.9 KiB
Python
110 lines
2.9 KiB
Python
category = "Nilor Nodes 👺"
|
|
subcategories = {
|
|
"io": "/IO",
|
|
}
|
|
|
|
from .controllers import CONTROLLER_HOOK
|
|
|
|
|
|
class NilorUserInput_String:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input_name": (
|
|
"STRING",
|
|
{"default": "my_string_input", "multiline": False},
|
|
),
|
|
"value": ("STRING", {"default": "", "multiline": True}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING", CONTROLLER_HOOK)
|
|
RETURN_NAMES = ("string", "_controller_hook")
|
|
FUNCTION = "get_value"
|
|
CATEGORY = category + subcategories["io"]
|
|
|
|
def get_value(self, input_name, value):
|
|
return (value, None)
|
|
|
|
|
|
class NilorUserInput_Int:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input_name": (
|
|
"STRING",
|
|
{"default": "my_int_input", "multiline": False},
|
|
),
|
|
"value": ("INT", {"default": 0}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("INT", CONTROLLER_HOOK)
|
|
RETURN_NAMES = ("int", "_controller_hook")
|
|
FUNCTION = "get_value"
|
|
CATEGORY = category + subcategories["io"]
|
|
|
|
def get_value(self, input_name, value):
|
|
return (value, None)
|
|
|
|
|
|
class NilorUserInput_Float:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input_name": (
|
|
"STRING",
|
|
{"default": "my_float_input", "multiline": False},
|
|
),
|
|
"value": ("FLOAT", {"default": 0.0}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("FLOAT", CONTROLLER_HOOK)
|
|
RETURN_NAMES = ("float", "_controller_hook")
|
|
FUNCTION = "get_value"
|
|
CATEGORY = category + subcategories["io"]
|
|
|
|
def get_value(self, input_name, value):
|
|
return (value, None)
|
|
|
|
|
|
class NilorUserInput_Boolean:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"input_name": (
|
|
"STRING",
|
|
{"default": "my_bool_input", "multiline": False},
|
|
),
|
|
"value": ("BOOLEAN", {"default": False}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("BOOLEAN", CONTROLLER_HOOK)
|
|
RETURN_NAMES = ("boolean", "_controller_hook")
|
|
FUNCTION = "get_value"
|
|
CATEGORY = category + subcategories["io"]
|
|
|
|
def get_value(self, input_name, value):
|
|
return (value, None)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"NilorUserInput_String": NilorUserInput_String,
|
|
"NilorUserInput_Int": NilorUserInput_Int,
|
|
"NilorUserInput_Float": NilorUserInput_Float,
|
|
"NilorUserInput_Boolean": NilorUserInput_Boolean,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"NilorUserInput_String": "👺 User Input (String)",
|
|
"NilorUserInput_Int": "👺 User Input (Int)",
|
|
"NilorUserInput_Float": "👺 User Input (Float)",
|
|
"NilorUserInput_Boolean": "👺 User Input (Boolean)",
|
|
}
|