pydantic 사용

app.py

@@ -5,8 +5,7 @@ import random
 import json
 import time
 import wandb
-from copy import deepcopy
-from collections import defaultdict
+from utils import Info
 
 wandb.login(key=os.environ['WANDB_KEY'])
 run = wandb.init(project="pokemon-quiz", entity="yoon-gu")
@@ -41,7 +40,7 @@ def get_question_answer(pokemons_set, user):
     candidates = random.choices([poke['name'] for poke in pokemons_set], k=3)
     candidates.append(a)
     random.shuffle(candidates)
-    info[user]['candidates'] = candidates
+    infos[user].candidates = candidates
     return q, a
 
 initial_info = {"done" : True,
@@ -55,13 +54,11 @@ try:
     folder = run.use_artifact("settings:latest").download()
     with open(os.path.join(folder, "users.json"), "r") as f:
         USERS = json.load(f)
-    with open(os.path.join(folder, "info.json"), "r") as f:
-        info = json.load(f)
+    with open(os.path.join(folder, "infos.json"), "r") as f:
+        infos = json.load(f)
 except:
-    info = defaultdict(lambda : deepcopy(initial_info))
     USERS = ["김서현", "김우주", "Anonymous"]
-    for user in USERS:
-        print(info[user])
+    infos = {user: Info(name=user) for user in USERS}
 
 MD = """# 포켓몬 퀴즈 🦑
 ## 공부 방법
@@ -117,43 +114,43 @@ with gr.Blocks() as demo:
         if user.strip() == "":
             raise gr.Error("사용자 이름을 입력하세요.")
         message = message.strip()
-        done = info[user]['done']
+        done = infos[user].done
         start, end = GEN_RANGE[gen]
         sdf = df[start:end]
         pokemons_set = sdf[sdf['types'].apply(lambda x: (types in x)) | (types == "모든 타입")]
         pokemons_set = pokemons_set.to_dict("records")
         global random_buttons
         if done:
-            info[user]['history'] = []
+            infos[user].history = []
             if "퀴즈시작" == message.replace(" ", ""):
                 q, a = get_question_answer(pokemons_set, user)
                 bot_message = f"퀴즈를 시작합니다.\n{q}"
-                info[user]['answer'] = a
-                info[user]['done'] = False
-                info[user]['score'] = 0
-                info[user]['count'] = 0
-                info[user]['time'] = time.time()
+                infos[user].answer = a
+                infos[user].done = False
+                infos[user].score = 0
+                infos[user].count = 0
+                infos[user].time = time.time()
             else:
                 bot_message = "퀴즈를 시작하고 싶으시면, **퀴즈 시작** 버튼을 누르세요."
         else:
-            if info[user]['answer'] == message:
+            if infos[user].answer == message:
                 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']:3.1f}점\n- 소요 시간: {time.time() - info[user]['time']:4.3f}초"
+                infos[user].answer = a
+                infos[user].score += 1
+                infos[user].count += 1
+                bot_message = f"🎉정답입니다! 다음 문제입니다.\n{q}\n- 현재 점수: {infos[user].score:3.1f}점\n- 소요 시간: {time.time() - infos[user].time:4.3f}초"
 
             elif "포기하기" == message.replace(" ", ""):
                 bot_message = f"퀴즈를 강제 종료합니다."
-                info[user]['done'] = True
+                infos[user].done = True
             elif "문제 넘어가기" == message:
-                info[user]['count'] += 1
+                infos[user].count += 1
                 q, a = get_question_answer(pokemons_set, user)
-                info[user]['answer'] = a
+                infos[user].answer = a
                 bot_message = f"문제를 넘어갑니다. 다음문제입니다.\n{q}"
             else:
                 hint1 = ""
-                for i, y in enumerate(info[user]['answer']):
+                for i, y in enumerate(infos[user].answer):
                     if i < len(message):
                         if message[i] == y:
                             hint1 += y
@@ -162,41 +159,41 @@ with gr.Blocks() as demo:
                     else:
                         hint1 += "X"
                 bot_message = f"***{message}***!? 🧐 다시 한번 생각해보세요.\n힌트: {hint1}"
-                info[user]['score'] -= 0.1
-
-            gr.Info(f"{user}님, 총 {quiz_count}개 문제 중 {info[user]['count']+1}번째 문제 진행 중입니다.")
-
-            if quiz_count == info[user]['count']:
-                bot_message = f"모든 퀴즈를 다 풀었습니다. 점수는 {info[user]['score']:3.1f}점 입니다."
-                info[user]['done'] = True
-                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'])
-                    info[user]['best_time'] = round(info[user]['best_time'], 2)
-                    info[user]['comment'] = f"{gen.replace(' ', '')}+{types.replace(' ', '')}"
-                with open("info.json", "w") as f:
-                    json.dump(info, f, indent=4)
+                infos[user].score -= 1
+
+            gr.Info(f"{user}님, 총 {quiz_count}개 문제 중 {infos[user].count+1}번째 문제 진행 중입니다.")
+
+            if quiz_count == infos[user].count:
+                bot_message = f"모든 퀴즈를 다 풀었습니다. 점수는 {infos[user].score:3.1f}점 입니다."
+                infos[user].done = True
+                if infos[user].score >= infos[user].best_score:
+                    infos[user].best_score = infos[user].score
+                    infos[user].best_time = min(time.time() - infos[user].time, infos[user].best_time)
+                    infos[user].best_time = round(infos[user].best_time, 2)
+                    infos[user].comment = f"{gen.replace(' ', '')}+{types.replace(' ', '')}"
+                with open("infos.json", "w") as f:
+                    json.dump({k: v.model_dump() for k, v in infos.items()}, f, indent=4, ensure_ascii=False)
                 with open("users.json", "w") as f:
-                    json.dump(USERS, f, indent=4)
-                artifact = wandb.Artifact("settings", type="info")
-                artifact.add_file("info.json")
+                    json.dump(USERS, f, indent=4, ensure_ascii=False)
+                artifact = wandb.Artifact("settings", type="History")
+                artifact.add_file("infos.json")
                 artifact.add_file("users.json")
                 run.log_artifact(artifact)
-                info[user]['history'] = []
+                infos[user].history = []
 
-        info[user]['history'].append((message, bot_message))
-        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])
+        infos[user].history.append((message, bot_message))
+        leader_board = sorted(infos.items(), key=lambda x: (x[1].best_score, -x[1].best_time), reverse=True)
+        lbdf = pd.DataFrame([dict(**a[1].model_dump()) for a in leader_board])
         lbdf.rename(columns={'name': "이름", 'best_score': "최고점수", 'best_time': "시간기록", 'comment': "퀴즈유형"}, inplace=True)
         lbdf.index += 1
         md = lbdf[['이름', '최고점수', '시간기록', '퀴즈유형']]
-        if info[user]['done']:
+        if infos[user].done:
             btn = gr.Button(value="퀴즈 시작", label="퀴즈 시작")
         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
+        random_buttons = [gr.Button(value=c) for c in infos[user].candidates]
+        return btn, infos[user].history, md, *random_buttons
 
     play.click(respond, 
                inputs=[play, chatbot, user, quiz_count, generation, poke_types],
@@ -215,19 +212,19 @@ with gr.Blocks() as demo:
         if user not in USERS:
             USERS += [user]
         new_dropdown = gr.Dropdown(sorted(USERS), value=user, label="사용자", info="당신은 누구신가요?", allow_custom_value=True)
-        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])
+        leader_board = sorted(infos.items(), key=lambda x: (x[1].best_score, -x[1].best_time), reverse=True)
+        lbdf = pd.DataFrame([dict(**a[1].model_dump()) for a in leader_board])
         lbdf.rename(columns={'name': "이름", 'best_score': "최고점수", 'best_time': "시간기록", 'comment': "퀴즈유형"}, inplace=True)
         lbdf.index += 1
         md = lbdf[['이름', '최고점수', '시간기록', '퀴즈유형']]
 
-        if info[user]['done']:
+        if infos[user].done:
             btn = gr.Button(value="퀴즈 시작", label="퀴즈 시작")
         else:
             btn = gr.Button(value="포기하기", label="퀴즈 중단")
 
-        random_buttons = [gr.Button(value=c) for c in info[user]['candidates']]
-        return new_dropdown, btn, md, gr.Dropdown(USERS, value=user, label="사용자", info="당신은 누구신가요?", allow_custom_value=True), info[user]['history'], *random_buttons
+        random_buttons = [gr.Button(value=c) for c in infos[user].candidates]
+        return new_dropdown, btn, md, gr.Dropdown(USERS, value=user, label="사용자", info="당신은 누구신가요?", allow_custom_value=True), infos[user].history, *random_buttons
     
     user.blur(update_table,
               inputs=user,

utils.py

@@ -0,0 +1,23 @@
+from pydantic import BaseModel
+
+initial_info = {"done" : True,
+                "score": 0, "count": 0,
+                "best_score": 0, "best_time": float("inf"),
+                "time": 0.0, "comment": "",
+                "history": [],
+                "candidates": ['1번', '2번', '3번', '4번']}
+
+class Info(BaseModel):
+    done: bool = True
+    score: int = 0
+    count: int = 0
+    answer: str = None
+    best_score: int = 0
+    best_time: float = float("inf")
+    time: float = 0.0
+    comment: str = ""
+    history: list = []
+    candidates: list = ['1번', '2번', '3번', '4번']
+    name: str = None
+    generations: list = []
+    types: list = []