Update app.py

app.py

@@ -74,7 +74,7 @@ def sky_fn(inp):
     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 outp,out_url
+    return outp,out_url,outrs
 
 def star_fn(inp):
     rand = random.randint(1,200)
@@ -88,7 +88,7 @@ def star_fn(inp):
     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
+    return out,out_url,outrs
 
 def enemy_fn(inp):
     rand = random.randint(1,200)
@@ -102,7 +102,7 @@ def enemy_fn(inp):
     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
+    return out,out_url,outrs
 
 def save_game(name,score,background_im,star_im,enemy_im):
     user_repo=save_data.split('datasets/',1)[1].split('/raw',1)[0]
@@ -132,7 +132,7 @@ def save_game(name,score,background_im,star_im,enemy_im):
         t1.write(i1)
     try: 
         api.upload_file(
-        path_or_fileobj='tmp1.png',
+        path_or_fileobj=background_im,
         path_in_repo=f"{save_data.split('main/',1)[1]}/images/background/{game}-background_img.png",
         repo_id=save_data.split('datasets/',1)[1].split('/raw',1)[0],
         token=token_self,
@@ -148,7 +148,7 @@ def save_game(name,score,background_im,star_im,enemy_im):
         t2.write(i2)    
     try: 
         api.upload_file(
-        path_or_fileobj='tmp2.png',
+        path_or_fileobj=star_im,
         path_in_repo=f"{save_data.split('main/',1)[1]}/images/star/{game}-star_img.png",
         repo_id=save_data.split('datasets/',1)[1].split('/raw',1)[0],
         token=token_self,
@@ -164,7 +164,7 @@ def save_game(name,score,background_im,star_im,enemy_im):
         t3.write(i3)     
     try: 
         api.upload_file(
-        path_or_fileobj='tmp3.png',
+        path_or_fileobj=enemy_im,
         path_in_repo=f"{save_data.split('main/',1)[1]}/images/enemy/{game}-enemy_img.png",
         repo_id=save_data.split('datasets/',1)[1].split('/raw',1)[0],
         token=token_self,
@@ -301,14 +301,19 @@ with gr.Blocks() as app:
         text_input=gr.Textbox()
         url_params = gr.JSON({}, visible=True, label="")
         acc=gr.Textbox()
+    with gr.Row():
+        hid_sky=gr.Image(type='filepath')
+        hid_star=gr.Image(type='filepath')
+        hid_enemy=gr.Image(type='filepath')
+        
     def return_score(text):
         return text
     get_score.click(return_score,score,[score],_js=score_js)
-    save_btn.click(return_score,score,[score],_js=score_js).then(save_game,[game_name,score,out_sky_url,out_star_url,out_enemy_url],message)
+    save_btn.click(return_score,score,[score],_js=score_js).then(save_game,[game_name,score,hid_sky,hid_star,hid_enemy],message)
     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])
+    btn_sky.click(sky_fn,prompt_sky,[out_im_sky,out_sky_url,hid_sky])
+    btn_star.click(star_fn,prompt_star,[out_im_star,out_star_url,hid_star])
+    btn_enemy.click(enemy_fn,prompt_enemy,[out_im_enemy,out_enemy_url,hid_enemy])
     app.load(predict,[text_input,url_params],[text_input,html_game], _js=get_window_url_params) 
 
     #app.load(sky_fn,prompt_sky,[out_im_sky,out_sky_url]).then(game_fn,[out_sky_url],html_game)