deepgeek/gradio-webrtc
2
0
Fork 0
mirror of https://github.com/HumanAIGC-Engineering/gradio-webrtc.git synced 2026-02-05 18:09:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add code (#276)

Browse Source
This commit is contained in:
Freddy Boulton
2025-04-14 09:57:15 -04:00
committed by GitHub
parent 5835e74377
commit 54d07bc3c8
2 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
backend/fastrtc/websocket.py
View File

@@ -15,9 +15,11 @@ from .tracks import AsyncStreamHandler, StreamHandlerImpl
from .utils import (
AdditionalOutputs,
CloseStream,
Context,
DataChannel,
audio_to_float32,
audio_to_int16,
current_context,
split_output,
)
@@ -141,7 +143,7 @@ class WebSocketHandler:
)
else:
await run_sync(
self.stream_handler.receive,
self.receive_with_context,
(self.stream_handler.input_sample_rate, audio_array),
)
except Exception as e:
@@ -156,6 +158,7 @@ class WebSocketHandler:
self.stream_id = cast(str, message["streamSid"])
else:
self.stream_id = cast(str, message["websocket_id"])
current_context.set(Context(webrtc_id=self.stream_id))
self.set_additional_outputs = self.set_additional_outputs_factory(
self.stream_id
)
@@ -186,13 +189,21 @@ class WebSocketHandler:
self.clean_up(cast(str, self.stream_id))
def emit_with_context(self):
current_context.set(Context(webrtc_id=cast(str, self.stream_id)))
return self.stream_handler.emit()
def receive_with_context(self, frame: tuple[int, np.ndarray]):
current_context.set(Context(webrtc_id=cast(str, self.stream_id)))
return self.stream_handler.receive(frame)
async def _emit_to_queue(self):
try:
while not self.quit.is_set():
if isinstance(self.stream_handler, AsyncStreamHandler):
output = await self.stream_handler.emit()
else:
output = await run_sync(self.stream_handler.emit)
output = await run_sync(self.emit_with_context)
self.queue.put_nowait(output)
except asyncio.CancelledError:
logger.debug("Emit loop cancelled")
@@ -270,7 +281,7 @@ class WebSocketHandler:
audio_payload = base64.b64encode(mulaw_audio).decode("utf-8")
if self.websocket and self.stream_id:
sample_rate, audio_array = frame
sample_rate, audio_array = frame[:2]
duration = len(audio_array) / sample_rate
self.playing_durations.append(duration)