From 86b97d0039cc559a4902605f7f12053cac4fc068 Mon Sep 17 00:00:00 2001 From: SevaSk Date: Sat, 13 May 2023 10:46:05 -0400 Subject: [PATCH] minor refactoring --- AudioTranscriber.py | 14 +++++++------- main.py | 2 +- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/AudioTranscriber.py b/AudioTranscriber.py index 042e70a..667e501 100644 --- a/AudioTranscriber.py +++ b/AudioTranscriber.py @@ -16,24 +16,24 @@ PHRASE_TIMEOUT = 3.01 MAX_PHRASES = 10 class AudioTranscriber: - def __init__(self, default_mic : DefaultMicRecorder, default_speaker : DefaultSpeakerRecorder): + def __init__(self, mic_source, speaker_source): self.transcript_data = {"You": [], "Speaker": []} self.transcript_changed_event = threading.Event() self.audio_model = whisper.load_model(os.path.join(os.getcwd(), 'tiny.en.pt')) self.audio_sources = { "You": { - "sample_rate": default_mic.source.SAMPLE_RATE, - "sample_width": default_mic.source.SAMPLE_WIDTH, - "channels": default_mic.source.channels, + "sample_rate": mic_source.SAMPLE_RATE, + "sample_width": mic_source.SAMPLE_WIDTH, + "channels": mic_source.channels, "last_sample": bytes(), "last_spoken": None, "new_phrase": True, "process_data_func": self.process_mic_data }, "Speaker": { - "sample_rate": default_speaker.source.SAMPLE_RATE, - "sample_width": default_speaker.source.SAMPLE_WIDTH, - "channels": default_speaker.source.channels, + "sample_rate": speaker_source.SAMPLE_RATE, + "sample_width": speaker_source.SAMPLE_WIDTH, + "channels": speaker_source.channels, "last_sample": bytes(), "last_spoken": None, "new_phrase": True, diff --git a/main.py b/main.py index f294ed7..793eb42 100644 --- a/main.py +++ b/main.py @@ -72,7 +72,7 @@ def main(): speaker_audio_recorder = AudioRecorder.DefaultSpeakerRecorder() speaker_audio_recorder.record_into_queue(audio_queue) - global_transcriber = AudioTranscriber(user_audio_recorder, speaker_audio_recorder) + global_transcriber = AudioTranscriber(user_audio_recorder.source, speaker_audio_recorder.source) transcribe = threading.Thread(target=global_transcriber.transcribe_audio_queue, args=(audio_queue,)) transcribe.start()