Update app.py

app.py

@@ -50,6 +50,7 @@ def update_game(inp):
     return game_fn(sky=sky)
 
 def sky_fn(inp):
+    uid=uuid.uuid4()
     rand = random.randint(1,200)
     for i in range(rand):
         inp+=" "
@@ -72,6 +73,7 @@ def sky_fn(inp):
     return outp,out_url
 
 def star_fn(inp):
+    uid=uuid.uuid4()
     rand = random.randint(1,200)
     for i in range(rand):
         inp+=" "
@@ -86,6 +88,7 @@ def star_fn(inp):
     return out,out_url
 
 def enemy_fn(inp):
+    uid=uuid.uuid4()
     rand = random.randint(1,200)
     for i in range(rand):
         inp+=" "
@@ -99,7 +102,11 @@ 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,sky_im,star_im,enemy_im):
+    sky_im=sky_im.split("app/",1)[1]
+    star_im=star_im.split("app/",1)[1]
+    #sky_im=sky_im.split("app/",1)[1]
+    print(star_im)
     user_repo=save_data.split('datasets/',1)[1].split('/raw',1)[0]
     timestamp=str(datetime.datetime.now())
     timename=timestamp.replace(" ","--").replace(":","-").replace(".","-")    
@@ -121,7 +128,7 @@ def save_game(name,score):
         pass
     try: 
         api.upload_file(
-        path_or_fileobj=f"{uid}_sky.png",
+        path_or_fileobj=sky_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,
@@ -132,7 +139,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,
@@ -143,7 +150,7 @@ def save_game(name,score):
         pass  
     try: 
         api.upload_file(
-        path_or_fileobj=f"{uid}_enemy.png",
+        path_or_fileobj=enemy_im.split("app/",1)[1],
         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,
@@ -348,7 +355,7 @@ with gr.Blocks() as app:
 
     get_high_score_btn.click(get_high_score,None,score_html)    
     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).then(get_high_score,None,score_html)
     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])