import gradio as gr

HEADER_NAMES = ['x-zerogpu-token', 'x-zerogpu-uuid', 'x-ip-token']

def greet(name, request: gr.Request):
    return {k:v for k,v in request.headers.items() if k in HEADER_NAMES}

def headers(request: gr.Request):
    return {k:v for k,v in request.headers.items() if k in HEADER_NAMES}

iface = gr.Interface(fn=greet, inputs="text", outputs="json")
with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column():
            iface.render()
        with gr.Column():
            load_headers = gr.Json(label="Headers on Page Load")
    demo.load(headers, inputs=None, outputs=load_headers)
demo.launch()