Update app.py

app.py

@@ -694,8 +694,7 @@ with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
     stage_hist = gr.Textbox(visible=dev_mod, value="stage history: ", interactive=False, label='stage history')
     chat_hist = gr.Textbox(visible=dev_mod, interactive=False, label='chatting_history')
     response_examples_text = gr.Textbox(visible=dev_mod, interactive=False, value="이번 주에 친구들과 모임이 있는데, 훌륭한 와인 한 병을 추천해줄래?|입문자에게 좋은 와인을 추천해줄래?|연인과 가기 좋은 와인바를 알려줘", label='response_examples')
-    # btn = gr.Button("Flag", visible=dev_mod)
-    # hf_writer.setup(components=[chat_hist, stage_hist, response_examples_text], flagging_dir="chatwine-korean")
+    hf_writer.setup(components=[chat_hist, stage_hist, response_examples_text], flagging_dir="chatwine-korean")
 
     def click_flag_btn(*args):
         hf_writer.flag(flag_data=[*args])
@@ -766,10 +765,9 @@ with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
         bot_chat, [msg, chat_hist, chatbot, cur_stage], [chatbot, chat_hist]
     ).then(
         bot_response_pred, chat_hist, [user_response_examples, response_examples_text, msg]
+    ).then(
+        click_flag_btn, [chat_hist, stage_hist, response_examples_text], None
     )
-    # .then(
-    #     click_flag_btn, [chat_hist, stage_hist, response_examples_text], None
-    # )
 
 
 
@@ -781,10 +779,9 @@ with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
         bot_chat, [msg, chat_hist, chatbot, cur_stage], [chatbot, chat_hist]
     ).then(
         bot_response_pred, chat_hist, [user_response_examples, response_examples_text, msg]
+    ).then(
+        click_flag_btn, [chat_hist, stage_hist, response_examples_text], None
     )
-    # .then(
-    #     click_flag_btn, [chat_hist, stage_hist, response_examples_text], None
-    # )
 
 
 
@@ -794,6 +791,5 @@ with gr.Blocks(css='#chatbot .overflow-y-auto{height:750px}') as demo:
         outputs=[user_response_examples, cur_stage, stage_hist, chat_hist, response_examples_text], 
         queue=False)
     user_response_examples.click(load_example, inputs=[response_examples_text, user_response_examples], outputs=[msg], queue=False)
-    # btn.click(lambda *args: hf_writer.flag(args), [chat_hist, stage_hist, response_examples_text], None, preprocess=False)
 demo.queue(concurrency_count=100)
 demo.launch()