admin-valentin/WhisperLiveKit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #147 from choomegan/diar_queue

Browse source 
Ensure diarization_queue receives only latest PCM chunk
This commit is contained in:
Quentin Fuxa 2025-07-16 12:04:53 +02:00 committed by GitHub
commit 2ca1156b7e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
whisperlivekit/audio_processor.py
View file

@ -193,12 +193,6 @@ class AudioProcessor:
self.pcm_buffer.extend(chunk) self.pcm_buffer.extend(chunk)
# Send to diarization if enabled
if self.args.diarization and self.diarization_queue:
await self.diarization_queue.put(
self.convert_pcm_to_float(self.pcm_buffer).copy()
)
# Process when enough data # Process when enough data
if len(self.pcm_buffer) >= self.bytes_per_sec: if len(self.pcm_buffer) >= self.bytes_per_sec:
if len(self.pcm_buffer) > self.max_bytes_per_sec: if len(self.pcm_buffer) > self.max_bytes_per_sec:
@ -215,6 +209,10 @@ class AudioProcessor:
if self.args.transcription and self.transcription_queue: if self.args.transcription and self.transcription_queue:
await self.transcription_queue.put(pcm_array.copy()) await self.transcription_queue.put(pcm_array.copy())
# Send to diarization if enabled
if self.args.diarization and self.diarization_queue:
await self.diarization_queue.put(pcm_array.copy())
# Sleep if no processing is happening # Sleep if no processing is happening
if not self.args.transcription and not self.args.diarization: if not self.args.transcription and not self.args.diarization:
await asyncio.sleep(0.1) await asyncio.sleep(0.1)