Update app.py

app.py

@@ -6,9 +6,12 @@ import uuid
 import random
 import bg
 from rembg import remove as rm
+import datatime 
 proc1=diff
 proc2=diff 
 
+save_data='https://huggingface.co/datasets/Omnibus/game-data-1/raw/main/'
+token_self = os.environ['HF_TOKEN']
 
 #bg1=gr.Interface.load("spaces/Omnibus/fast-bg-remove-mod")
 #proc2=gr.Interface.load("spaces/Omnibus/fast-diff")
@@ -88,6 +91,64 @@ def enemy_fn(inp):
     out_url = f'https://omnibus-game-test.hf.space/file={out_file}'    
     return out,out_url
 
+def save_game(background_img,star_img,enemy_img,platform_img,dude_img):
+    #background_image.save(f"{uid}-background_img.png")
+    #star_img.save(f"{uid}-star_img.png")
+    #enemy_img.save(f"{uid}-enemy_img.png")
+    #platform_img.save(f"{uid}-platform_img.png")
+    #dude_img.save(f"{uid}-dude_img.png")
+    
+    user_repo=game_data.split('datasets/',1)[1].split('/raw',1)[0]
+    try:
+        r = requests.get(f'{save_data}/game_data.json') 
+        lod = json.loads(r.text)
+    except:
+        lod=[]
+        pass
+    timestamp=str(datetime.datetime.now())
+    print(timestamp)
+    try: 
+        api.upload_file(
+        path_or_fileobj=background_img,
+        path_in_repo=f"{game_data.split('main/',1)[1]}/images/background/{uid}-background_img.png",
+        repo_id=game_data.split('datasets/',1)[1].split('/raw',1)[0],
+        token=token_self,
+        repo_type="dataset",
+        )
+        #os.remove("tmp1.json")
+
+    except Exception as e:
+        print (e)
+        pass    
+    block = {'index': len(lod) + 1,
+            'timestamp': timestamp,
+            'background_url': f'https://huggingface.co/datasets/{user_repo}/raw/main/images/background/{uid}-background_img.png',
+            'star_url': f'https://huggingface.co/datasets/{user_repo}/raw/main/images/star/{star_img}',
+            'enemy_url': f'https://huggingface.co/datasets/{user_repo}/raw/main/images/enemy/{enemy_img}',
+            #'platform_url': f'https://huggingface.co/datasets/{repo}/{space}/raw/main/{platform_img}',
+            #'dude_url': f'https://huggingface.co/datasets/{repo}/{space}/raw/main/{dude_img}',
+            }
+    print(block)
+    lod.append(block)    
+    
+    json_object = json.dumps(lod, indent=4)
+    with open("tmp1.json", "w") as outfile:
+        outfile.write(json_object)
+    try: 
+        api.upload_file(
+        path_or_fileobj="tmp1.json",
+        path_in_repo=f"{save_data.split('main/',1)[1]}/game_data.json",
+        repo_id=save_data.split('datasets/',1)[1].split('/raw',1)[0],
+        token=token_self,
+        repo_type="dataset",
+        )
+        os.remove("tmp1.json")
+        print("success")
+    except Exception as e:
+        print (e)
+        pass
+
+
 with gr.Blocks() as app:
     with gr.Row():
         with gr.Column():
@@ -112,7 +173,9 @@ with gr.Blocks() as app:
         update_game=gr.Button("Make Game")
         #start_prompt=gr.Textbox(value="beautiful landscape, real, 8k",visible=False)
     html_game = gr.HTML()
-    
+    save_btn=gr.Button("Save (Experimental)")
+
+    save_btn.click(save_game,[out_im_sky,out_im_star,out_im_enemy,out_im_enemy,out_im_enemy],None)
     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])