Update app.py

app.py

@@ -1,38 +1,28 @@
+import threading
 import asyncio
 import websockets
 
-async def run_code(websocket, path):
+
+async def websocket_consumer(websocket, path):
+    
     try:
-        print("WebSocket connection established.")
-        
-        # Read the code from the WebSocket connection
         code = await websocket.recv()
-        print("Received code:", code)
-
-        # Establish connection to external WebSocket server
-        async with websockets.connect('wss://ramesh-vani-wspython.hf.space') as web:
-            print("Connection to external WebSocket server established.")
+        async with websockets.connect('wss://ramesh-vani-wspython.hf.space') as wssocket:
             
-            await web.send(code)
-            print("Code sent to external WebSocket server.")
-            print(code)
-
+            await wssocket.send(code)
             while True:
-                response = await web.recv()
-                print("Received response:", response)
-                await send_message(websocket, response)
-
-    except Exception as e:
-        print(f'Error: {str(e)}')
-        await websocket.send(f'Error: {str(e)}')
-    finally:
-        print("Closing WebSocket connection.")
-        await websocket.close()
-
-async def send_message(websocket, message):
-    await websocket.send(f'data: {message}')
+                response = await wssocket.recv()
+                await websocket.send(response)
+        
+    except websockets.exceptions.ConnectionClosedOK:
+         pass
 
-start_server = websockets.serve(run_code, "0.0.0.0", 7860)
+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()
 
-asyncio.get_event_loop().run_until_complete(start_server)
-asyncio.get_event_loop().run_forever()
+websocket_thread = WebSocketThread()
+websocket_thread.start()