Update app.py

app.py

@@ -33,14 +33,14 @@ def update_game(inp):
     
     return game_fn(sky=sky)
 
-def star_fn(inp):
+def sky_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=proc1.send_it(inp,5,1)
     #output=dif(inp)
     print(output)
     outp=Image.open(output[0])
@@ -51,22 +51,21 @@ def star_fn(inp):
     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")
+        outp = outp.resize((800,536))
+    outp.save(f"{uid}_sky.png")
+    out = os.path.abspath(f"{uid}_sky.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):
+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=proc2.send_it(inp,5,1)
     #output=dif(inp)
     print(output)
     outp=Image.open(output[0])
@@ -77,9 +76,9 @@ def sky_fn(inp):
     elif width < height:
         outp = outp.resize((800,800*rat))     
     else:
-        outp = outp.resize((800,536))
-    outp.save(f"{uid}_sky.png")
-    out = os.path.abspath(f"{uid}_sky.png")
+        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
@@ -93,11 +92,13 @@ with gr.Blocks() as app:
                 update_game=gr.Button("Use Image")
         out_im_sky=gr.Image(type='filepath',height=50)
         out_im_star=gr.Image(type='filepath',height=50)
-        out_url=gr.Textbox(visible=False)
+        out_sky_url=gr.Textbox(visible=False)
+        out_star_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_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)
+    update_game.click(game_fn,[out_sky_url,out_star_url],html_game)
+    btn.click(sky_fn,prompt,[out_im_sky,out_sky_url])
+    btn.click(star_fn,prompt,[out_im_star,out_star_url])
+    app.load(sky_fn,start_prompt,[out_im_sky,out_sky_url]).then(game_fn,[out_sky_url],html_game)
 app.launch()