위치 변경 및 2번 나오는 내용 수정

gradio_interface.py

@@ -5,39 +5,47 @@ import time
 def create_interface():
     def handle_user_response(user_input, selected_response, chatbot_history):
         input_text = user_input if user_input else selected_response
-        new_history, offender_response, choices = process_user_input(input_text, chatbot_history)
         
         if input_text.strip().lower() == "종료":
-            return new_history, gr.update(choices=[], interactive=False)
+            chatbot_history.append((input_text, "실험에 참가해 주셔서 감사합니다. 후속 지시를 따라주세요"))
+            return chatbot_history, gr.update(choices=[], interactive=False)
         
         # Add user's input to history
-        new_history = new_history + [(input_text, None)]
+        chatbot_history.append((input_text, None))
+        yield chatbot_history, gr.update(choices=[])  # Immediately show user input
         
-        # 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)
+        # Get offender's response
+        offender_response, _ = chatbot_response(input_text, 'offender', n=1)
         time.sleep(2)  # 2-second delay
-        new_history = new_history + [(None, offender_response)]
-        yield new_history, gr.update(choices=choices)
+        
+        # Add offender's response to history
+        chatbot_history.append((None, offender_response))
+        
+        # Generate victim choices for the next turn
+        _, victim_choices = chatbot_response(offender_response, 'victim', n=3)
+        
+        yield chatbot_history, gr.update(choices=victim_choices)
 
     def handle_case_selection():
         initial_message = "발표가 망한 건 제 잘못도 좀 있지만, 팀장님은 아무것도 안 하면서 이러는 건 선 넘은거죠"
         chatbot_history = [(initial_message, None)]
+        yield chatbot_history, gr.update(choices=[])
+        
         offender_response, _ = chatbot_response(initial_message, 'offender', n=1)
+        time.sleep(2)  # 2-second delay
+        
+        chatbot_history.append((None, offender_response))
         _, victim_choices = chatbot_response(offender_response, 'victim', n=3)
         
-        # 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:
+        case_selection_button = gr.Button("발표가 망한 건 제 잘못도 좀 있지만, 팀장님은 아무것도 안 하면서 이러는 건 선 넘은거죠")
         screen = gr.Chatbot()
-        user_input = gr.Textbox(label="Your input")
-        response_choices = gr.Dropdown(label="Select a Response", choices=[], interactive=True)
-        submit_button = gr.Button(value="Submit")
+        user_input = gr.Textbox(label="이곳에 대답을 입력하세요")
+        response_choices = gr.Dropdown(label="또는 대답을 선택해 주세요", choices=[], interactive=True)
+        submit_button = gr.Button(value="제출")
 
-        case_selection_button = gr.Button("발표가 망한 건 제 잘못도 좀 있지만, 팀장님은 아무것도 안 하면서 이러는 건 선 넘은거죠")
         case_selection_button.click(handle_case_selection, inputs=[], outputs=[screen, response_choices])
         
         submit_button.click(handle_user_response, inputs=[user_input, response_choices, screen], outputs=[screen, response_choices])