import gradio as gr
import asyncio
import websockets
import threading

async def send_wss_request():
    uri = "wss://clone3-imagex-clone-advance.hf.space/queue/join"
    uri2 = "wss://clone3-imagex-clone-advance.hf.space/queue/join"

    while True:
        try:
            async with websockets.connect(uri) as websocket:
                # Receive the response
                response = await websocket.recv()
                print(f"Received: {response}")
            
            async with websockets.connect(uri2) as websocket:
                # Receive the response
                response = await websocket.recv()
                print(f"Received: {response}")

        except websockets.exceptions.ConnectionClosed:
            print("Connection closed. Reconnecting in 30 minutes.")
            
        # Sleep for 30 minutes before the next iteration
        await asyncio.sleep(30 * 60)

def run_websocket():
    asyncio.get_event_loop().run_until_complete(send_wss_request())

# Create a thread for the WebSocket function
websocket_thread = threading.Thread(target=run_websocket)
websocket_thread.start()

def greet(name):
    return "Hello " + name + "!"

demo = gr.Interface(
    fn=greet,
    inputs=gr.Textbox(lines=2, placeholder="Name Here..."),
    outputs="text",
)
demo.launch()