From d6edf8e847b0a6ae261d2ea4de88d99d745393e2 Mon Sep 17 00:00:00 2001
From: makaveli10 <suryanvineet47@gmail.com>
Date: Wed, 7 Feb 2024 23:29:22 +0530
Subject: [PATCH] update on_error; on_close

---
 whisper_live/client.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/whisper_live/client.py b/whisper_live/client.py
index f8df387..15850a0 100644
--- a/whisper_live/client.py
+++ b/whisper_live/client.py
@@ -233,10 +233,15 @@ class Client:
             print(element)
 
     def on_error(self, ws, error):
-        print(error)
+        print(f"[ERROR] WebSocket Error: {error}")
+        self.server_error = True
+        self.error_message = error
 
     def on_close(self, ws, close_status_code, close_msg):
         print(f"[INFO]: Websocket connection closed: {close_status_code}: {close_msg}")
+        self.recording = False
+        self.server_error = False
+        self.waiting = False
 
     def on_open(self, ws):
         """