Update chat_session/state_orchestrator.py

chat_session/state_orchestrator.py

@@ -87,6 +87,9 @@ class StateOrchestrator:
         self.event_bus.subscribe(EventType.SESSION_STARTED, self._handle_session_started)
         self.event_bus.subscribe(EventType.SESSION_ENDED, self._handle_session_ended)
 
+        # ✅ WebSocket events
+        self.event_bus.subscribe(EventType.WEBSOCKET_DISCONNECTED, self._handle_websocket_disconnected)
+        
         # STT events
         self.event_bus.subscribe(EventType.STT_READY, self._handle_stt_ready)
         self.event_bus.subscribe(EventType.STT_RESULT, self._handle_stt_result)
@@ -106,6 +109,24 @@ class StateOrchestrator:
         # Error events
         self.event_bus.subscribe(EventType.CRITICAL_ERROR, self._handle_critical_error)
 
+    async def _handle_websocket_disconnected(self, event: Event):
+        """Handle WebSocket disconnection"""
+        session_id = event.session_id
+        context = self.sessions.get(session_id)
+        
+        if not context:
+            return
+            
+        log_info(f"🔌 Handling WebSocket disconnect | session_id={session_id}, state={context.state.value}")
+        
+        # Eğer conversation aktifse, önce conversation'ı sonlandır
+        if context.state not in [ConversationState.IDLE, ConversationState.ENDED]:
+            await self._handle_conversation_ended(Event(
+                type=EventType.CONVERSATION_ENDED,
+                session_id=session_id,
+                data={"reason": "websocket_disconnected"}
+            ))
+            
     async def _handle_conversation_started(self, event: Event) -> None:
         """Handle conversation start within existing session"""
         session_id = event.session_id