debug figure out timeout

client.py

@@ -50,7 +50,7 @@ async def send_heartbeat(websocket):
         except websockets.ConnectionClosed:
             print("Connection closed, stopping heartbeat")
             break
-        await asyncio.sleep(600)  # Send ping every 30 seconds (adjust as needed)
+        await asyncio.sleep(3)  # Send ping every 30 seconds (adjust as needed)
 
 
 async def run_client():
@@ -58,12 +58,15 @@ async def run_client():
     ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
     ssl_context.check_hostname = False
     ssl_context.verify_mode = ssl.CERT_NONE
-
-    async with websockets.connect(uri, ssl=ssl_context, ping_timeout=120,ping_interval=10) as websocket:
-        await asyncio.gather(
-            send_audio(websocket),
-            receive_transcription(websocket),
-            send_heartbeat(websocket)
-        )
+    while True:
+        try:
+            async with websockets.connect(uri, ssl=ssl_context, ping_timeout=120,ping_interval=None) as websocket:
+                await asyncio.gather(
+                    send_audio(websocket),
+                    receive_transcription(websocket),
+                    send_heartbeat(websocket)
+                )
+        except websockets.ConnectionClosedError as e:
+            print(f"web closed :{e}")
 
 asyncio.run(run_client())

infer.py

@@ -13,7 +13,7 @@ import sys
 import asyncio
 
 # Configure logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s: %(message)s',handlers=[logging.StreamHandler(sys.stdout)])
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(levelname)s: %(message)s',handlers=[logging.StreamHandler(sys.stdout)])
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 logging.info(f'Device selected: {device}')
@@ -192,19 +192,6 @@ async def websocket_transcribe(websocket: WebSocket):
     await websocket.accept()
     logging.info("WebSocket connection established successfully.")
 
-    async def send_ping():
-        """Function to send periodic ping to keep the connection alive."""
-        while True:
-            try:
-                await websocket.ping()
-                logging.info("Sent keepalive ping to client.")
-                await asyncio.sleep(10)  # Ping every 10 seconds (adjust the interval as needed)
-            except Exception as e:
-                logging.error(f"Error sending ping: {e}")
-                break
-
-    ping_task = asyncio.create_task(send_ping())
-
     try:
         processed_segments = []  # Keeps track of the segments already transcribed
         accumulated_audio_size = 0  # Track how much audio data has been buffered