Update app.py

app.py

@@ -22,6 +22,7 @@ GEN_RANGE = {
     "9세대": [906, 1017]
 }
 
+USERS = ["June", "Sean", "Woojoo", "Taejoo", "Dummy"]
 QUESTION_TEMPLATE = {"question": "다음 포켓몬의 이름은 뭘까요?![]({img_url})", "answer": "{name}"}
 
 def get_question_answer(pokemons_set):
@@ -34,52 +35,70 @@ def get_question_answer(pokemons_set):
     a = QUESTION_TEMPLATE['answer'].format(name=name)
     return q, a
 
+info = {u: {"done" : True, "score": 0, "count": 0} for u in USERS}
 MD = """# 포켓몬 퀴즈
+## 사용방법
+
+1. 사용자를 선택하세요.
+2. 총 퀴즈 개수를 선택하세요.
+3. 포켓몬 세대를 선택하세요.
+4. 포켓몬 타입을 선택하세요.
+
+## 점수판
+{content}
 """
 with gr.Blocks() as demo:
-    quiz_start = gr.State(value=False)
-    score = gr.State(value=0)
     answer = gr.State(value="")
     with gr.Row():
         with gr.Column():
-            gr.Markdown(MD)
+            markdown = gr.Markdown(MD.format(content='\n'.join([f"- {u}({info[u]['score']}점)" for u in USERS])))
+            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="1세대", label="포켓몬 세대", info="원하는 포켓몬 세대를 선택하세요."
+                            [f"{k}세대" for k in range(1, 10)] + ["모든 세대"], value="모든 세대", label="포켓몬 세대", info="원하는 포켓몬 세대를 선택하세요."
                         )
                 poke_types = gr.Dropdown(
                             pokemons_types, value="모든 타입", label="포켓몬 타입", info="원하는 포켓몬 타입을 선택하세요."
                     )
             chatbot = gr.Chatbot(bubble_full_width=False)
-            msg = gr.Textbox(value="퀴즈 시작")
-            clear = gr.ClearButton([msg, chatbot, quiz_start])
+            msg = gr.Textbox(value="퀴즈 시작", label="답")
+            clear = gr.ClearButton([msg, chatbot])
 
-    def respond(gen, types, message, chat_history, request: gr.Request):
+    def respond(user, quiz_count, gen, types, message, chat_history, request: gr.Request):
+        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 not quiz_start.value:
+        if done:
             if "퀴즈 시작" == message:
                 q, a = get_question_answer(pokemons_set)
                 bot_message = f"퀴즈를 시작합니다.\n{q}"
                 answer.value = a
-                quiz_start.value = True
+                info[user]['done'] = False
+                info[user]['count'] = 0
             else:
                 bot_message = "퀴즈를 시작하고 싶으시면, **퀴즈 시작**이라고 말씀해주세요."
         else:
             if answer.value == message:
                 q, a = get_question_answer(pokemons_set)
                 answer.value = a
+                info[user]['score'] += 1
+                info[user]['count'] += 1
                 bot_message = f"🎉정답입니다! 다음 문제입니다.\n{q}"
+                if quiz_count == info[user]['count']:
+                    bot_message = f"모든 퀴즈를 다 풀었습니다. 점수는 {info[user]['score']}점 입니다."
             else:
                 bot_message = f"***{message}***!? 🧐 다시 한번 생각해보세요."
+                info[user]['score'] = -0.1
 
         chat_history.append((message, bot_message))
-        return "", chat_history
+        print(info)
+        return "", chat_history, MD.format(content='\n'.join([f"{u}({info[u]['score']}점)" for u in USERS]))
     
-    msg.submit(respond, [generation, poke_types, msg, chatbot], [msg, chatbot])
+    msg.submit(respond, [user, quiz_count, generation, poke_types, msg, chatbot], [msg, chatbot, markdown])
 
-demo.queue(concurrency_count=3)
+demo.queue(concurrency_count=1)
 demo.launch()