Update app.py

app.py

@@ -9,31 +9,33 @@ star = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/star
 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 game_fn(sky=sky,platform=platform,star=star,bomb=bomb,dude=dude)
+    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>"""
+    return html_mod
 
 def dif_fn(inp):
     output=diff.send_it(inp,5,1)
     return output[0]
 
-
-
 with gr.Blocks() as app:
     with gr.Row():
         with gr.Column():
             prompt=gr.Textbox()
-            btn=gr.Button()
+            with gr.Row():
+                btn=gr.Button("Make Image")
+                game_btn=gr.Button("Use Image")
         out_im=gr.Image()
-    
-    gr.HTML(html_mod)
+    game_btn.click(game_fn,[out_im],html_game)
+    html_game = gr.HTML(html_mod)
     btn.click(dif_fn,prompt,out_im)
+    app.load(game_fn,None,html_game)
 app.launch()