30 · API 与自动化工作流

http://127.0.0.1:8188

import json
import requests
import time
 
# 1. 加载工作流 JSON
with open("my_workflow.json", "r") as f:
    workflow = json.load(f)
 
# 2. 你可以动态修改工作流里的参数
# 比如改提示词
for node_id, node in workflow.items():
    if node["class_type"] == "CLIPTextEncode":
        node["inputs"]["text"] = "新的提示词"
 
# 3. 提交任务
resp = requests.post(
    "http://127.0.0.1:8188/prompt",
    json={"prompt": workflow}
)
prompt_id = resp.json()["prompt_id"]
 
# 4. 轮询等结果
while True:
    history = requests.get(f"http://127.0.0.1:8188/history/{prompt_id}").json()
    if prompt_id in history:
        break
    time.sleep(1)
 
print("生成完成！")

prompts = ["猫", "狗", "兔子", "鸟", "鱼"]
for p in prompts:
    workflow = load_template()
    set_prompt(workflow, p)
    submit_and_wait(workflow)

for cfg in [3, 5, 7, 9, 12]:
    for sampler in ["euler", "dpmpp_2m", "dpmpp_sde"]:
        workflow = load_template()
        set_param(workflow, "cfg", cfg)
        set_param(workflow, "sampler", sampler)
        submit_and_wait(workflow)

from fastapi import FastAPI
import requests
 
app = FastAPI()
 
@app.post("/generate")
async def generate(prompt: str):
    workflow = load_template()
    set_prompt(workflow, prompt)
    result = submit_comfyui(workflow)
    return {"status": "done", "image_url": result}