Update app.py

app.py

@@ -1,38 +1,44 @@
-import threading
-import asyncio
 import websockets
+import asyncio
+import aioconsole
+import sys
 
+URL = "wss://ramesh-vani-collab.hf.space"
 
-async def websocket_consumer(websocket, path):
+async def receiver(ws,connection):
     
-    try:
-        code = await websocket.recv()
-        print(code)
-        async with websockets.connect('wss://ramesh-vani-wspython.hf.space') as wssocket:
-            
-            
-            await wssocket.send(code)
-            print('code sent')
-            while True:
-                print('loop')
-                response = await wssocket.recv()
-                print(response)
-                await websocket.send(response)
-        
-    except websockets.exceptions.ConnectionClosedOK:
-         print('error')
+        try:
+          async for msg in connection:
+            # msg = await connection.recv()
+            print(f"\nServer: {msg}")
+            await ws.send(msg)
+        except websockets.exceptions.WebSocketException as e:
+            print(f"WebSocket exception: {e}")
+            # sys.exit(1)
 
-class WebSocketThread(threading.Thread):
-    def run(self):
-        asyncio.set_event_loop(asyncio.new_event_loop())
-        start_server = websockets.serve(websocket_consumer, '0.0.0.0', 7860)
-        asyncio.get_event_loop().run_until_complete(start_server)
-        asyncio.get_event_loop().run_forever()
+async def sender(ws,connection):
+    
+        try:
+          async for msg in ws:
+            # msg = await ws.recv()
+            await connection.send(msg)
+        except websockets.exceptions.WebSocketException as e:
+            print(f"WebSocket exception: {e}")
+            # sys.exit(1)
+            
 
-# websocket_thread = WebSocketThread()
-# websocket_thread.start()
-# asyncio.set_event_loop(asyncio.new_event_loop())
-start_server = websockets.serve(websocket_consumer, "0.0.0.0", 7860)
+async def chat(websocket, path) -> None:
+    try:
+        async with websockets.connect(URL) as ws:
+            await asyncio.gather(
+                receiver(websocket,ws),
+                sender(websocket,ws)
+            )
+    except websockets.exceptions.WebSocketException as e:
+        #print(f"WebSocket exception: {e}")
+        sys.exit(1)
+start_server = websockets.serve(chat,"0.0.0.0", 7860)
 
 asyncio.get_event_loop().run_until_complete(start_server)
-asyncio.get_event_loop().run_forever()
+asyncio.get_event_loop().run_forever()
+