### ComfyUI JSON API 实现图生图功能 以下是使用 Python 调用 ComfyUI 的 JSON API 来实现图生图功能的完整代码示例。该代码涵盖了读取 JSON 文件以及加载本地图片的核心操作。 #### 1. 安装必要的依赖项 为了完成此任务，需要安装 `requests` 库用于发送 HTTP 请求，以及 `Pillow` 库来处理图像文件。 ```bash pip install requests pillow ``` #### 2. 完整代码示例 以下是一个完整的 Python 脚本，展示如何通过 ComfyUI 的 JSON API 进行图生图的功能： ```python import os import json import base64 from PIL import Image import io import requests def load_image(image_path): """加载本地图片并将其转换为 Base64 编码""" with open(image_path, "rb") as image_file: encoded_string = base64.b64encode(image_file.read()).decode('utf-8') return f"data:image/png;base64,{encoded_string}" def read_json(json_file): """读取 JSON 配置文件""" with open(json_file, 'r') as file: data = json.load(file) return data def send_request(api_url, payload): """向 ComfyUI 发送 POST 请求""" headers = {'Content-Type': 'application/json'} response = requests.post(api_url, headers=headers, json=payload) if response.status_code != 200: raise Exception(f"请求失败: {response.text}") result_data = response.json() output_images = result_data.get("images", []) if not output_images: raise ValueError("未返回任何生成的图像") return output_images[0] def save_base64_image(base64_str, output_path): """保存 Base64 编码的图像到指定路径""" img_bytes = base64.b64decode(base64_str.split(',')[1]) img = Image.open(io.BytesIO(img_bytes)) img.save(output_path) if __name__ == "__main__": # 参数配置 comfy_ui_api_url = "http://localhost:8188/prompt" input_json_file = "./input_config.json" # 输入 JSON 文件路径 local_image_path = "./example_input.png" # 输入本地图片路径 output_image_path = "./generated_output.png" # 输出生成图片路径 # 步骤一：读取 JSON 文件 prompt_payload = read_json(input_json_file)[^4] # 步骤二：加载本地图片 init_image_b64 = load_image(local_image_path)[^5] prompt_payload["inputs"]["image"] = init_image_b64 # 步骤三：发送请求至 ComfyUI 并接收响应 generated_image_b64 = send_request(comfy_ui_api_url, prompt_payload)[^6] # 步骤四：保存生成的图像 save_base64_image(generated_image_b64, output_image_path)[^7] ``` --- #### 关键说明 1. **JSON 文件结构** 提供给 ComfyUI 的 JSON 文件应包含节点定义及其输入参数。例如： ```json { "prompt": { "1": {"class_type": "ImageLoader"}, "2": {"class_type": "CheckpointLoaderSimple"} }, "workflow": [ ["1", "out", "2", "in"] ] } ``` 2. **Base64 图片编码** 使用 `base64` 将本地图片转为字符串形式传递给 ComfyUI API[^5]。 3. **API URL 地址** 默认情况下，ComfyUI 启动后会监听 `http://localhost:8188` 端口上的 `/prompt` 接口[^6]。 4. **错误处理** 如果服务器端出现问题或无法解析请求，则抛出异常提示用户检查环境设置和数据格式[^7]。 --- ###