Update app.py

app.py

@@ -1,10 +1,13 @@
 import os
-os.system("pip install gradio==3.47.1")
 import gradio as gr
 import pandas as pd
 import random
 import json
 import time
+import wandb
+
+wandb.login(key=os.environ['WANDB_KEY'])
+run = wandb.init(project="pokemon-quiz", entity="yoon-gu")
 
 with open('pokemon.json', 'r') as f:
     pokemons = json.load(f)
@@ -35,7 +38,12 @@ def get_question_answer(pokemons_set):
     a = QUESTION_TEMPLATE['answer'].format(name=name)
     return q, a
 
-info = {u: {"done" : True, "score": 0, "count": 0, "best_score": 0, "best_time": float("inf"), "time": 0.0} for u in USERS}
+try:
+    folder = run.use_artifact("settings:latest").download()
+    with open(os.path.join(folder, "info.json"), "r") as f:
+        info = json.load(f)
+except:
+    info = {u: {"done" : True, "score": 0, "count": 0, "best_score": 0, "best_time": float("inf"), "time": 0.0} for u in USERS}
 
 MD = """# 포켓몬 퀴즈
 ## 공부 방법
@@ -135,6 +143,11 @@ with gr.Blocks() as demo:
                 if info[user]['score'] >= info[user]['best_score']:
                     info[user]['best_score'] = info[user]['score']
                     info[user]['best_time'] = min(time.time() - info[user]['time'], info[user]['best_time'])
+                with open("info.json", "w") as f:
+                    json.dump(info, f, indent=4)
+                artifact = wandb.Artifact("settings", type="info")
+                artifact.add_file("info.json")
+                run.log_artifact(artifact)
 
         chat_history.append((message, bot_message))