| import gradio as gr |
| import os |
| import json |
|
|
| def get_msg(name, request: gr.Request): |
| header_origin = str(request.headers) |
| header_origin = json.loads(header_origin.replace("'", "\""))['origin'] |
| if header_origin == str(os.environ["ORIGIN"]): |
| return "Working" |
| return "Sign up on visualstoriesonline.com" |
|
|
| 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() |
