Update app.py

app.py

@@ -2,12 +2,16 @@ import asyncio
 import websockets
 
 async def handle_user_client(websocket, path):
+    code = await websocket.recv()
     external_uri = 'wss://ramesh-vani-wspython.hf.space'
     
     async with websockets.connect(external_uri) as external_ws:
         async def forward_user_to_external():
-            async for user_message in websocket:
-                await external_ws.send(user_message)
+            
+            await external_ws.send(code)
+            while True:
+                response = await external_ws.recv()
+                await websocket.send(f'{response}')
 
         async def forward_external_to_user():
             async for external_message in external_ws:
@@ -17,7 +21,7 @@ async def handle_user_client(websocket, path):
         # Start two tasks to handle message forwarding in both directions
         await asyncio.gather(
             forward_user_to_external(),
-            forward_external_to_user(),
+            # forward_external_to_user(),
         )
 
 # Set up WebSocket server for user