wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-18 10:20:16 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Merge branch 'comfyanonymous:master' into master

Browse Source
This commit is contained in:
patientx 2025-07-15 00:19:16 +03:00 committed by GitHub
commit d16487f4e6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 32 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -1,5 +1,5 @@
comfyui-frontend-package==1.23.4 comfyui-frontend-package==1.23.4
comfyui-workflow-templates==0.1.35 comfyui-workflow-templates==0.1.36
comfyui-embedded-docs==0.2.4 comfyui-embedded-docs==0.2.4
torch torch
torchsde torchsde

11
script_examples/websockets_api_example.py
View File

@ -10,11 +10,11 @@ import urllib.parse
server_address = "127.0.0.1:8188" server_address = "127.0.0.1:8188"
client_id = str(uuid.uuid4()) client_id = str(uuid.uuid4())
def queue_prompt(prompt): def queue_prompt(prompt, prompt_id):
p = {"prompt": prompt, "client_id": client_id} p = {"prompt": prompt, "client_id": client_id, "prompt_id": prompt_id}
data = json.dumps(p).encode('utf-8') data = json.dumps(p).encode('utf-8')
req = urllib.request.Request("http://{}/prompt".format(server_address), data=data) req = urllib.request.Request("http://{}/prompt".format(server_address), data=data)
return json.loads(urllib.request.urlopen(req).read()) urllib.request.urlopen(req).read()
def get_image(filename, subfolder, folder_type): def get_image(filename, subfolder, folder_type):
data = {"filename": filename, "subfolder": subfolder, "type": folder_type} data = {"filename": filename, "subfolder": subfolder, "type": folder_type}
@ -27,7 +27,8 @@ def get_history(prompt_id):
return json.loads(response.read()) return json.loads(response.read())
def get_images(ws, prompt): def get_images(ws, prompt):
prompt_id = queue_prompt(prompt)['prompt_id'] prompt_id = str(uuid.uuid4())
queue_prompt(prompt, prompt_id)
output_images = {} output_images = {}
while True: while True:
out = ws.recv() out = ws.recv()

2
server.py
View File

@ -678,7 +678,7 @@ class PromptServer():
if "prompt" in json_data: if "prompt" in json_data:
prompt = json_data["prompt"] prompt = json_data["prompt"]
prompt_id = str(uuid.uuid4()) prompt_id = str(json_data.get("prompt_id", uuid.uuid4()))
valid = await execution.validate_prompt(prompt_id, prompt) valid = await execution.validate_prompt(prompt_id, prompt)
extra_data = {} extra_data = {}
if "extra_data" in json_data: if "extra_data" in json_data: