diff options
| -rw-r--r-- | Scripts/transcribe_v2.py | 34 |
1 files changed, 24 insertions, 10 deletions
diff --git a/Scripts/transcribe_v2.py b/Scripts/transcribe_v2.py index 6eba3e8..2639697 100644 --- a/Scripts/transcribe_v2.py +++ b/Scripts/transcribe_v2.py @@ -13,6 +13,7 @@ import argparse import ctranslate2 import editdistance import keybind_event_machine +import keyboard import langcodes import lang_compat import math @@ -29,6 +30,7 @@ import transformers import typing import vad import wave +import winsound class ThreadControl: def __init__(self, cfg): @@ -537,7 +539,7 @@ class Plugin: def transform(self, commit: TranscriptCommit) -> TranscriptCommit: return commit - def stop(): + def stop(self): pass class TranslationPlugin(Plugin): @@ -897,6 +899,11 @@ def vrInputThread(ctrl: ThreadControl): last_rising = time.time() last_medium_press_end = 0 + waveform0 = os.path.abspath("Resources/Sounds/Noise_On_Quiet.wav") + waveform1 = os.path.abspath("Resources/Sounds/Noise_Off_Quiet.wav") + waveform2 = os.path.abspath("Resources/Sounds/Dismiss_Noise_Quiet.wav") + waveform3 = os.path.abspath("Resources/Sounds/KB_Noise_Off_Quiet.wav") + button_generator = steamvr.pollButtonPress(hand=hand_id, button=button_id, ctrl=ctrl) while ctrl.run_app: @@ -924,12 +931,14 @@ def vrInputThread(ctrl: ThreadControl): if last_rising - last_medium_press_end < 1.0: # Type transcription if ctrl.cfg["enable_local_beep"]: - #audio_state.audio_events.append(audio_state.AUDIO_EVENT_UPDATE) + winsound.PlaySound(waveform3, winsound.SND_FILENAME | winsound.SND_ASYNC) pass - #keyboard.write(audio_state.filtered_text) + # TODO(yum) this is broken! Audio is not being collected + # while paused anymore. + #keyboard.write(ctrl.preview) else: if ctrl.cfg["enable_local_beep"]: - #audio_state.audio_events.append(audio_state.AUDIO_EVENT_TOGGLE_OFF) + winsound.PlaySound(waveform1, winsound.SND_FILENAME | winsound.SND_ASYNC) pass elif now - last_rising > 0.5: @@ -939,7 +948,7 @@ def vrInputThread(ctrl: ThreadControl): state = PAUSE_STATE if ctrl.cfg["enable_local_beep"]: - #audio_state.audio_events.append(audio_state.AUDIO_EVENT_DISMISS) + winsound.PlaySound(waveform2, winsound.SND_FILENAME | winsound.SND_ASYNC) pass if not ctrl.cfg["use_builtin"]: @@ -968,7 +977,7 @@ def vrInputThread(ctrl: ThreadControl): ctrl.stream.pause(True) if ctrl.cfg["enable_local_beep"]: - #audio_state.audio_events.append(audio_state.AUDIO_EVENT_TOGGLE_OFF) + winsound.PlaySound(waveform1, winsound.SND_FILENAME | winsound.SND_ASYNC) pass elif state == PAUSE_STATE: print("RECORDING", file=sys.stderr) @@ -994,7 +1003,7 @@ def vrInputThread(ctrl: ThreadControl): ctrl.pager.clear() if ctrl.cfg["enable_local_beep"]: - #audio_state.audio_events.append(audio_state.AUDIO_EVENT_TOGGLE_ON) + winsound.PlaySound(waveform0, winsound.SND_FILENAME | winsound.SND_ASYNC) pass def kbInputThread(ctrl: ThreadControl): @@ -1008,6 +1017,11 @@ def kbInputThread(ctrl: ThreadControl): PAUSE_STATE = 1 state = PAUSE_STATE + waveform0 = os.path.abspath("Resources/Sounds/Noise_On_Quiet.wav") + waveform1 = os.path.abspath("Resources/Sounds/Noise_Off_Quiet.wav") + waveform2 = os.path.abspath("Resources/Sounds/Dismiss_Noise_Quiet.wav") + waveform3 = os.path.abspath("Resources/Sounds/KB_Noise_Off_Quiet.wav") + while ctrl.run_app: time.sleep(0.01) @@ -1030,7 +1044,7 @@ def kbInputThread(ctrl: ThreadControl): state = PAUSE_STATE if ctrl.cfg["enable_local_beep"]: - #audio_state.audio_events.append(audio_state.AUDIO_EVENT_DISMISS) + winsound.PlaySound(waveform2, winsound.SND_FILENAME | winsound.SND_ASYNC) pass if not ctrl.cfg["use_builtin"]: @@ -1060,7 +1074,7 @@ def kbInputThread(ctrl: ThreadControl): ctrl.stream.pause(True) if ctrl.cfg["enable_local_beep"]: - #audio_state.audio_events.append(audio_state.AUDIO_EVENT_TOGGLE_OFF) + winsound.PlaySound(waveform1, winsound.SND_FILENAME | winsound.SND_ASYNC) pass elif state == PAUSE_STATE: print("RECORDING", file=sys.stderr) @@ -1086,7 +1100,7 @@ def kbInputThread(ctrl: ThreadControl): ctrl.pager.clear() if ctrl.cfg["enable_local_beep"]: - #audio_state.audio_events.append(audio_state.AUDIO_EVENT_TOGGLE_ON) + winsound.PlaySound(waveform0, winsound.SND_FILENAME | winsound.SND_ASYNC) pass def oscThread(ctrl: ThreadControl): |