버그 픽싱 중

gradio_interface.py

@@ -1,5 +1,6 @@
 import gradio as gr
-from chatbot_utils import process_user_input, chatbot_response, delayed_offender_response
+from chatbot_utils import process_user_input, chatbot_response
+import time
 
 def create_interface():
     def handle_user_response(user_input, selected_response, chatbot_history):
@@ -9,10 +10,14 @@ def create_interface():
         if input_text.strip().lower() == "종료":
             return new_history, gr.update(choices=[], interactive=False)
         
-        # Schedule the offender's response to be added after a delay
-        gr.Result(delayed_offender_response, [new_history, offender_response], _js="() => setTimeout(() => _js_return_(), 2000)")
+        # Add user's input to history
+        new_history = new_history + [(input_text, None)]
         
-        return new_history, gr.update(choices=choices)
+        # Use a yield to first return the user's input, then the offender's response after a delay
+        yield new_history, gr.update(choices=choices)
+        time.sleep(2)  # 2-second delay
+        new_history = new_history + [(None, offender_response)]
+        yield new_history, gr.update(choices=choices)
 
     def handle_case_selection():
         initial_message = "발표가 망한 건 제 잘못도 좀 있지만, 팀장님은 아무것도 안 하면서 이러는 건 선 넘은거죠"
@@ -20,10 +25,11 @@ def create_interface():
         offender_response, _ = chatbot_response(initial_message, 'offender', n=1)
         _, victim_choices = chatbot_response(offender_response, 'victim', n=3)
         
-        # Schedule the offender's response to be added after a delay
-        gr.Result(delayed_offender_response, [chatbot_history, offender_response], _js="() => setTimeout(() => _js_return_(), 2000)")
-        
-        return chatbot_history, gr.update(choices=victim_choices)
+        # Use a yield to first return the initial message, then the offender's response after a delay
+        yield chatbot_history, gr.update(choices=victim_choices)
+        time.sleep(2)  # 2-second delay
+        chatbot_history = chatbot_history + [(None, offender_response)]
+        yield chatbot_history, gr.update(choices=victim_choices)
 
     with gr.Blocks() as demo:
         screen = gr.Chatbot()