Update app.py

app.py

@@ -18,7 +18,6 @@ from fastrtc import (
 )
 from gradio.utils import get_space
 from openai.types.beta.realtime import ResponseAudioTranscriptDoneEvent
-from websockets.exceptions import ConnectionClosedOK
 
 load_dotenv()
 
@@ -28,7 +27,9 @@ SAMPLE_RATE = 24000
 
 
 class OpenAIHandler(AsyncStreamHandler):
-    def __init__(self) -> None:
+    def __init__(
+        self,
+    ) -> None:
         super().__init__(
             expected_layout="mono",
             output_sample_rate=SAMPLE_RATE,
@@ -41,18 +42,16 @@ class OpenAIHandler(AsyncStreamHandler):
     def copy(self):
         return OpenAIHandler()
 
-    async def start_up(self):
+    async def start_up(
+        self,
+    ):
         """Connect to realtime API. Run forever in separate thread to keep connection open."""
         self.client = openai.AsyncOpenAI()
         async with self.client.beta.realtime.connect(
             model="gpt-4o-mini-realtime-preview-2024-12-17"
         ) as conn:
-            # 시스템 프롬프트 추가 및 turn_detection 설정
             await conn.session.update(
-                session={
-                    "system_prompt": "너의 이름은 '비드래프트'이다. 당신은 도움이 되는 정보를 제공하는 조수입니다. 항상 정확하고 친절하게 응답하세요.",
-                    "turn_detection": {"type": "server_vad"},
-                }
+                session={"turn_detection": {"type": "server_vad"}}
             )
             self.connection = conn
             async for event in self.connection:
@@ -74,11 +73,7 @@ class OpenAIHandler(AsyncStreamHandler):
         _, array = frame
         array = array.squeeze()
         audio_message = base64.b64encode(array.tobytes()).decode("utf-8")
-        try:
-            await self.connection.input_audio_buffer.append(audio=audio_message)  # type: ignore
-        except ConnectionClosedOK:
-            # 연결이 정상적으로 종료된 경우 예외를 무시합니다.
-            pass
+        await self.connection.input_audio_buffer.append(audio=audio_message)  # type: ignore
 
     async def emit(self) -> tuple[int, np.ndarray] | AdditionalOutputs | None:
         return await wait_for_item(self.output_queue)
@@ -143,4 +138,4 @@ if __name__ == "__main__":
     else:
         import uvicorn
 
-        uvicorn.run(app, host="0.0.0.0", port=7860)
+        uvicorn.run(app, host="0.0.0.0", port=7860)