Update app.py

app.py

@@ -23,7 +23,7 @@ dude = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/dude
 
 def game_fn(sky=sky,star=star,bomb=bomb,platform=platform,dude=dude):
     html_mod=f"""
-        <div id="demo" style="height:600px">
+        <div id="my_game" 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}"        
@@ -66,19 +66,26 @@ def star_fn(inp):
         inp+=" "
     uid=uuid.uuid4()
     output=proc2.send_it(inp,5,1)
-    print(output)
     outp=Image.open(output[0])
-    #outp.save(f"{uid}_star.png")
-    print("removing background")
     out=rm(outp)
-    print("resizing")
-    #outrs = out.resize((24,24))
     outrs = out.resize((36,36))
-    
     outrs.save(f"{uid}_star.png")
     out_file = os.path.abspath(f"{uid}_star.png")
     out_url = f'https://omnibus-game-test.hf.space/file={out_file}'    
-    
+    return out,out_url
+
+def enemy_fn(inp):
+    rand = random.randint(1,200)
+    for i in range(rand):
+        inp+=" "
+    uid=uuid.uuid4()
+    output=proc2.send_it(inp,5,1)
+    outp=Image.open(output[0])
+    out=rm(outp)
+    outrs = out.resize((24,24))
+    outrs.save(f"{uid}_enemy.png")
+    out_file = os.path.abspath(f"{uid}_enemy.png")
+    out_url = f'https://omnibus-game-test.hf.space/file={out_file}'    
     return out,out_url
 
 with gr.Blocks() as app:
@@ -93,17 +100,23 @@ with gr.Blocks() as app:
             btn_star=gr.Button("Make")
             out_im_star=gr.Image(type='filepath')
             out_star_url=gr.Textbox(visible=False)
-        gr.Column()
+        with gr.Column():
+            prompt_enemy=gr.Textbox(label="Enemy",value="Ball on fire, blank background")
+            btn_enemy=gr.Button("Make")
+            out_im_enemy=gr.Image(type='filepath')
+            out_enemy_url=gr.Textbox(visible=False)
         gr.Column()
         gr.Column()
         
     with gr.Row():
-        update_game=gr.Button("Use Image")
+        update_game=gr.Button("Make Game")
         #start_prompt=gr.Textbox(value="beautiful landscape, real, 8k",visible=False)
     html_game = gr.HTML()
     
-    update_game.click(game_fn,[out_sky_url,out_star_url],html_game)
+    update_game.click(game_fn,[out_sky_url,out_star_url,out_enemy_url],html_game)
     btn_sky.click(sky_fn,prompt_sky,[out_im_sky,out_sky_url])
     btn_star.click(star_fn,prompt_star,[out_im_star,out_star_url])
+    btn_enemy.click(enemy_fn,prompt_enemy,[out_im_enemy,out_enemy_url])
+
     #app.load(sky_fn,prompt_sky,[out_im_sky,out_sky_url]).then(game_fn,[out_sky_url],html_game)
 app.queue(concurrency_count=10).launch(show_api=False)