Update app.py

app.py

@@ -5,7 +5,10 @@ def hello():
     
     # Connect to the WebSocket
     websocket = websockets.connect(uri)
-    
+
+    # Establish the connection
+    websocket = websocket.__enter__()
+
     messages = [
         '{"fn_index":2,"session_hash":"kb46puzhzr"}',
         '{"data":["spider",1683852825,8,4,true],"event_data":null,"fn_index":2,"session_hash":"kb46puzhzr"}'
@@ -22,7 +25,7 @@ def hello():
         print(f"Received: {response}")
 
     # Close the WebSocket connection
-    websocket.close()
+    websocket.__exit__()
 
 if __name__ == "__main__":
     hello()