import gradio as gr
from gradio_client import Client

def set_client_for_session(request: gr.Request):
    x_ip_token = request.headers['x-ip-token']
    return Client("hysts/SDXL", headers={"x-ip-token": x_ip_token})

def text_to_image(prompt, request: gr.Request):
    client = set_client_for_session(request)
    img = client.predict(prompt, "", "", "", api_name="/run")
    return img

with gr.Blocks() as demo:
    image = gr.Image()
    prompt = gr.Textbox(max_lines=1)
    prompt.submit(text_to_image, [prompt], [image])

demo.launch()

# demo = gr.load("spaces/hysts/SDXL")
# demo.launch()