Update app.py

app.py

@@ -65,12 +65,12 @@ def sky_fn(inp):
     width, height = outp.size
     rat = width/height
     if width > height:
-        outp = outp.resize((600*rat,600))
+        outrs = outp.resize((600*rat,600))
     elif width < height:
-        outp = outp.resize((800,800*rat))     
+        outrs = outp.resize((800,800*rat))     
     else:
-        outp = outp.resize((800,536))
-    outp.save(f"{uid}_sky.png")
+        outrs = outp.resize((800,536))
+    outrs.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}'    
@@ -104,7 +104,7 @@ def enemy_fn(inp):
     out_url = f'https://omnibus-game-test.hf.space/file={out_file}'    
     return out,out_url
 
-def save_game(name,score):
+def save_game(name,score,background_im,star_im,enemy_im):
     user_repo=save_data.split('datasets/',1)[1].split('/raw',1)[0]
     timestamp=str(datetime.datetime.now())
     timename=timestamp.replace(" ","--").replace(":","-").replace(".","-")    
@@ -129,7 +129,7 @@ def save_game(name,score):
     print(timestamp)
     try: 
         api.upload_file(
-        path_or_fileobj=f"{uid}_sky.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,
@@ -140,7 +140,7 @@ def save_game(name,score):
         pass    
     try: 
         api.upload_file(
-        path_or_fileobj=f"{uid}_star.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,
@@ -151,7 +151,7 @@ def save_game(name,score):
         pass   
     try: 
         api.upload_file(
-        path_or_fileobj=f"{uid}_enemy.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,
@@ -288,7 +288,7 @@ with gr.Blocks() as app:
     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],message)
+    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)
     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])