import gradio as gr
import os
import json

def get_msg(name, request: gr.Request):
    header_origin = str(request.headers)
    #header_origin = json.loads(str(header_origin[8:-1]).replace("'", "\""))['origin']
    if header_origin == str(os.environ["ORIGIN"]):
        return "Working"
    return "Hello " + name + "!" + str(header_origin)

with gr.Blocks() as demo:

    input = gr.Textbox(label="Prompt", elem_id="textbox")
    output = gr.Textbox(label="Output", elem_id="textbox")
    button = gr.Button(value="Make Magic", elem_id="button")
    button.click(get_msg, inputs=[input], outputs=[output])

demo.launch()