Update ui/tabs.py

ui/tabs.py

@@ -200,11 +200,14 @@ def create_audio_tab(audio_service: AudioService):
         outputs=[transcription_output, response_output, tts_audio_output, language_display]  # UPDATED
     )
 def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
-    """Tạo tab streaming voice với VAD optimized"""
+    """Tạo tab streaming voice với VAD optimized - FIXED VERSION"""
     
     with gr.Blocks() as streaming_tab:
         gr.Markdown("## 🎤 Trò chuyện giọng nói thời gian thực - Tối ưu hóa")
         
+        # FIXED: Thêm state để lưu kết quả VAD
+        vad_result_state = gr.State(value=None)
+        
         with gr.Row():
             with gr.Column(scale=1):
                 # Controls
@@ -268,14 +271,19 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
         # State variables
         is_vad_active = gr.State(value=False)
         
+        def vad_callback(result):
+            """Callback khi VAD phát hiện speech - FIXED VERSION"""
+            print(f"🎯 Nhận kết quả từ VAD: {result['transcription']}")
+            vad_result_state.value = result
+        
         def start_vad():
-            """Bắt đầu VAD"""
-            success = streaming_service.start_listening(lambda result: None)
+            """Bắt đầu VAD - FIXED VERSION"""
+            success = streaming_service.start_listening(vad_callback)
             if success:
                 is_vad_active.value = True
                 status = "✅ VAD đang chạy - Hãy nói gì đó!"
-                state_info = streaming_service.get_conversation_state()
-                state_text = f"VAD: Đang hoạt động\nQueue: {state_info['queue_size']}\nThreads: {state_info['worker_threads']}"
+                state = streaming_service.get_conversation_state()
+                state_text = f"VAD: Đang hoạt động\nQueue: {state['queue_size']}\nThreads: {state['worker_threads']}"
             else:
                 status = "❌ Không thể khởi động VAD"
                 state_text = "Lỗi khởi động"
@@ -286,8 +294,8 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
             """Dừng VAD"""
             streaming_service.stop_listening()
             is_vad_active.value = False
-            state_info = streaming_service.get_conversation_state()
-            state_text = f"VAD: Đã dừng\nHistory: {state_info['history_length']} messages"
+            state = streaming_service.get_conversation_state()
+            state_text = f"VAD: Đã dừng\nHistory: {state['history_length']} messages"
             return "🛑 VAD đã dừng", state_text
         
         def process_microphone(audio_data):
@@ -297,17 +305,29 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
             
             try:
                 result = streaming_service.process_streaming_audio(audio_data)
-                state_info = streaming_service.get_conversation_state()
-                state_text = f"Manual mode\nHistory: {state_info['history_length']} messages"
+                state = streaming_service.get_conversation_state()
+                state_text = f"Manual mode\nHistory: {state['history_length']} messages"
                 return result['transcription'], result['response'], result['tts_audio'], state_text
             except Exception as e:
                 return f"Lỗi: {e}", "Xin lỗi, có lỗi xảy ra", None, "Lỗi xử lý"
         
+        def check_vad_results():
+            """Kiểm tra kết quả VAD - FIXED: Cập nhật UI khi có kết quả mới"""
+            if vad_result_state.value is not None:
+                result = vad_result_state.value
+                vad_result_state.value = None  # Reset sau khi dùng
+                
+                state = streaming_service.get_conversation_state()
+                state_text = f"VAD mode\nHistory: {state['history_length']} messages\nQueue: {state['queue_size']}"
+                
+                return result['transcription'], result['response'], result['tts_audio'], state_text
+            return gr.skip(), gr.skip(), gr.skip(), gr.skip()
+        
         def clear_chat():
             """Xóa hội thoại"""
             streaming_service.clear_conversation()
-            state_info = streaming_service.get_conversation_state()
-            state_text = f"Đã xóa hội thoại\nHistory: {state_info['history_length']} messages"
+            state = streaming_service.get_conversation_state()
+            state_text = f"Đã xóa hội thoại\nHistory: {state['history_length']} messages"
             return "", "", None, state_text
         
         def refresh_latency():
@@ -349,7 +369,13 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
             outputs=[latency_display]
         )
         
-        # Tự động cập nhật mỗi 3 giây
+        # FIXED: Timer để kiểm tra kết quả VAD
+        gr.Timer(1.0).tick(
+            fn=check_vad_results,
+            outputs=[transcription_box, response_box, audio_output, state_info]
+        )
+        
+        # Timer để cập nhật system info
         gr.Timer(3.0).tick(
             fn=update_state_info,
             outputs=[state_info, latency_display]