Update app.py

app.py

@@ -6,50 +6,24 @@ import random
 import json
 import time
 
-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']]))
-df = pd.DataFrame(pokemons)
-GEN_RANGE = {
-    "모든 세대": [1, 1017],
-    "1세대": [1, 151],
-    "2세대": [152, 251],
-    "3세대": [252, 386],
-    "4세대": [387, 493],
-    "5세대": [494, 649],
-    "6세대": [650, 721],
-    "7세대": [722, 809],
-    "8세대": [810, 905],
-    "9세대": [906, 1017]
-}
-
 USERS = ["June", "Sean", "Woojoo", "Taejoo", "Dummy"]
 QUESTION_TEMPLATE = {"question": "다음 포켓몬의 이름은 뭘까요?![]({img_url})", "answer": "{name}"}
 
-def get_question_answer(pokemons_set):
-    chosen = random.choice(pokemons_set)
-    name = chosen['name'].replace("♀", "암컷").replace("♂", "수컷")
-    image_path = chosen['image_path']
-    answer.value = QUESTION_TEMPLATE['answer'].format(name=name)
-    img_url = f"https://huggingface.co/spaces/yoon-gu/pokemon/resolve/main/{image_path}"
-    q = QUESTION_TEMPLATE["question"].format(img_url=img_url)
-    a = QUESTION_TEMPLATE['answer'].format(name=name)
+def get_question_answer():
+    a = random.randint(0, 10)
+    b = random.randint(0, 10)
+    c = a + b
+    q = f"$${a} + {b} = \square$$"
+    a = f"{c}"
     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}
 
-MD = """# 포켓몬 퀴즈
-## 공부 방법
-
-아래 포켓몬 도감을 보고 공부하면 도움이 됩니다.
-- https://huggingface.co/spaces/yoon-gu/pokemon
-
+MD = """# 수학 퀴즈
 ## 사용방법
 
 1. 사용자를 선택하세요.
 2. 총 퀴즈 개수를 선택하세요.
-3. 포켓몬 세대를 선택하세요.
-4. 포켓몬 타입을 선택하세요.
 
 ## 점수판
 {content}
@@ -63,18 +37,6 @@ with gr.Blocks() as demo:
             user = gr.Radio(USERS, value="Dummy", label="사용자", info="당신은 누구신가요?")
             quiz_count = gr.Radio([10, 20, 30], value=10, label="총 퀴즈 개수", info="퀴즈를 몇 개 풀 예정인가요?")
         with gr.Column():
-            with gr.Row():
-                generation = gr.Dropdown(
-                            [f"{k}세대" for k in range(1, 10)] + ["모든 세대"],
-                            value="모든 세대",
-                            label="포켓몬 세대",
-                            info="원하는 포켓몬 세대를 선택하세요."
-                        )
-                poke_types = gr.Dropdown(
-                            pokemons_types, value="모든 타입",
-                            label="포켓몬 타입",
-                            info="원하는 포켓몬 타입을 선택하세요."
-                    )
             with gr.Row():
                 play = gr.Button(value="퀴즈 시작", label="퀴즈 시작")
                 skip = gr.Button(value="문제 넘어가기", label="문제 스킵")
@@ -84,17 +46,13 @@ with gr.Blocks() as demo:
                                                 "https://huggingface.co/spaces/yoon-gu/pokemon/resolve/main/images/No_0155_브케인.png"])
             msg = gr.Textbox(placeholder="문제의 답을 입력하세요.", label="답")
 
-    def respond(message, chat_history, user, quiz_count, gen, types, request: gr.Request):
+    def respond(message, chat_history, user, quiz_count, request: gr.Request):
         message = message.strip()
         done = info[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")
         if done:
             if "퀴즈시작" == message.replace(" ", ""):
-                q, a = get_question_answer(pokemons_set)
-                bot_message = f"퀴즈를 시작합니다.\n{q}"
+                q, a = get_question_answer()
+                bot_message = f"퀴즈를 시작합니다. 빈칸에 들어갈 숫자를 맞춰주세요.\n{q}"
                 answer.value = a
                 info[user]['done'] = False
                 info[user]['score'] = 0
@@ -104,31 +62,22 @@ with gr.Blocks() as demo:
                 bot_message = "퀴즈를 시작하고 싶으시면, **퀴즈 시작** 버튼을 누르세요."
         else:
             if answer.value == message:
-                q, a = get_question_answer(pokemons_set)
+                q, a = get_question_answer()
                 answer.value = 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- 현재 점수: {info[user]['score']}점\n- 소요 시간: {time.time() - info[user]['time']:4.3f}초\n. 빈칸에 들어갈 숫자를 맞���주세요.{q}"
 
             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()
                 answer.value = a
-                bot_message = f"문제를 넘어갑니다. 다음문제입니다.\n{q}"
+                bot_message = f"문제를 넘어갑니다. 다음 문제입니다.\n{q}"
             else:
-                hint1 = ""
-                for i, y in enumerate(answer.value):
-                    if i < len(message):
-                        if message[i] == y:
-                            hint1 += y
-                        else:
-                            hint1 += "X"
-                    else:
-                        hint1 += "X"
-                bot_message = f"***{message}***!? 🧐 다시 한번 생각해보세요.\n힌트: {hint1}"
+                bot_message = f"***{message}***!? 🧐 다시 한번 생각해보세요."
                 info[user]['score'] -= 0.1
             
             if quiz_count == info[user]['count']:
@@ -147,13 +96,13 @@ with gr.Blocks() as demo:
         return "", chat_history, MD.format(content=md)
 
     play.click(respond, 
-               inputs=[play, chatbot, user, quiz_count, generation, poke_types],
+               inputs=[play, chatbot, user, quiz_count],
                outputs=[msg, chatbot, markdown])
     skip.click(respond, 
-               inputs=[skip, chatbot, user, quiz_count, generation, poke_types],
+               inputs=[skip, chatbot, user, quiz_count],
                outputs=[msg, chatbot, markdown])
     stop.click(respond, 
-               inputs=[stop, chatbot, user, quiz_count, generation, poke_types],
+               inputs=[stop, chatbot, user, quiz_count],
                outputs=[msg, chatbot, markdown])
     def update_table():
         leader_board = sorted(info.items(), key=lambda x: (x[1]['best_score'], -x[1]['best_time']), reverse=True)
@@ -164,7 +113,7 @@ with gr.Blocks() as demo:
     demo.load(update_table,
               inputs=None,
               outputs=markdown)
-    msg.submit(respond, [msg, chatbot, user, quiz_count, generation, poke_types], [msg, chatbot, markdown])
+    msg.submit(respond, [msg, chatbot, user, quiz_count], [msg, chatbot, markdown])
 
 demo.queue(concurrency_count=1)
 demo.launch()