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

fix: output node status

Browse Source
This commit is contained in:
bennykok 2024-10-10 11:20:12 -07:00
parent 44a91d2093
commit a139424b91
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
custom_routes.py
View File

@ -1009,6 +1009,12 @@ async def send_json_override(self, event, data, sid=None):
asyncio.create_task(self.send_json_original(event, data, sid)) asyncio.create_task(self.send_json_original(event, data, sid))
]) ])
if prompt_id in comfy_message_queues:
comfy_message_queues[prompt_id].put_nowait({
"event": event,
"data": data
})
asyncio.create_task(update_run_ws_event(prompt_id, event, data)) asyncio.create_task(update_run_ws_event(prompt_id, event, data))
if event == 'execution_start': if event == 'execution_start':
@ -1083,17 +1089,17 @@ async def send_json_override(self, event, data, sid=None):
} }
if class_type == "PreviewImage": if class_type == "PreviewImage":
logger.info("Skipping preview image") logger.info("Skipping preview image")
return else:
await update_run_with_output(prompt_id, data.get('output'), node_id=data.get('node'), node_meta=node_meta) await update_run_with_output(prompt_id, data.get('output'), node_id=data.get('node'), node_meta=node_meta)
if prompt_id in comfy_message_queues:
comfy_message_queues[prompt_id].put_nowait({
"event": "output_ready",
"data": data
})
logger.info(f"Executed {class_type} {data}") logger.info(f"Executed {class_type} {data}")
else: else:
logger.info(f"Executed {data}") logger.info(f"Executed {data}")
if prompt_id in comfy_message_queues:
comfy_message_queues[prompt_id].put_nowait({
"event": event,
"data": data
})
# Global variable to keep track of the last read line number # Global variable to keep track of the last read line number
last_read_line_number = 0 last_read_line_number = 0