Update app.py

app.py

@@ -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 outrs,out_url
+    return out,out_url
 
 def enemy_fn(inp):
     rand = random.randint(1,200)
@@ -102,9 +102,9 @@ 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 outrs,out_url
+    return out,out_url
 
-def save_game(name,score,background_img,star_img,enemy_img,platform_img,dude_img):
+def save_game(name,score):
     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,background_img,star_img,enemy_img,platform_img,dude_img
     print(timestamp)
     try: 
         api.upload_file(
-        path_or_fileobj=background_img,
+        path_or_fileobj=f"{uid}_sky.png",
         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,background_img,star_img,enemy_img,platform_img,dude_img
         pass    
     try: 
         api.upload_file(
-        path_or_fileobj=star_img,
+        path_or_fileobj=f"{uid}_star.png",
         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,background_img,star_img,enemy_img,platform_img,dude_img
         pass   
     try: 
         api.upload_file(
-        path_or_fileobj=enemy_img,
+        path_or_fileobj=f"{uid}_enemy.png",
         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,out_im_sky,out_im_star,out_im_enemy,out_im_enemy,out_im_enemy],message)
+    save_btn.click(return_score,score,[score],_js=score_js).then(save_game,[game_name,score],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])