mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-18 14:32:49 +08:00
fix: do not spam frontend with each chunk arrival
Signed-off-by: bigcat88 <bigcat88@icloud.com>
This commit is contained in:
parent
06c108ce0e
commit
2070c99c14
@ -180,6 +180,7 @@ async def _stream_sonilo_music(
|
|||||||
|
|
||||||
node_id = get_node_id(cls)
|
node_id = get_node_id(cls)
|
||||||
start_ts = time.monotonic()
|
start_ts = time.monotonic()
|
||||||
|
last_chunk_status_ts = 0.0
|
||||||
audio_streams: dict[int, list[bytes]] = {}
|
audio_streams: dict[int, list[bytes]] = {}
|
||||||
title: str | None = None
|
title: str | None = None
|
||||||
|
|
||||||
@ -242,14 +243,17 @@ async def _stream_sonilo_music(
|
|||||||
audio_streams[stream_idx] = []
|
audio_streams[stream_idx] = []
|
||||||
audio_streams[stream_idx].append(chunk_data)
|
audio_streams[stream_idx].append(chunk_data)
|
||||||
|
|
||||||
total_chunks = sum(len(chunks) for chunks in audio_streams.values())
|
now = time.monotonic()
|
||||||
elapsed = int(time.monotonic() - start_ts)
|
if now - last_chunk_status_ts >= 1.0:
|
||||||
status_lines = ["Status: Receiving audio"]
|
total_chunks = sum(len(chunks) for chunks in audio_streams.values())
|
||||||
if title:
|
elapsed = int(now - start_ts)
|
||||||
status_lines.append(f"Title: {title}")
|
status_lines = ["Status: Receiving audio"]
|
||||||
status_lines.append(f"Chunks received: {total_chunks}")
|
if title:
|
||||||
status_lines.append(f"Time elapsed: {elapsed}s")
|
status_lines.append(f"Title: {title}")
|
||||||
PromptServer.instance.send_progress_text("\n".join(status_lines), node_id)
|
status_lines.append(f"Chunks received: {total_chunks}")
|
||||||
|
status_lines.append(f"Time elapsed: {elapsed}s")
|
||||||
|
PromptServer.instance.send_progress_text("\n".join(status_lines), node_id)
|
||||||
|
last_chunk_status_ts = now
|
||||||
elif evt_type == "complete":
|
elif evt_type == "complete":
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user