Update app.py

app.py

@@ -1,31 +1,28 @@
 import websockets
+import asyncio
 
-def hello():
+async def hello():
     uri = "ws://localhost:8765"
-    
+
     # Connect to the WebSocket
-    websocket = websockets.connect(uri)
+    async with websockets.connect(uri) as websocket:
+        messages = [
+            '{"fn_index":2,"session_hash":"kb46puzhzr"}',
+            '{"data":["spider",1683852825,8,4,true],"event_data":null,"fn_index":2,"session_hash":"kb46puzhzr"}'
+        ]
 
-    # Establish the connection
-    websocket = websocket.__enter__()
+        for message in messages:
+            print(f"Sending: {message}")
 
-    messages = [
-        '{"fn_index":2,"session_hash":"kb46puzhzr"}',
-        '{"data":["spider",1683852825,8,4,true],"event_data":null,"fn_index":2,"session_hash":"kb46puzhzr"}'
-    ]
+            # Send the message
+            await websocket.send(message)
 
-    for message in messages:
-        print(f"Sending: {message}")
-        
-        # Send the message
-        websocket.send(message)
-        
-        # Receive the response
-        response = websocket.recv()
-        print(f"Received: {response}")
+            # Receive the response
+            response = await websocket.recv()
+            print(f"Received: {response}")
 
-    # Close the WebSocket connection
-    websocket.__exit__()
+async def main():
+    await hello()
 
 if __name__ == "__main__":
-    hello()
+    asyncio.run(main())