새 수정

chatbot_utils.py

@@ -3,7 +3,7 @@ from openai import OpenAI
 import json
 from datetime import datetime
 from scenario_handler import ScenarioHandler
-import asyncio
+import time
 
 client = OpenAI(api_key="sk-proj-3IEelWYK3Wl251k9qNriT3BlbkFJ9M7GpUGBijobUj1LETdu")
 
@@ -32,18 +32,6 @@ def chatbot_response(response, handler_type='offender', n=1):
     choices = [choice.message.content for choice in api_response.choices]
     return choices[0], choices
 
-async def async_chatbot_response(response, handler_type='offender', n=1):
-    return chatbot_response(response, handler_type, n)
-
-async def get_both_responses(user_input):
-    victim_task = asyncio.create_task(async_chatbot_response(user_input, 'victim', n=3))
-    offender_task = asyncio.create_task(async_chatbot_response(user_input, 'offender', n=1))
-    
-    victim_response, victim_choices = await victim_task
-    offender_response, _ = await offender_task
-    
-    return victim_response, victim_choices, offender_response
-
 def save_history(history):
     os.makedirs('logs', exist_ok=True)
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -52,12 +40,22 @@ def save_history(history):
         json.dump(history, file, ensure_ascii=False, indent=4)
     print(f"History saved to {filename}")
 
-async def process_user_input(user_input, chatbot_history):
+def process_user_input(user_input, chatbot_history):
     if user_input.strip().lower() == "종료":
         save_history(chatbot_history)
         return chatbot_history + [("종료", "실험에 참가해 주셔서 감사합니다. 후속 지시를 따라주세요")], []
 
-    victim_response, victim_choices, offender_response = await get_both_responses(user_input)
-    new_history = chatbot_history + [(user_input, victim_response), (None, offender_response)]
+    # First, add the user's input to the history
+    new_history = chatbot_history + [(user_input, None)]
     
-    return new_history, victim_choices
+    # Then, get the offender's response
+    offender_response, _ = chatbot_response(user_input, 'offender', n=1)
+    
+    # Generate victim choices for the next turn
+    _, victim_choices = chatbot_response(offender_response, 'victim', n=3)
+    
+    return new_history, offender_response, victim_choices
+
+def delayed_offender_response(history, offender_response):
+    # This function will be called after a delay to add the offender's response
+    return history + [(None, offender_response)]

gradio_interface.py

@@ -1,41 +1,39 @@
 import gradio as gr
-from chatbot_utils import process_user_input, get_both_responses
-import asyncio
+from chatbot_utils import process_user_input, chatbot_response, delayed_offender_response
 
 def create_interface():
-    async def handle_user_response(user_input, selected_response, chatbot_history):
+    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() == "종료":
-            new_history = chatbot_history + [(input_text, "실험에 참가해 주셔서 감사합니다. 후속 지시를 따라주세요")]
             return new_history, gr.update(choices=[], interactive=False)
         
-        new_history, victim_choices = await process_user_input(input_text, chatbot_history)
-        return new_history, gr.update(choices=victim_choices)
+        # 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)")
+        
+        return new_history, gr.update(choices=choices)
 
-    async def handle_case_selection():
+    def handle_case_selection():
         initial_message = "발표가 망한 건 제 잘못도 좀 있지만, 팀장님은 아무것도 안 하면서 이러는 건 선 넘은거죠"
+        chatbot_history = [(initial_message, None)]
+        offender_response, _ = chatbot_response(initial_message, 'offender', n=1)
+        _, victim_choices = chatbot_response(offender_response, 'victim', n=3)
         
-        victim_response, victim_choices, offender_response = await get_both_responses(initial_message)
+        # 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)")
         
-        chatbot_history = [(initial_message, victim_response), (None, offender_response)]
         return 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="입력하세요")
-        response_choices = gr.Dropdown(label="답안을 골라주세요", choices=[], interactive=True)
-        submit_button = gr.Button(value="전송")
+        user_input = gr.Textbox(label="Your input")
+        response_choices = gr.Dropdown(label="Select a Response", choices=[], interactive=True)
+        submit_button = gr.Button(value="Submit")
 
+        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])
 
-    return demo
-
-# Gradio의 launch 함수를 사용하여 인터페이스를 실행합니다.
-if __name__ == "__main__":
-    demo = create_interface()
-    demo.launch()
+    return demo