chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: log out the status

Browse Source
This commit is contained in:
BennyKok 2024-01-12 20:30:45 +08:00
parent d0373ffb8b
commit c4c64a5f04
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
custom_routes.py
View File

@ -249,6 +249,7 @@ def update_run(prompt_id, status: Status):
"status": status.value, "status": status.value,
} }
prompt_metadata[prompt_id]['status'] = status prompt_metadata[prompt_id]['status'] = status
print(f"Status: {status.value}")
try: try:
requests.post(status_endpoint, json=body) requests.post(status_endpoint, json=body)
@ -371,7 +372,6 @@ async def update_file_status(prompt_id, data, uploading, have_error=False, node_
if uploading: if uploading:
if prompt_metadata[prompt_id]['status'] != Status.UPLOADING: if prompt_metadata[prompt_id]['status'] != Status.UPLOADING:
update_run(prompt_id, Status.UPLOADING) update_run(prompt_id, Status.UPLOADING)
print("Status: UPLOADING")
await send("uploading", { await send("uploading", {
"prompt_id": prompt_id, "prompt_id": prompt_id,
}) })