객관식으로 변경

app.py

@@ -37,7 +37,12 @@ def get_question_answer(pokemons_set):
     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)
-    return q, a
+
+    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
 
 initial_info = {"done" : True,
                 "score": 0, "count": 0,
@@ -99,7 +104,11 @@ with gr.Blocks() as demo:
             chatbot = gr.Chatbot(bubble_full_width=False,
                                  avatar_images=["https://huggingface.co/spaces/yoon-gu/pokemon/resolve/main/images/No_0001_이상해씨.png",
                                                 "https://huggingface.co/spaces/yoon-gu/pokemon/resolve/main/images/No_0155_브케인.png"])
-            msg = gr.Textbox(placeholder="문제의 답을 입력하세요.", label="답")
+            with gr.Row():
+                button1 = gr.Button(value="1번")
+                button2 = gr.Button(value="2번")
+                button3 = gr.Button(value="3번")
+                button4 = gr.Button(value="4번")
 
     def respond(message, chat_history, user, quiz_count, gen, types, request: gr.Request):
         message = message.strip()
@@ -110,7 +119,7 @@ with gr.Blocks() as demo:
         pokemons_set = pokemons_set.to_dict("records")
         if done:
             if "퀴즈시작" == message.replace(" ", ""):
-                q, a = get_question_answer(pokemons_set)
+                q, a, random_buttons = get_question_answer(pokemons_set)
                 bot_message = f"퀴즈를 시작합니다.\n{q}"
                 info[user]['answer'] = a
                 info[user]['done'] = False
@@ -121,7 +130,7 @@ with gr.Blocks() as demo:
                 bot_message = "퀴즈를 시작하고 싶으시면, **퀴즈 시작** 버튼을 누르세요."
         else:
             if info[user]['answer'] == message:
-                q, a = get_question_answer(pokemons_set)
+                q, a, random_buttons = get_question_answer(pokemons_set)
                 info[user]['answer'] = a
                 info[user]['score'] += 1
                 info[user]['count'] += 1
@@ -167,7 +176,7 @@ with gr.Blocks() as demo:
                 artifact.add_file("users.json")
                 run.log_artifact(artifact)
 
-        chat_history.append((message, bot_message))
+        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])
@@ -178,15 +187,21 @@ with gr.Blocks() as demo:
             btn = gr.Button(value="퀴즈 시작", label="퀴즈 시작")
         else:
             btn = gr.Button(value="포기하기", label="퀴즈 중단")
-        info[user]['history'] = chat_history
-        return btn, "", chat_history, md
+
+        
+        return btn, info[user]['history'], md, *random_buttons
 
     play.click(respond, 
                inputs=[play, chatbot, user, quiz_count, generation, poke_types],
-               outputs=[play, msg, chatbot, leader_board])
+               outputs=[play, chatbot, leader_board, button1, button2, button3, button4])
     skip.click(respond,
                inputs=[skip, chatbot, user, quiz_count, generation, poke_types],
-               outputs=[play, msg, chatbot, leader_board])
+               outputs=[play, chatbot, leader_board, button1, button2, button3, button4])
+    
+    for btn in [button1, button2, button3, button4]:
+        btn.click(respond, 
+                inputs=[btn, chatbot, user, quiz_count, generation, poke_types],
+                outputs=[play, chatbot, leader_board, button1, button2, button3, button4])
     
     def rs_change(rs):
         global USERS
@@ -213,7 +228,6 @@ with gr.Blocks() as demo:
     demo.load(update_table,
               inputs=user,
               outputs=[play, leader_board, user, chatbot])
-    msg.submit(respond, [msg, chatbot, user, quiz_count, generation, poke_types], [play, msg, chatbot, leader_board])
 
-demo.queue(concurrency_count=1)
+demo.queue(concurrency_count=3)
 demo.launch()