import gradio as gr
import diff

dif = diff.SendIt()


sky = "https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/sky.png"
platform = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/platform.png'
star = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/star.png'
bomb = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/bomb.png'
dude = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/dude.png'


html_mod=f"""
    <div id="demo" style="height:600px">
        <iframe
        id="myIframe"
        src="https://omnibus-game-test-static.static.hf.space/index.html?sky={sky}&platform={platform}&star={star}&bomb={bomb}&dude={dude}"        
    	frameborder="0"
    	width="100%"
        height="100%"
        ></iframe>
    </div>"""

def dif_fn(inp):
    output=dif(inp)
    return output[0]



with gr.Blocks() as app:
    with gr.Row():
        with gr.Column():
            prompt=gr.Textbox()
            btn=gr.Button()
        out_im=gr.Image()
    
    gr.HTML(html_mod)
    btn.click(dif_fn,prompt,out_im)
app.launch()