Update app.py

app.py

@@ -16,7 +16,7 @@ dude = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/dude
 
 
 
-def game_fn(sky=sky,platform=platform,star=star,bomb=bomb,dude=dude):
+def game_fn(sky=sky,star=star,bomb=bomb,platform=platform,dude=dude):
     html_mod=f"""
         <div id="demo" style="height:600px">
             <iframe
@@ -33,7 +33,33 @@ def update_game(inp):
     
     return game_fn(sky=sky)
 
-def dif_fn(inp):
+def star_fn(inp):
+    rand = random.randint(1,200)
+    for i in range(rand):
+        inp+=" "
+    #dif = gr.Interface.load("models/stabilityai/stable-diffusion-2-1-base", live=True),
+    #dif = models[0]
+    uid=uuid.uuid4()
+    output=diff.send_it(inp,5,1)
+    #output=dif(inp)
+    print(output)
+    outp=Image.open(output[0])
+    width, height = outp.size
+    rat = width/height
+    if width > height:
+        outp = outp.resize((600*rat,600))
+    elif width < height:
+        outp = outp.resize((800,800*rat))     
+    else:
+        outp = outp.resize((24,24))
+    outp.save(f"{uid}_star.png")
+    out = os.path.abspath(f"{uid}_star.png")
+    #out = os.path.abspath(outp)
+    out_url = f'https://omnibus-game-test.hf.space/file={out}'    
+    return output[0],out_url
+
+
+def sky_fn(inp):
     rand = random.randint(1,200)
     for i in range(rand):
         inp+=" "
@@ -65,12 +91,13 @@ with gr.Blocks() as app:
             with gr.Row():
                 btn=gr.Button("Make Image")
                 update_game=gr.Button("Use Image")
-        out_im=gr.Image(type='filepath')
+        out_im_sky=gr.Image(type='filepath',height=50)
+        out_im_star=gr.Image(type='filepath',height=50)
         out_url=gr.Textbox(visible=False)
         start_prompt=gr.Textbox(value="beautiful landscape, real, 8k",visible=False)
     html_game = gr.HTML()
     
-    update_game.click(game_fn,[out_url],html_game)
-    btn.click(dif_fn,prompt,[out_im,out_url])
-    app.load(dif_fn,start_prompt,[out_im,out_url]).then(game_fn,[out_url],html_game)
+    update_game.click(game_fn,[out_sky_url,star_url],html_game)
+    btn.click(sky_fn,prompt,[out_im_sky,out_sky_url]).then(star_fn,prompt,[out_im_star,out_star_url])
+    app.load(sky_fn,start_prompt,[out_im,out_url]).then(game_fn,[out_sky_url,out_star_url],html_game)
 app.launch()