fix(client): improve error handling and readability

Author: janaab11

src/diart/console/client.py

@@ -64,8 +64,12 @@ def run():
         type=str,
         help="Path to an audio file | 'microphone' | 'microphone:<DEVICE_ID>'",
     )
-    parser.add_argument("--host", required=True, type=str, help="Server host")
-    parser.add_argument("--port", required=True, type=int, help="Server port")
+    parser.add_argument(
+        "--host", default="0.0.0.0", type=str, help="Server host. Defaults to 0.0.0.0"
+    )
+    parser.add_argument(
+        "--port", default=7007, type=int, help="Server port. Defaults to 7007"
+    )
     parser.add_argument(
         "--step", default=0.5, type=float, help=f"{argdoc.STEP}. Defaults to 0.5"
     )
@@ -93,7 +97,7 @@ def run():
 
         # Wait for READY signal from server
         print("Waiting for server to be ready...", end="", flush=True)
-        while True:
+        while not stop_event.is_set():
             try:
                 message = ws.recv()
                 if message.strip() == "READY":
@@ -113,10 +117,19 @@ def run():
         sender.start()
         receiver.start()
 
+        try:
+            # Wait for threads to complete or for keyboard interrupt
+            sender.join()
+            receiver.join()
+        except KeyboardInterrupt:
+            print("\nShutting down...")
+            stop_event.set()
+
     except Exception as e:
         print(f"Error: {e}")
-        stop_event.set()
+
     finally:
+        stop_event.set()
         try:
             ws.close()
         except: