Update main.py

main.py

@@ -10,7 +10,6 @@ logger = logging.getLogger(__name__)
 app = FastAPI()
 
 WEBSOCKET_URL = "wss://b917-160-179-178-105.ngrok-free.app/ws"
-
 call_count = 0
 
 async def websocket_client():
@@ -24,18 +23,16 @@ async def websocket_client():
                     payload = {"count": call_count}
                     await websocket.send(json.dumps(payload))
                     logger.info(f"Sent: {payload}")
-
                     response = await websocket.recv()
                     data = json.loads(response)
                     logger.info(f"Received: {data}")
-
-                    await asyncio.sleep(1)  # Adjust this delay as needed
+                    await asyncio.sleep(1)  # Keep this delay as 1 second between messages
         except websockets.exceptions.ConnectionClosed:
             logger.error("WebSocket connection closed. Retrying...")
-            await asyncio.sleep(5)
+            await asyncio.sleep(0.1)  # 100ms delay before retrying
         except Exception as e:
             logger.error(f"Error: {e}")
-            await asyncio.sleep(5)
+            await asyncio.sleep(0.1)  # 100ms delay before retrying
 
 @app.on_event("startup")
 async def startup_event():