simplifications to the audio part

Author: andimarafioti

audio_streaming_client.py

@@ -51,25 +51,15 @@ def start(self):
 
         # Wait for the WebSocket to be ready
         self.ws_ready.wait()
+        self.start_audio_streaming()
 
-        send_thread = threading.Thread(target=self.send_audio)
-        play_thread = threading.Thread(target=self.play_audio)
+    def start_audio_streaming(self):
+        self.send_thread = threading.Thread(target=self.send_audio)
+        self.play_thread = threading.Thread(target=self.play_audio)
 
         with sd.InputStream(samplerate=self.args.sample_rate, channels=1, dtype='int16', callback=self.audio_input_callback, blocksize=self.args.chunk_size):
-            send_thread.start()
-            play_thread.start()
-
-            try:
-                input("Press Enter to stop streaming...")
-            except KeyboardInterrupt:
-                print("\nStreaming interrupted by user.")
-            finally:
-                self.stop_event.set()
-                send_thread.join()
-                play_thread.join()
-                self.ws.close()
-                ws_thread.join()
-                print("Audio streaming stopped.")
+            self.send_thread.start()
+            self.play_thread.start()
 
     def on_open(self, ws):
         print("WebSocket connection opened.")
@@ -92,6 +82,14 @@ def on_error(self, ws, error):
     def on_close(self, ws, close_status_code, close_msg):
         print("WebSocket connection closed.")
 
+    def on_shutdown(self):
+        self.stop_event.set()
+        self.send_thread.join()
+        self.play_thread.join()
+        self.ws.close()
+        self.ws_thread.join()
+        print("Service shutdown.")
+
     def send_audio(self):
         while not self.stop_event.is_set():
             if not self.send_queue.empty():