Update app.py

app.py

@@ -1,4 +1,8 @@
 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'
@@ -18,6 +22,19 @@ html_mod=f"""
         ></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()