Update websocket_handler.py

websocket_handler.py

@@ -189,14 +189,27 @@ class RealtimeSession:
         # Chunk counter için attribute
         self.chunk_counter = 0
         
+        # Session management - YENİ
+        self.stt_session_count = 0
+        self.last_stt_stop_time = None
+        
     async def initialize_stt(self):
         """Initialize STT provider with clean state"""
         try:
+            # Session numarasını artır
+            self.stt_session_count += 1
+            log_info(f"🎤 Initializing STT session #{self.stt_session_count}", session_id=self.session.session_id)
+            
             # Önce mevcut STT'yi tamamen temizle
             await self.stop_stt_streaming()
             
-            # Biraz bekle - temizlik işleminin tamamlanması için
-            await asyncio.sleep(0.1)
+            # Önceki stop'tan bu yana yeterli zaman geçtiğinden emin ol
+            if self.last_stt_stop_time:
+                elapsed = (datetime.now() - self.last_stt_stop_time).total_seconds()
+                if elapsed < 0.5:
+                    wait_time = 0.5 - elapsed
+                    log_info(f"⏳ Waiting {wait_time:.2f}s for proper cleanup", session_id=self.session.session_id)
+                    await asyncio.sleep(wait_time)
             
             # Tüm değişkenleri yeniden başlat
             self.chunk_counter = 0
@@ -251,39 +264,19 @@ class RealtimeSession:
             # Hata durumunda da temizlik yap
             await self.stop_stt_streaming()
             return False
-        
-    async def restart_stt_if_needed(self):
-        """Restart STT if it's not active"""
-        try:
-            # Sadece LISTENING state'inde ve WebSocket aktifse restart yap
-            if not self.is_streaming and self.is_websocket_active and self.state == ConversationState.LISTENING:
-                log_info(f"🔄 Restarting STT stream...", session_id=self.session.session_id)
-                
-                # Önce mevcut stream'i temizle
-                await self.stop_stt_streaming()
-                
-                # Sonra yeniden başlat
-                stt_initialized = await self.initialize_stt()
-                if stt_initialized:
-                    log_info(f"✅ STT stream restarted successfully", session_id=self.session.session_id)
-                    return True
-                else:
-                    log_error(f"❌ Failed to restart STT stream", session_id=self.session.session_id)
-                    return False
-            return True
-        except Exception as e:
-            log_error(f"❌ Error restarting STT", error=str(e), session_id=self.session.session_id)
-            return False
 
     async def stop_stt_streaming(self):
         """Stop STT streaming completely and reset all STT-related variables"""
         try:
+            log_info(f"🛑 Stopping STT session #{self.stt_session_count}", session_id=self.session.session_id)
+            
+            # STT manager varsa durdur
             if self.stt_manager:
-                log_info(f"🛑 Stopping STT stream and resetting all STT data", session_id=self.session.session_id)
-                
-                # STT manager'ı durdur
                 if self.is_streaming:
-                    await self.stt_manager.stop_streaming()
+                    try:
+                        await self.stt_manager.stop_streaming()
+                    except Exception as e:
+                        log_warning(f"⚠️ Error during STT stop_streaming: {e}", session_id=self.session.session_id)
                 
                 # STT manager'ı tamamen sil
                 self.stt_manager = None
@@ -303,37 +296,33 @@ class RealtimeSession:
             if hasattr(self, 'speech_started'):
                 delattr(self, 'speech_started')
             
-            log_info(f"✅ STT stream stopped and all STT data reset", session_id=self.session.session_id)
+            # Stop zamanını kaydet
+            self.last_stt_stop_time = datetime.now()
+            
+            log_info(f"✅ STT session #{self.stt_session_count} stopped and all data reset", session_id=self.session.session_id)
             
         except Exception as e:
-            log_warning(f"⚠️ Error stopping STT stream: {e}", session_id=self.session.session_id)
+            log_error(f"❌ Error in stop_stt_streaming", error=str(e), session_id=self.session.session_id)
             # Hata olsa bile değişkenleri resetle
             self.stt_manager = None
             self.is_streaming = False
             self.chunk_counter = 0
             self.current_transcription = ""
-            await self.audio_buffer.clear()
-            self.silence_detector.reset()
+            if self.audio_buffer:
+                await self.audio_buffer.clear()
+            if self.silence_detector:
+                self.silence_detector.reset()
+            self.last_stt_stop_time = datetime.now()
 
     async def restart_stt_if_needed(self):
         """Restart STT if it's not active"""
         try:
             # Sadece LISTENING state'inde ve WebSocket aktifse restart yap
             if not self.is_streaming and self.is_websocket_active and self.state == ConversationState.LISTENING:
-                log_info(f"🔄 Restarting STT stream...", session_id=self.session.session_id)
+                log_info(f"🔄 Restarting STT stream (session #{self.stt_session_count} -> #{self.stt_session_count + 1})", 
+                        session_id=self.session.session_id)
                 
-                # Önce audio buffer'ı temizle - eski chunk'ları sil
-                await self.audio_buffer.clear()
-                self.chunk_counter = 0
-                
-                # Önce mevcut stream'i temizle (eğer varsa)
-                if self.stt_manager:
-                    await self.stop_stt_streaming()
-                
-                # Biraz bekle - Google API'nin toparlanması için
-                await asyncio.sleep(0.5)
-                
-                # Sonra yeniden başlat
+                # Yeni session başlat (initialize_stt zaten stop_stt_streaming'i çağırıyor)
                 stt_initialized = await self.initialize_stt()
                 if stt_initialized:
                     log_info(f"✅ STT stream restarted successfully", session_id=self.session.session_id)
@@ -343,7 +332,8 @@ class RealtimeSession:
                     return False
             return True
         except Exception as e:
-            log_error(f"❌ Error restarting STT", error=str(e), traceback=traceback.format_exc(), session_id=self.session.session_id)
+            log_error(f"❌ Error restarting STT", error=str(e), traceback=traceback.format_exc(), 
+                     session_id=self.session.session_id)
             return False
     
     async def change_state(self, new_state: ConversationState):
@@ -364,13 +354,21 @@ class RealtimeSession:
     
     async def reset_for_new_utterance(self):
         """Reset for new user utterance"""
+        log_info(f"🔄 Resetting for new utterance", session_id=self.session.session_id)
+        
+        # Buffer ve detector'ı temizle
         await self.audio_buffer.clear()
         self.silence_detector.reset()
+        
+        # Transcription ve counter'ı sıfırla
         self.current_transcription = ""
-        self.chunk_counter = 0  # Chunk counter'ı reset et
+        self.chunk_counter = 0
+        
+        # Speech started flag'ini temizle
         if hasattr(self, 'speech_started'):
             delattr(self, 'speech_started')
-        log_info(f"🔄 Reset for new utterance complete", session_id=self.session.session_id)
+            
+        log_info(f"✅ Reset for new utterance complete", session_id=self.session.session_id)
     
     async def cleanup(self):
         """Clean up resources"""