yumoqing/rtcllm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2024-09-18 16:01:08 +08:00
parent 492df342a6
commit 54c9b4c14f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rtcllm/vad.py
View File

@ -4,6 +4,7 @@ import asyncio
import collections
import contextlib
from appPublic.folderUtils import temp_file
from appPublic.worker import awaitify
from aiortc import MediaStreamTrack
from aiortc.contrib.media import MediaBlackhole, MediaPlayer, MediaRecorder, MediaRelay
import webrtcvad
@ -137,6 +138,7 @@ class AudioTrackVad(MediaStreamTrack):
super().__init__()
self.track = track
self.vad = MyVad(callback=onvoiceend)
self.vad_check = awaitify(self.vad.vad_check)
# self.sample_rate = self.track.getSettings().sampleRate
# frameSize = self.track.getSettings().frameSize
# self.frame_duration_ms = (1000 * frameSize) / self.sample_rate
@ -165,7 +167,7 @@ class AudioTrackVad(MediaStreamTrack):
try:
frames = to16000_160_frames(frame)
for frame in frames:
await self.vad.vad_check(frame)
await self.vad_check(frame)
except Exception as e:
print(f'{e=}')
print_exc()