import threading
import asyncio
import websockets


async def websocket_consumer(websocket, path):
    
    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:
         pass

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()

# 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)

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