| 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).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() |