fix: logic of adding of pcm_array to diarization_queue

This commit is contained in:
choomegan
2025-07-15 15:27:44 +08:00
parent 156b9a133f
commit 64e44fb24f

View File

@@ -192,12 +192,6 @@ class AudioProcessor:
continue
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
if len(self.pcm_buffer) >= self.bytes_per_sec:
@@ -214,7 +208,11 @@ class AudioProcessor:
# Send to transcription if enabled
if self.args.transcription and self.transcription_queue:
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
if not self.args.transcription and not self.args.diarization:
await asyncio.sleep(0.1)