Update ui/tabs.py

ui/tabs.py

@@ -266,21 +266,22 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
                     autoplay=True
                 )
         
-        # Biến để lưu kết quả từ VAD callback
-        vad_result = gr.State(value={
-            'transcription': "",
-            'response': "", 
-            'tts_audio': None,
-            'status': 'waiting'
-        })
+        # Biến state để lưu kết quả từ VAD
+        vad_transcription = gr.State(value="")
+        vad_response = gr.State(value="")
+        vad_audio = gr.State(value=None)
+        vad_status = gr.State(value="waiting")
         
         def start_vad():
             """Bắt đầu VAD listening"""
             def vad_callback(result):
                 """Callback khi VAD phát hiện speech"""
-                # Lưu kết quả vào state
-                vad_result.value = result
-                print(f"VAD Result: {result}")
+                # Cập nhật state với kết quả mới
+                vad_transcription.value = result.get('transcription', '')
+                vad_response.value = result.get('response', '')
+                vad_audio.value = result.get('tts_audio', None)
+                vad_status.value = result.get('status', 'completed')
+                print(f"VAD Callback: {result.get('transcription', '')}")
             
             success = streaming_service.start_listening(vad_callback)
             status = "✅ Đang lắng nghe với VAD..." if success else "❌ Lỗi khởi động VAD"
@@ -299,7 +300,6 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
                 return "❌ Không có âm thanh", "Vui lòng nói lại", None, "Đang chờ...", {}
             
             try:
-                # SỬA LỖI: Gọi đúng phương thức
                 result = streaming_service.process_streaming_audio(audio_data)
                 state = streaming_service.get_conversation_state()
                 status = "✅ Đã xử lý manual audio"
@@ -312,15 +312,30 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
             """Xóa hội thoại"""
             streaming_service.clear_conversation()
             state = streaming_service.get_conversation_state()
+            # Reset các state
+            vad_transcription.value = ""
+            vad_response.value = ""
+            vad_audio.value = None
             return "", "", None, "🗑️ Đã xóa hội thoại", state
         
-        def update_from_vad():
-            """Cập nhật UI từ kết quả VAD"""
-            result = vad_result.value
+        def get_vad_results():
+            """Lấy kết quả từ VAD và cập nhật UI"""
+            if vad_status.value == "completed":
+                # Reset status sau khi lấy kết quả
+                vad_status.value = "waiting"
+                state = streaming_service.get_conversation_state()
+                return (vad_transcription.value, vad_response.value, vad_audio.value, 
+                       "✅ VAD đã xử lý", state)
+            else:
+                # Không có kết quả mới, trả về None để không cập nhật
+                return gr.skip(), gr.skip(), gr.skip(), gr.skip(), gr.skip()
+        
+        def refresh_state():
+            """Chỉ cập nhật state display"""
             state = streaming_service.get_conversation_state()
-            return result['transcription'], result['response'], result['tts_audio'], "✅ VAD đã xử lý", state
+            return state
         
-        # Event handlers
+        # Event handlers chính
         start_vad_btn.click(
             start_vad,
             outputs=[status_display, state_display]
@@ -331,7 +346,6 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
             outputs=[status_display, state_display]
         )
         
-        # SỬA LỖI: Sử dụng đúng phương thức stream
         microphone.stream(
             process_manual_audio,
             inputs=[microphone],
@@ -343,13 +357,20 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
             outputs=[realtime_transcription, ai_response, tts_output, status_display, state_display]
         )
         
-        # Timer để cập nhật kết quả từ VAD
-        streaming_tab.poll(
-            update_from_vad,
-            1000,
+        # Nút refresh để cập nhật kết quả VAD thủ công
+        refresh_btn = gr.Button("🔄 Kiểm tra kết quả VAD", variant="secondary", visible=False)
+        
+        refresh_btn.click(
+            get_vad_results,
             outputs=[realtime_transcription, ai_response, tts_output, status_display, state_display]
         )
-
+        
+        # Tự động cập nhật state display mỗi 2 giây
+        streaming_tab.load(
+            refresh_state,
+            outputs=[state_display],
+            every=2
+        )
     
     return streaming_tab
 def create_image_tab(image_service: ImageService):