ComfyUI Extension: AIYang_TripleAPI

Authored by AlexXia007

Created

Updated

0 stars

ComfyUI custom nodes integrating three Chinese AI image generation platforms (Hi Dream, Douban, NanoBanana) supporting text-to-image and image-to-image generation with parallel processing and robust error handling. (Description by CC)

Custom Nodes (0)

    README

    ComfyUI 三合一 AI 图像生成节点

    这个项目为 ComfyUI 提供了两个强大的自定义节点，集成了三个主流的中文AI图像生成平台：即梦AI、火山引擎豆包和NanoBanana。支持文生图和图生图功能。

    功能特点

    • 🎨 三平台集成：同时支持即梦AI、火山引擎豆包、NanoBanana三个平台
    • 🔄 双模式支持：文生图 (Text-to-Image) 和 图生图 (Image-to-Image)
    • ⚡ 并行生成：可同时调用多个平台生成图片，提高效率
    • 🎯 智能轮询：自动等待图片生成完成，无需手动查询
    • 🛡️ 容错处理：单个平台失败不影响其他平台正常工作
    • 📊 详细状态：实时显示每个平台的生成状态和错误信息
    • 🔗 URL输出：提供每个API生成的图片URL，便于后续处理和分享
    • 🔄 重试机制：支持API调用失败时自动重试，提高成功率
    • ⏱️ 超时控制：可配置请求和轮询超时时间，避免长时间等待

    系统要求

    • Python 3.8 或更高版本
    • ComfyUI 已安装并运行
    • 至少 4GB 可用内存

    依赖说明

    本项目需要以下 Python 包：

    • Pillow (>=9.0.0): 用于图像处理
    • numpy (>=1.21.0): 用于数值计算和数组操作
    • torch (>=1.12.0): PyTorch 深度学习框架（ComfyUI 通常已包含）

    注意: ComfyUI 通常已经包含了 torch 和 numpy，如果您的 ComfyUI 环境已配置好，可能只需要安装 Pillow。

    安装方法

    方法一：使用 Git 克隆（推荐）

    1. 将此仓库克隆到 ComfyUI 的 custom_nodes 目录：
    cd ComfyUI/custom_nodes
git clone https://github.com/AlexXia007/AIYang_TripleAPI.git
    1. 进入项目目录并安装依赖：
    cd AIYang_comfyui_tripapi
