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