애초에 함수 자체가 동기적으로 되어있으니까...

chatbot_utils.py

@@ -3,6 +3,7 @@ from openai import OpenAI
 import json
 from datetime import datetime
 from scenario_handler import ScenarioHandler
+import asyncio
 
 client = OpenAI(api_key="sk-proj-3IEelWYK3Wl251k9qNriT3BlbkFJ9M7GpUGBijobUj1LETdu")
 
@@ -31,6 +32,18 @@ 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")
@@ -39,13 +52,12 @@ def save_history(history):
         json.dump(history, file, ensure_ascii=False, indent=4)
     print(f"History saved to {filename}")
 
-def process_user_input(user_input, chatbot_history):
+async def process_user_input(user_input, chatbot_history):
     if user_input.strip().lower() == "종료":
         save_history(chatbot_history)
         return chatbot_history + [("종료", "실험에 참가해 주셔서 감사합니다. 후속 지시를 따라주세요")], []
 
-    offender_response, _ = chatbot_response(user_input, 'offender', n=1)
-    new_history = chatbot_history + [(user_input, offender_response)]
+    victim_response, victim_choices, offender_response = await get_both_responses(user_input)
+    new_history = chatbot_history + [(user_input, victim_response), (None, offender_response)]
     
-    _, victim_choices = chatbot_response(offender_response, 'victim', n=3)
     return new_history, victim_choices

gradio_interface.py

@@ -1,35 +1,24 @@
 import gradio as gr
-from chatbot_utils import process_user_input, chatbot_response
+from chatbot_utils import process_user_input, get_both_responses
+import asyncio
 
 def create_interface():
-    def handle_user_response(user_input, selected_response, chatbot_history):
+    async def handle_user_response(user_input, selected_response, chatbot_history):
         input_text = user_input if user_input else selected_response
         
         if input_text.strip().lower() == "종료":
-            new_history = chatbot_history + [("종료", "실험에 참가해 주셔서 감사합니다. 후속 지시를 따라주세요")]
+            new_history = chatbot_history + [(input_text, "실험에 참가해 주셔서 감사합니다. 후속 지시를 따라주세요")]
             return new_history, gr.update(choices=[], interactive=False)
         
-        # Get victim's response first
-        victim_response, victim_choices = chatbot_response(input_text, 'victim', n=3)
-        
-        # Then get offender's response
-        offender_response, _ = chatbot_response(input_text, 'offender', n=1)
-        
-        new_history = chatbot_history + [(input_text, victim_response), (None, offender_response)]
+        new_history, victim_choices = await process_user_input(input_text, chatbot_history)
         return new_history, gr.update(choices=victim_choices)
 
-    def handle_case_selection():
+    async def handle_case_selection():
         initial_message = "발표가 망한 건 제 잘못도 좀 있지만, 팀장님은 아무것도 안 하면서 이러는 건 선 넘은거죠"
-        chatbot_history = [(initial_message, None)]
         
-        # Get victim's response first
-        victim_response, victim_choices = chatbot_response(initial_message, 'victim', n=3)
-        chatbot_history.append((None, victim_response))
-        
-        # Then get offender's response
-        offender_response, _ = chatbot_response(initial_message, 'offender', n=1)
-        chatbot_history.append((None, offender_response))
+        victim_response, victim_choices, offender_response = await get_both_responses(initial_message)
         
+        chatbot_history = [(initial_message, victim_response), (None, offender_response)]
         return chatbot_history, gr.update(choices=victim_choices)
 
     with gr.Blocks() as demo:
@@ -44,4 +33,9 @@ def create_interface():
         
         submit_button.click(handle_user_response, inputs=[user_input, response_choices, screen], outputs=[screen, response_choices])
 
-    return demo
+    return demo
+
+# Gradio의 launch 함수를 사용하여 인터페이스를 실행합니다.
+if __name__ == "__main__":
+    demo = create_interface()
+    demo.launch()