Update app.py

app.py

@@ -70,7 +70,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,rand
 
 def star_fn(inp):
     uid=uuid.uuid4()
@@ -85,7 +85,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,rand
 
 def enemy_fn(inp):
     uid=uuid.uuid4()
@@ -100,9 +100,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 out,out_url
+    return out,out_url,rand
 
-def save_game(name,score,sky_im,star_im,enemy_im):
+def save_game(name,score,sky_im,star_im,enemy_im,sky_p,star_p,enemy_p,background_s,star_s,enemy_s):
     sky_im=sky_im.split("app/",1)[1]
     star_im=star_im.split("app/",1)[1]
     #sky_im=sky_im.split("app/",1)[1]
@@ -184,7 +184,13 @@ def save_game(name,score,sky_im,star_im,enemy_im):
     block = {'index': len(lod) + 1,
             'timestamp': timestamp,
             'game_name': game,
-            'score': score, 
+            'score': score,
+            'background_p': sky_p,
+            'star_p': star_p,
+            'enemy_p': enemy_p,
+            'background_s': background_s,
+            'star_s': star_s,
+            'enemy_s': enemy_s,             
             'background_url': f'https://huggingface.co/datasets/{user_repo}/resolve/main/images/background/{game}-background_img.png',
             'star_url': f'https://huggingface.co/datasets/{user_repo}/resolve/main/images/star/{game}-star_img.png',
             'enemy_url': f'https://huggingface.co/datasets/{user_repo}/resolve/main/images/enemy/{game}-enemy_img.png',
@@ -214,6 +220,12 @@ def save_game(name,score,sky_im,star_im,enemy_im):
     block2 = {'game_name': game,
             'score':score,
             'timestamp': timestamp,
+            'background_p': sky_p,
+            'star_p': star_p,
+            'enemy_p': enemy_p,    
+            'background_s': background_s,
+            'star_s': star_s,
+            'enemy_s': enemy_s,                     
             'background_url': f'https://huggingface.co/datasets/{user_repo}/resolve/main/images/background/{game}-background_img.png',
             'star_url': f'https://huggingface.co/datasets/{user_repo}/resolve/main/images/star/{game}-star_img.png',
             'enemy_url': f'https://huggingface.co/datasets/{user_repo}/resolve/main/images/enemy/{game}-enemy_img.png',
@@ -348,6 +360,13 @@ with gr.Blocks() as app:
         text_input=gr.Textbox()
         url_params = gr.JSON({}, visible=True, label="")
         acc=gr.Textbox()
+    with gr.Row(visible=False):
+        sky_p=gr.Textbox()
+        star_p=gr.Textbox()
+        enemy_p=gr.Textbox()
+        sky_s=gr.Textbox()
+        star_s=gr.Textbox()
+        enemy_s=gr.Textbox()        
 
         get_high_score_btn=gr.Button()
     def return_score(text):
@@ -356,11 +375,12 @@ with gr.Blocks() as app:
         return None
     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,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,out_sky_url,out_star_url,out_enemy_url,sky_p,star_p,enemy_p,sky_s,star_s,enemy_s],message)
     update_game.click(refresh_frame,None,html_game).then(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])
-    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,sky_p,sky_s])
+    btn_star.click(star_fn,prompt_star,[out_im_star,out_star_url,star_p,star_s])
+    btn_enemy.click(enemy_fn,prompt_enemy,[out_im_enemy,out_enemy_url,enemy_p,enemy_s])
     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)