Update websocket_handler.py

websocket_handler.py

@@ -409,16 +409,27 @@ class RealtimeSession:
             
         elif action == "audio_ended":
             # Audio playback ended on client
+            log_info(f"🎵 Client reported audio ended, current state: {session.state.value}", session_id=session.session.session_id)
+            
             if session.state == ConversationState.PLAYING_AUDIO:
-                log_info(f"🎵 Client reported audio ended", session_id=session.session.session_id)
+                # State'i listening'e çevir
                 await session.change_state(ConversationState.LISTENING)
                 await websocket.send_json({
                     "type": "state_change",
                     "from": "playing_audio",
                     "to": "listening"
                 })
-                # STT'yi yeniden başlat
-                success = await session.restart_stt_if_needed()
+                
+                # STT'yi başlat
+                log_info(f"🎤 Starting STT after audio playback ended", session_id=session.session.session_id)
+                
+                # Önce mevcut STT varsa temizle
+                if session.stt_manager:
+                    await session.stop_stt_streaming()
+                    await asyncio.sleep(0.1)  # Kısa bekleme
+                
+                # Yeni STT başlat
+                success = await session.initialize_stt()
                 
                 # STT hazır olduğunda sinyal gönder
                 if success and session.is_streaming:
@@ -428,12 +439,14 @@ class RealtimeSession:
                         "message": "STT is ready to receive audio"
                     })
                 else:
-                    log_error(f"❌ STT not ready after restart", session_id=session.session.session_id)
+                    log_error(f"❌ STT initialization failed", session_id=session.session.session_id)
                     await websocket.send_json({
                         "type": "error",
                         "error_type": "stt_init_failed",
                         "message": "Failed to initialize STT after audio playback"
                     })
+            else:
+                log_warning(f"⚠️ audio_ended received but state is not playing_audio: {session.state.value}", session_id=session.session.session_id)
                 
         elif action == "restart_stt":
             # Manual STT restart request
@@ -441,15 +454,6 @@ class RealtimeSession:
             await session.stop_stt_streaming()
             await session.restart_stt_if_needed()
 
-    async def cleanup(self):
-        """Clean up resources"""
-        try:
-            self.is_websocket_active = False
-            await self.stop_stt_streaming()  # STT'yi düzgün durdur
-            log_info(f"Cleaned up realtime session", session_id=self.session.session_id)
-        except Exception as e:
-            log_warning(f"Cleanup error", error=str(e), session_id=self.session.session_id)
-
 # ========================= MAIN HANDLER =========================
 async def websocket_endpoint(websocket: WebSocket, session_id: str):
     """Main WebSocket endpoint for real-time conversation"""