grizzly/interprete
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve interruption handling

Browse source
This commit is contained in:
grizzly 2026-03-01 21:42:37 -05:00
parent 82b86216e9
commit 45148327a6
1 changed files with 16 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

24
main.py
View file

@ -1,5 +1,6 @@
import sounddevice as sd import sounddevice as sd
import numpy as np import numpy as np
import time
import queue import queue
import threading import threading
import argparse import argparse
@ -12,7 +13,7 @@ def audio_callback(indata, frames, time, status):
print(f"Audio Status: {status}") print(f"Audio Status: {status}")
audio_queue.put(indata.copy()) audio_queue.put(indata.copy())
def transcribe_loop(): def transcribe_loop(stop_event):
"""Listen for chunks of audio, transcribe, and save to file.""" """Listen for chunks of audio, transcribe, and save to file."""
sample_rate = 16000 sample_rate = 16000
chunk_seconds = 3 chunk_seconds = 3
@ -23,9 +24,12 @@ def transcribe_loop():
buffer = [] buffer = []
print(">> Transcribing") print(">> Transcribing")
while True: while not stop_event.is_set():
buffer.append(audio_queue.get()) # get the latest data from the audio queue
try:
buffer.append(audio_queue.get(timeout=1)) # get the latest data from the audio queue
except:
continue
# check if there is enought audio for transcription # check if there is enought audio for transcription
total_samples = sum(c.shape[0] for c in buffer) total_samples = sum(c.shape[0] for c in buffer)
if total_samples >= sample_rate * chunk_seconds: if total_samples >= sample_rate * chunk_seconds:
@ -43,7 +47,7 @@ def transcribe_loop():
if text: if text:
print(text) print(text)
def listen(): def listen(stop_event):
print(">> Listening") print(">> Listening")
with sd.InputStream( with sd.InputStream(
samplerate=16000, samplerate=16000,
@ -53,20 +57,24 @@ def listen():
device=None, # default system device device=None, # default system device
callback=audio_callback): callback=audio_callback):
try: try:
threading.Event().wait() while not stop_event.is_set():
time.sleep(0.1)
except KeyboardInterrupt: except KeyboardInterrupt:
print("\nListening Thread Stopped") print("\nListening Thread Stopped")
def main(args): def main(args):
stop_event = threading.Event()
# Start listening thread # Start listening thread
t_listen = threading.Thread(target=listen) t_listen = threading.Thread(target=listen, args=(stop_event,), daemon=True)
t_listen.start() t_listen.start()
# Start transcription thread # Start transcription thread
t_transcribe = threading.Thread(target=transcribe_loop) t_transcribe = threading.Thread(target=transcribe_loop, args=(stop_event,), daemon=True)
t_transcribe.start() t_transcribe.start()
try: try:
threading.Event().wait() threading.Event().wait()
except KeyboardInterrupt: except KeyboardInterrupt:
@ -77,7 +85,7 @@ if __name__ == "__main__":
prog='Interprete', prog='Interprete',
description='Live EN->FR Voice-to-text Translation', description='Live EN->FR Voice-to-text Translation',
epilog='Well... go and use it now.') epilog='Well... go and use it now.')
parser.add_argument('-w', '--wmodel', default="base") parser.add_argument('-w', '--wmodel', default="small")
parser.add_argument('--whisperdevice', default="cpu") parser.add_argument('--whisperdevice', default="cpu")
args = parser.parse_args() args = parser.parse_args()