button 리스트도 유저별 관리

app.py

@@ -11,8 +11,6 @@ from collections import defaultdict
 wandb.login(key=os.environ['WANDB_KEY'])
 run = wandb.init(project="pokemon-quiz", entity="yoon-gu")
 
-random_buttons = [gr.Button(value=f"{name}") for name in ['1번', '2번', '3번', '4번']]
-
 with open('pokemon.json', 'r') as f:
     pokemons = json.load(f)
 pokemons_types = ["모든 타입"] + sorted(set([t for poke in pokemons for t in poke['types']]))
@@ -32,7 +30,7 @@ GEN_RANGE = {
 
 QUESTION_TEMPLATE = {"question": "다음 포켓몬의 이름은 뭘까요?![]({img_url})", "answer": "{name}"}
 
-def get_question_answer(pokemons_set):
+def get_question_answer(pokemons_set, user):
     chosen = random.choice(pokemons_set)
     name = chosen['name'].replace("♀", "암컷").replace("♂", "수컷")
     image_path = chosen['image_path']
@@ -43,14 +41,15 @@ def get_question_answer(pokemons_set):
     candidates = random.choices([poke['name'] for poke in pokemons_set], k=3)
     candidates.append(a)
     random.shuffle(candidates)
-    random_buttons = [gr.Button(value=f"{name}") for name in candidates]
-    return q, a, random_buttons
+    info[user]['candidates'] = candidates
+    return q, a
 
 initial_info = {"done" : True,
                 "score": 0, "count": 0,
                 "best_score": 0, "best_time": float("inf"),
                 "time": 0.0, "comment": "",
-                "history": []}
+                "history": [],
+                "candidates": ['1번', '2번', '3번', '4번']}
 
 try:
     folder = run.use_artifact("settings:latest").download()
@@ -58,9 +57,6 @@ try:
         USERS = json.load(f)
     with open(os.path.join(folder, "info.json"), "r") as f:
         info = json.load(f)
-    for user in USERS:
-        if "history" not in info[user]:
-            info[user]['history'] = []
 except:
     info = defaultdict(lambda : deepcopy(initial_info))
     USERS = ["김서현", "김우주", "Anonymous"]
@@ -126,7 +122,7 @@ with gr.Blocks() as demo:
         global random_buttons
         if done:
             if "퀴즈시작" == message.replace(" ", ""):
-                q, a, random_buttons = get_question_answer(pokemons_set)
+                q, a = get_question_answer(pokemons_set, user)
                 bot_message = f"퀴즈를 시작합니다.\n{q}"
                 info[user]['answer'] = a
                 info[user]['done'] = False
@@ -137,18 +133,18 @@ with gr.Blocks() as demo:
                 bot_message = "퀴즈를 시작하고 싶으시면, **퀴즈 시작** 버튼을 누르세요."
         else:
             if info[user]['answer'] == message:
-                q, a, random_buttons = get_question_answer(pokemons_set)
+                q, a = get_question_answer(pokemons_set, user)
                 info[user]['answer'] = a
                 info[user]['score'] += 1
                 info[user]['count'] += 1
-                bot_message = f"🎉정답입니다! 다음 문제입니다.\n{q}\n- 현재 점수: {info[user]['score']}점\n- 소요 시간: {time.time() - info[user]['time']:4.3f}초"
+                bot_message = f"🎉정답입니다! 다음 문제입니다.\n{q}\n- 현재 점수: {info[user]['score']:3.1f}점\n- 소요 시간: {time.time() - info[user]['time']:4.3f}초"
 
             elif "포기하기" == message.replace(" ", ""):
                 bot_message = f"퀴즈를 강제 종료합니다."
                 info[user]['done'] = True
             elif "문제 넘어가기" == message:
                 info[user]['count'] += 1
-                q, a = get_question_answer(pokemons_set)
+                q, a = get_question_answer(pokemons_set, user)
                 info[user]['answer'] = a
                 bot_message = f"문제를 넘어갑니다. 다음문제입니다.\n{q}"
             else:
@@ -195,7 +191,7 @@ with gr.Blocks() as demo:
         else:
             btn = gr.Button(value="포기하기", label="퀴즈 중단")
 
-        
+        random_buttons = [gr.Button(value=c) for c in info[user]['candidates']]
         return btn, info[user]['history'], md, *random_buttons
 
     play.click(respond, 
@@ -217,8 +213,6 @@ with gr.Blocks() as demo:
         new_dropdown = gr.Dropdown(sorted(USERS), value=rs, label="사용자", info="당신은 누구신가요?", allow_custom_value=True)
         return new_dropdown
 
-    user.blur(rs_change, user, user)
-
     def update_table(user):
         leader_board = sorted(info.items(), key=lambda x: (x[1]['best_score'], -x[1]['best_time']), reverse=True)
         lbdf = pd.DataFrame([dict(**a[1], name=a[0]) for a in leader_board])
@@ -231,10 +225,17 @@ with gr.Blocks() as demo:
         else:
             btn = gr.Button(value="포기하기", label="퀴즈 중단")
 
-        return btn, md, gr.Dropdown(USERS, value=user, label="사용자", info="당신은 누구신가요?", allow_custom_value=True), info[user]['history']
+        random_buttons = [gr.Button(value=c) for c in info[user]['candidates']]
+        return btn, md, gr.Dropdown(USERS, value=user, label="사용자", info="당신은 누구신가요?", allow_custom_value=True), info[user]['history'], *random_buttons
+    
+    user.blur(rs_change, user, user)
+    user.select(update_table,
+              inputs=user,
+              outputs=[play, leader_board, user, chatbot, button1, button2, button3, button4])
+
     demo.load(update_table,
               inputs=user,
-              outputs=[play, leader_board, user, chatbot])
+              outputs=[play, leader_board, user, chatbot, button1, button2, button3, button4])
 
 demo.queue(concurrency_count=3)
 demo.launch()