import asyncio
import websockets

async def run_code(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.")
            
            await web.send(code)
            print("Code sent to external WebSocket server.")

            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}')

start_server = websockets.serve(run_code, "0.0.0.0", 7860)

asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()