import gradio as gr def get_msg(name, request: gr.Request): if dict(request.headers)['origin'] in os.environ["ORIGIN"]: return "Working" return "Hello " + name + "!" 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()