API를 사용하여 ComfyUI 호출하기

1. 그림 작업 제출하기

그림 작업을 제출하려면 다음 POST 요청을 사용해야 합니다:

POST /prompt

요청 매개변수:

• client_id string 작업 시작자를 식별하기 위해 클라이언트가 생성한 작업 ID
• prompt json 그림 매개변수를 포함하는 JSON 데이터

예시

{
  "client_id": "unique_client_id",
  "prompt": {
    "width": 768,
    "height": 512,
    "text": "아름다운 풍경"
  }
}

2. WebSocket을 사용하여 작업 상태 수신하기

작업을 제출한 후, WebSocket을 통해 실시간 업데이트를 받을 수 있습니다. 다음 주소에 연결하세요:

ws://<your_server>:<port>/ws?client_id=unique_client_id

연결되면 작업 실행 상태, 진행 상황 등에 대한 정보를 받게 됩니다.

데이터 형식

• 텍스트 데이터: 작업 변경 사항, 현재 실행 단계 및 진행 상황을 알리는 데 사용됩니다.
• 이진 데이터: 생성된 이미지 미리 보기를 전송하는 데 사용됩니다.

Python 예제 코드

다음은 Python 및 WebSocket 클라이언트 라이브러리를 사용하여 그림 요청을 제출하고 결과를 수신하는 예제 코드입니다:

import websocket
import json
import uuid

server_address = "127.0.0.1:8188"
client_id = str(uuid.uuid4())

def queue_prompt(prompt):
    p = {"client_id": client_id, "prompt": prompt}
    data = json.dumps(p).encode('utf-8')
    req = urllib.request.Request(f"http://{server_address}/prompt", data=data)
    response = urllib.request.urlopen(req)
    return json.loads(response.read())

def on_message(ws, message):
    print(f"수신한 메시지: {message}")

ws = websocket.WebSocketApp(f"ws://{server_address}/ws?client_id={client_id}",
                            on_message=on_message)
ws.run_forever()

# 그림 요청 제출하기
prompt_data = {
    "width": 768,
    "height": 512,
    "text": "아름다운 풍경"
}
queue_prompt(prompt_data)