ComfyUI web API 接口 python 调用工具 简单实现

提示: 本文适合熟悉"confyUI"工作流希望自动化生图的人,并且能读懂和执行python代码的人以及"AI程序开发者"阅读

主要想法和初衷:

1. 通过python 程序调用comfyUI workflow 生成图片
2. 批量替换workflow 中的prompt 节点, 批量生成图片.
3. 自动化, 一键完成, 无需等待图片生成, 让GPU 干活, 我们睡觉.
4. 图片生成后, 自动保存到指定文件夹, 方便后续使用.
   

使用条件和步骤:

1. 本地安装comfyUI 或 调用远程server 上的comfyUI
2. 有一个在comfyUi上可用的工作流,测试可以生成图片.
3. 开启DEV模式 在comfyUI功能栏上点击小齿轮，在弹出的页面中开启DEV模式.
4. 保存工作流到本地. workdir, 运行程序获取图片.
5. 如果需要更换提示词需要下面步骤.

   根据工作流, 找到需要替换的节点, 并进行替换相应的代码
   生成一个批量替换的提示词, 并保存到xlsx 文件中.
   运行python 程序, 自动替换节点, 并生成图片.

ComfyUI web API 接口 python 调用工具:

(参考程序https://zhuanlan.zhihu.com/p/686893291 进行改进)

• ComfyUI提供了一套强大的API接口，方便开发者进行绘图任务的提交、查询历史任务、获取绘图结果等操作。
• 以下包装了一个ComfyUI API 接口调用的tool 类 方便进行工作流调用,并获取结果。
• 这个类的基本功能是：

  初始化ComfyUI API 接口的URL
  提交绘图任务
  查询历史任务
  获取下载绘图结果

# ! pip install websocket-client
import json
import websocket  
import uuid
import urllib.request
import urllib.parse
import random
import pandas as pd

COMFYUI_ENDPOINT = '127.0.0.1:8188'

class ComfyUITool:
    def __init__(self, server_address, seed, workflowfile, working_dir):
        self.server_address = server_address
        self.client_