import gradio as gr
from gradio_client import Client

# client = Client("hysts/SDXL")

def text_to_image(prompt, client):
    print(">>>>>>>>MMMMMMMM")
    from gradio.context import LocalContext
    print("localcontext.request", LocalContext.request)
    print("localcontext", LocalContext.request.get().headers.get("x-ip-token"))
    img = client.predict(prompt, "", "", "", api_name="/run")
    print("<<<<<<<<NNNNNNNN")
    return img

with gr.Blocks() as demo:
    client = gr.State()
    image = gr.Image()
    prompt = gr.Textbox(max_lines=1)
    prompt.submit(text_to_image, [prompt, client], [image])
    demo.load(lambda: Client("hysts/SDXL"), inputs=None, outputs=client)

demo.launch()

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