29 · 自定义节点开发入门

cd ComfyUI/custom_nodes/
mkdir my-first-nodes
cd my-first-nodes
# 创建以下文件：
touch __init__.py   # 节点注册
touch nodes.py      # 节点逻辑

class HelloWorld:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "text": ("STRING", {"default": "Hello ComfyUI!"}),
            }
        }
    
    RETURN_TYPES = ("STRING",)
    FUNCTION = "say_hello"
    CATEGORY = "my_nodes"
 
    def say_hello(self, text):
        return (f"[MyNode] {text}",)
 
NODE_CLASS_MAPPINGS = {
    "HelloWorld": HelloWorld,
}
 
NODE_DISPLAY_NAME_MAPPINGS = {
    "HelloWorld": "Hello World Node",
}

from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
 
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']

import requests
 
class TranslatePrompt:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "chinese_prompt": ("STRING", {"multiline": True}),
            }
        }
    
    RETURN_TYPES = ("STRING",)
    FUNCTION = "translate"
    CATEGORY = "my_nodes"
 
    def translate(self, chinese_prompt):
        # 调用翻译 API
        # ... 省略 API 调用代码
        english_prompt = translated_text
        return (english_prompt,)