pip install -r requirements.txt
    1. 重启 ComfyUI

    方法二：手动安装

    1. 下载项目 ZIP 文件并解压到 ComfyUI/custom_nodes 目录

    2. 在项目目录下打开终端，执行：

    pip install -r requirements.txt
    1. 重启 ComfyUI

    验证安装

    重启 ComfyUI 后，在节点菜单中应该能看到：

    • TripleAPIGenerate (三合一文生图节点)
    • TripleAPIImg2Img (三合一图生图节点)

    如果节点未出现，请检查：

    • 依赖是否安装成功
    • ComfyUI 控制台是否有错误信息
    • 项目是否在正确的 custom_nodes 目录下

    节点说明

    1. TripleAPIGenerate (三合一文生图节点)

    用于根据文字描述生成图片。

    输入参数详解

    基础参数

    • prompt (STRING):

      • 描述: 图片生成的主要提示词
      • 填写: 详细描述你想要生成的图片内容，如"一只可爱的小猫坐在花园里，阳光明媚，高清摄影风格"
      • 建议: 越详细越好，包含风格、场景、光线、质量等描述

    • negative_prompt (STRING):

      • 描述: 负面提示词，指定不想要的元素
      • 填写: 如"模糊，低质量，变形，多余的手指，nsfw"
      • 可选: 可以留空，但建议填写以提高图片质量

    • image_urls_input (STRING):

      • 描述: 参考图片的URL地址（多个URL用换行分隔）
      • 填写: 如"https://example.com/image1.jpg"
      • 用途: 提供风格参考或构图参考
      • 可选: 纯文生图时可留空

    平台开关

    • enable_hidream (BOOLEAN): 是否启用即梦AI (默认: True)
    • enable_volcengine (BOOLEAN): 是否启用火山引擎豆包 (默认: True)
    • enable_nanobanana (BOOLEAN): 是否启用NanoBanana (默认: True)

    API配置

    • hidream_req_key (STRING): 即梦AI的API密钥
    • volcengine_api_key (STRING): 火山引擎的API密钥
    • nanobanana_api_key (STRING): NanoBanana的API密钥

    生成参数

    • image_width (INT): 生成图片宽度 (默认: 1024, 范围: 256-2048)
    • image_height (INT): 生成图片高度 (默认: 1024, 范围: 256-2048)
    • num_images (INT): 生成图片数量 (默认: 1, 范围: 1-4)

    高级设置

    • request_timeout (INT): 单次请求超时时间，秒 (默认: 240)
    • poll_interval (INT): 轮询间隔时间，秒 (默认: 3)
    • poll_timeout (INT): 总轮询超时时间，秒 (默认: 240)
    • max_retries (INT): 最大重试次数 (默认: 3, 范围: 1-5)
    • white_image_width (INT): 失败时白色占位图宽度 (默认: 512)
    • white_image_height (INT): 失败时白色占位图高度 (默认: 512)

    特定平台参数

    • nanobanana_model (STRING): NanoBanana使用的模型 (默认: "flux")
    • nanobanana_style (STRING): NanoBanana的艺术风格 (默认: "默认")

    输出说明

    图片输出

    • nanobanana_image (IMAGE): NanoBanana生成的图片张量
    • doubao_image (IMAGE): 豆包生成的图片张量
    • hidream_image (IMAGE): 即梦AI生成的图片张量

    任务信息

    • nanobanana_task_id (STRING): NanoBanana任务ID
    • hidream_task_id (STRING): 即梦AI任务ID

    API响应

    • nanobanana_response (STRING): NanoBanana的完整API响应 (JSON格式)
    • doubao_response (STRING): 豆包的完整API响应 (JSON格式)
    • hidream_response (STRING): 即梦AI的完整API响应 (JSON格式)

    状态信息

    • status_info (STRING): 详细的状态信息，包含每个平台的执行结果

    图片URL输出 ⭐ 新功能

    • nanobanana_image_url (STRING): NanoBanana生成的图片URL
    • doubao_image_url (STRING): 豆包生成的图片URL
    • hidream_image_url (STRING): 即梦AI生成的图片URL

    URL值说明：

    • 成功时: 返回实际的图片URL地址
    • API错误时: 返回 "api_error"
    • API禁用时: 返回 "api_disabled"

    2. TripleAPIImg2Img (三合一图生图节点)

    用于基于输入图片进行图像转换和编辑。

    输入参数详解

    图片输入

    • input_image (IMAGE):
      • 描述: 输入的基础图片张量
      • 连接: 从图片加载节点或其他图像处理节点连接
      • 必填: 图生图必须提供基础图片

    基础参数

    • prompt (STRING):

      • 描述: 对图片进行修改的描述
      • 填写: 如"将这个人的头发改为金色，背景改为海滩"
      • 建议: 描述你想要的变化，而不是重复描述原图内容

    • negative_prompt (STRING):

      • 描述: 不想要的效果
      • 填写: 如"模糊，扭曲，多余的肢体"

    其他参数: 与文生图节点相同

    输出说明

    输出与文生图节点完全相同。

    URL输出功能详解

    什么是URL输出？

    URL输出功能为每个API生成的图片提供直接的URL链接，让您可以：

    • 直接访问图片：无需下载到本地即可查看生成的图片
    • 分享图片：将URL分享给他人查看
    • 集成到其他系统：将URL传递给其他应用或服务
    • 调试和监控：快速检查API是否成功生成了图片

    URL值含义

    | 值 | 含义 | 说明 | |---|---|---| | https://example.com/image.jpg | 成功 | 返回实际的图片URL地址 | | api_error | API错误 | API调用失败或响应格式错误 | | api_disabled | API禁用 | 该API被禁用或未提供密钥 |

    使用场景

    1. 质量对比：同时获取三个平台的图片URL，对比生成效果
    2. 自动化流程：将URL传递给后续的图片处理节点
    3. 错误诊断：通过URL值快速判断哪个API出现问题
    4. 批量处理：收集多个生成的图片URL进行批量操作

    使用示例

    文生图工作流

    1. 添加 TripleAPIGenerate 节点
    2. 填写提示词: "一只橘色的小猫在阳光下睡觉，温暖的午后，高清摄影"
    3. 填写负面提示词: "模糊，低质量，变形"
    4. 配置API密钥
    5. 设置图片尺寸 (如 1024x1024)
    6. 运行生成

    获取图片URL：

    • 连接 nanobanana_image_url 输出到文本显示节点，查看NanoBanana生成的图片URL
    • 连接 doubao_image_url 输出到文本显示节点，查看豆包生成的图片URL
    • 连接 hidream_image_url 输出到文本显示节点，查看即梦AI生成的图片URL

    图生图工作流

    1. 添加图片加载节点，加载基础图片
    2. 添加 TripleAPIImg2Img 节点
    3. 将图片加载节点的输出连接到 input_image
    4. 填写修改提示词: "将背景改为星空，保持人物不变"
    5. 配置其他参数
    6. 运行生成

    API密钥获取

    即梦AI

    1. 访问 即梦AI官网
    2. 注册账号并获取API密钥

    火山引擎豆包

    1. 访问 火山引擎控制台
    2. 开通豆包大模型服务
    3. 创建API密钥

    NanoBanana

    1. 访问 NanoBanana官网
    2. 注册账号并获取API访问权限