버튼을 업데이트하는 방식으로!

app.py

@@ -95,7 +95,6 @@ with gr.Blocks() as demo:
             with gr.Row():
                 play = gr.Button(value="퀴즈 시작", label="퀴즈 시작")
                 skip = gr.Button(value="문제 넘어가기", label="문제 스킵")
-                stop = gr.Button(value="퀴즈 종료", label="퀴즈 종료")
             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"])
@@ -127,7 +126,7 @@ with gr.Blocks() as demo:
                 info[user]['count'] += 1
                 bot_message = f"🎉정답입니다! 다음 문제입니다.\n{q}\n- 현재 점수: {info[user]['score']}점\n- 소요 시간: {time.time() - info[user]['time']:4.3f}초"
 
-            elif "퀴즈종료" == message.replace(" ", ""):
+            elif "포기하기" == message.replace(" ", ""):
                 bot_message = f"퀴즈를 강제 종료합니다."
                 info[user]['done'] = True
             elif "문제 넘어가기" == message:
@@ -157,7 +156,7 @@ with gr.Blocks() as demo:
                     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}+{types}"
+                    info[user]['comment'] = f"{gen.replace(' ', '')}+{types.replace(' ', '')}"
                 with open("info.json", "w") as f:
                     json.dump(info, f, indent=4)
                 with open("users.json", "w") as f:
@@ -174,23 +173,25 @@ with gr.Blocks() as demo:
         lbdf.rename(columns={'name': "이름", 'best_score': "최고점수", 'best_time': "시간기록", 'comment': "퀴즈유형"}, inplace=True)
         lbdf.index += 1
         md = lbdf[['이름', '최고점수', '시간기록', '퀴즈유형']]
-        return "", chat_history, md
+        if info[user]['done']:
+            btn = gr.Button(value="퀴즈 시작", label="퀴즈 시작")
+        else:
+            btn = gr.Button(value="포기하기", label="퀴즈 중단")
+        return btn, "", chat_history, md
 
     play.click(respond, 
                inputs=[play, chatbot, user, quiz_count, generation, poke_types],
-               outputs=[msg, chatbot, leader_board])
+               outputs=[play, msg, chatbot, leader_board])
     skip.click(respond, 
                inputs=[skip, chatbot, user, quiz_count, generation, poke_types],
-               outputs=[msg, chatbot, leader_board])
-    stop.click(respond, 
-               inputs=[stop, chatbot, user, quiz_count, generation, poke_types],
-               outputs=[msg, chatbot, leader_board])    
+               outputs=[play, msg, chatbot, leader_board])
     
     def rs_change(rs):
         global USERS
         if rs not in USERS:
             USERS += [rs]
-        return gr.update(choices=sorted(USERS), value=rs)
+        new_dropdown = gr.Dropdown(sorted(USERS), value=rs, label="사용자", info="당신은 누구신가요?", allow_custom_value=True)
+        return new_dropdown
 
     user.blur(rs_change, user, user)
 
@@ -199,13 +200,18 @@ with gr.Blocks() as demo:
         lbdf = pd.DataFrame([dict(**a[1], name=a[0]) for a in leader_board])
         lbdf.rename(columns={'name': "이름", 'best_score': "최고점수", 'best_time': "시간기록", 'comment': "퀴즈유형"}, inplace=True)
         lbdf.index += 1
-        print(len(lbdf))
         md = lbdf[['이름', '최고점수', '시간기록', '퀴즈유형']]
-        return md, gr.Dropdown(USERS, value=user, label="사용자", info="당신은 누구신가요?", allow_custom_value=True)
+
+        if info[user]['done']:
+            btn = gr.Button(value="퀴즈 시작", label="퀴즈 시작")
+        else:
+            btn = gr.Button(value="포기하기", label="퀴즈 중단")
+
+        return btn, md, gr.Dropdown(USERS, value=user, label="사용자", info="당신은 누구신가요?", allow_custom_value=True)
     demo.load(update_table,
               inputs=user,
-              outputs=[leader_board, user])
-    msg.submit(respond, [msg, chatbot, user, quiz_count, generation, poke_types], [msg, chatbot, leader_board])
+              outputs=[play, leader_board, user])
+    msg.submit(respond, [msg, chatbot, user, quiz_count, generation, poke_types], [play, msg, chatbot, leader_board])
 
 demo.queue(concurrency_count=1)
 demo.launch()