### ComfyUI Python 图生图 JSON 调用 API 示例 为了在 Python 中使用 ComfyUI 实现图生图功能并通过 JSON 调用相关接口，可以参考以下方法。以下是详细的实现方式： #### 1. 环境准备 确保已经完成 ComfyUI 的安装并配置好运行环境[^1]。如果尚未安装必要的 Python 包，请执行以下命令来安装依赖项： ```bash pip install websocket-client requests ``` 这些库用于处理 WebSocket 和 HTTP 请求。 --- #### 2. 接口调用逻辑 ComfyUI 提供了一个基于 WebSocket 的 API 来控制其工作流。下面是一个完整的 Python 示例代码，展示如何通过 JSON 数据发送请求以实现图生图功能。 ##### 完整代码示例 ```python import json import websocket from threading import Thread def on_message(ws, message): """WebSocket消息回调函数""" try: msg_data = json.loads(message) if 'type' in msg_data and msg_data['type'] == 'executing': node_id = msg_data.get('node', {}).get('id') progress = msg_data.get('progress') print(f"Node ID: {node_id}, Progress: {progress}") elif 'type' in msg_data and msg_data['type'] == 'execution_result': output_node_id = msg_data.get('node', {}).get('id') result = msg_data.get('result', {}) images = result.get('images', []) for img_info in images: image_name = img_info.get('filename') subfolder = img_info.get('subfolder') full_image_path = f"http://localhost:8000/{subfolder}/{image_name}" if subfolder else "" print(f"Generated Image URL: {full_image_path}") except Exception as e: print(f"Error processing message: {e}") def on_error(ws, error): """错误处理回调函数""" print(f"WebSocket Error: {error}") def on_close(ws, close_status_code, close_msg): """关闭连接回调函数""" print("WebSocket Closed") def on_open(ws): """打开连接后的初始化操作""" def run(): request_payload = { "prompt": { "1": {"class_type": "ImageLoader", "inputs": {"image": "<base64_encoded_image>"}}, "2": {"class_type": "EmptyLatentImage", "inputs": {}}, "3": {"class_type": "CheckpointLoaderSimple", "inputs": {"ckpt_name": "model.ckpt"}}, "4": {"class_type": "KSampler", "inputs": {"latent": "2", "seed": 42}}, "5": {"class_type": "VAEDecode", "inputs": {"samples": "4"}}, "6": {"class_type": "SaveImage", "inputs": {"pixels": "5"}} }, "connections": [ ["1", "image", "5", "pixels"], ["2", "samples", "4", "samples"] ] } ws.send(json.dumps(request_payload)) thread = Thread(target=run) thread.start() if __name__ == "__main__": # 替换为您的实际 WebSocket 地址 websocket_url = "ws://127.0.0.1:8188/queue/join" ws = websocket.WebSocketApp( websocket_url, on_message=on_message, on_error=on_error, on_close=on_close ) ws.on_open = on_open # 启动 WebSocket 连接 ws.run_forever() ``` --- #### 3. 关键参数解释 - **`request_payload`**: 此对象定义了 ComfyUI 工作流中的节点及其输入数据结构。具体字段含义如下： - `class_type`: 表示节点的类型，例如 `"ImageLoader"` 或 `"CheckpointLoaderSimple"`。 - `inputs`: 每个节点所需的输入参数。对于图像加载器 (`ImageLoader`)，通常需要提供 Base64 编码的图片字符串。 - **`connections`**: 描述不同节点之间的连接关系。例如，将某个节点的输出作为另一个节点的输入。 - **Base64 图像编码**: 如果您希望传递一张现有图片作为输入，则需将其转换为 Base64 字符串形式。可以通过以下代码片段实现： ```python import base64 with open("input_image.png", "rb") as image_file: encoded_string = base64.b64encode(image_file.read()).decode('utf-8') print(encoded_string[:100]) # 打印前100字符验证 ``` --- #### 4. 输出解析 当服务器返回结果时，会包含生成图片的信息。路径类似于以下格式： ``` http://localhost:8000/<subfolder>/<filename> ``` 其中 `<subfolder>` 是存储子目录名称，而 `<filename>` 则是文件名。 --- ####