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