fixed ui not updating

I5UCC · Jan 3, 2023 · 8457edb · 8457edb
1 parent be3d097
commit 8457edb
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/src/TextboxSTT.py b/src/TextboxSTT.py
@@ -63,10 +63,12 @@ def flush(self):
 def set_status_label(text, color):
     status_lbl.configure(text=text)
     color_lbl.configure(bg=color)
+    ui.update()
 
 def set_text_label(text):
     text = text[:144]
     text_lbl.configure(text=text)
+    ui.update()
 
 VRC_INPUT_PARAM = "/chatbox/input"
 VRC_TYPING_PARAM = "/chatbox/typing"
@@ -118,6 +120,8 @@ def play_sound(filename):
     actionSetHandle = openvr.VRInput().getActionSetHandle(ACTIONSETHANDLE)
     buttonactionhandle = openvr.VRInput().getActionHandle(STTLISTENHANDLE)
     ovr_initialized = True
+    print("INITIALZIED")
+    set_status_label("INITIALZIED", "green")
 except Exception:
     ovr_initialized = False
 
@@ -140,6 +144,7 @@ def listen_and_transcribe():
         oscClient.send_message(VRC_TYPING_PARAM, True)
         print("TRANSCRIBING")
         set_status_label("TRANSCRIBING", "orange")
+
         if lang:
             result = audio_model.transcribe(torch_audio, language=lang)
         else:
@@ -158,17 +163,17 @@ def send_message():
         set_status_label("TRANSCRIBING", "#ff8800")
         oscClient.send_message(VRC_INPUT_PARAM, [trans, True, True])
         oscClient.send_message(VRC_TYPING_PARAM, False)
-        print("WAITING")
-        set_status_label("WAITING", "#00008b")
+        print("WAITING FOR INPUT")
+        set_status_label("WAITING FOR INPUT", "#00008b")
 
 
 def clear_chatbox():
     print("CLEARING OSC TEXTBOX")
     set_status_label("CLEARING OSC TEXTBOX", "#e0ffff")
     oscClient.send_message(VRC_INPUT_PARAM, ["", True])
     oscClient.send_message(VRC_TYPING_PARAM, False)
-    print("WAITING")
-    set_status_label("WAITING", "#00008b")
+    print("WAITING FOR INPUT")
+    set_status_label("WAITING FOR INPUT", "#00008b")
     set_text_label("---")
 
 
@@ -195,6 +200,7 @@ def handle_input():
         while pressed:
             if time.time() - curr_time > float(config["hold_time"]):
                 clear_chatbox()
+                set_status_label("CLEARED - WAITING FOR INPUT", "#000054")
                 held = True
                 break
             pressed = get_action_bstate()
@@ -212,10 +218,8 @@ def handle_input():
 keyboard.add_hotkey(config["record_hotkey"], send_message)
 keyboard.add_hotkey(config["clear_hotkey"], clear_chatbox)
 cls()
-print("-INITIALZIED-")
-set_status_label("INITIALZIED", "green")
-print("WAITING")
-set_status_label("WAITING", "#00008b")
+print("WAITING FOR INPUT")
+set_status_label("WAITING FOR INPUT", "#00008b")
 if ovr_initialized:
     try:
         ui.after(50, handle_input)