From 6e923dbc74a6a4190ad9a84a032ec204e91b8c2e Mon Sep 17 00:00:00 2001 From: yumoqing Date: Sat, 14 Sep 2024 12:08:08 +0800 Subject: [PATCH] bugfix --- rtcllm/vad.py | 23 +++++++++-------------- 1 file changed, 9 insertions(+), 14 deletions(-) diff --git a/rtcllm/vad.py b/rtcllm/vad.py index 6d627cb..7c32c01 100644 --- a/rtcllm/vad.py +++ b/rtcllm/vad.py @@ -47,21 +47,16 @@ class AudioTrackVad(MediaStreamTrack): return audio_bytes async def recv(self): - oldf = await self.track.recv() - self.sample_rate = oldf.sample_rate - duration = (oldf.samples * 1000) / oldf.sample_rate - # print(f'{self.__class__.__name__}.recv(): {duration=}, {oldf.samples=}, {oldf.sample_rate=}') + frame = await self.track.recv() + self.sample_rate = frame.sample_rate + duration = (frame.samples * 1000) / frame.sample_rate + # print(f'{self.__class__.__name__}.recv(): {duration=}, {frame.samples=}, {frame.sample_rate=}') try: - for f in frames: - if self.debug: - self.debug = False - # print(f'{type(f)}, {f.samples=}, {f.format.bytes=}, {f.sample_rate=}, {f.format=}, {f.is_corrupt=}, {f.layout=}, {f.planes=}, {f.side_data=}') - try: - await self.vad_check(f) - except Exception as e: - print(f'{e=}') - print_exc() - return + await self.vad_check(f) + except Exception as e: + print(f'{e=}') + print_exc() + return if self.task: self.task.cancel() if self.running: