Update ui/tabs.py

ui/tabs.py

@@ -199,11 +199,11 @@ def create_audio_tab(audio_service: AudioService):
         inputs=audio_input,
         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 - Phiên bản cực kỳ đơn giản"""
+def create_streaming_voice_tab(streaming_service: OptimizedStreamingVoiceService):
+    """Tạo tab streaming voice với VAD optimized"""
     
     with gr.Blocks() as streaming_tab:
-        gr.Markdown("## 🎤 Trò chuyện giọng nói thời gian thực")
+        gr.Markdown("## 🎤 Trò chuyện giọng nói thời gian thực - Tối ưu hóa")
         
         with gr.Row():
             with gr.Column(scale=1):
@@ -228,6 +228,7 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
                     type="numpy",
                     streaming=True
                 )
+                
                 with gr.Accordion("📊 Performance Metrics", open=False):
                     latency_display = gr.JSON(
                         label="Latency Statistics",
@@ -264,36 +265,30 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
                     autoplay=True
                 )
         
-        # Biến đơn giản để track VAD
+        # State variables
         is_vad_active = gr.State(value=False)
-        last_vad_result = gr.State(value=None)
         
         def start_vad():
             """Bắt đầu VAD"""
-            def callback(result):
-                last_vad_result.value = result
-                print(f"VAD: {result.get('transcription', '')}")
-            
-            success = streaming_service.start_listening(callback)
+            success = streaming_service.start_listening(lambda result: None)
             if success:
                 is_vad_active.value = True
                 status = "✅ VAD đang chạy - Hãy nói gì đó!"
-                state = f"VAD: Đang hoạt động\n"
-                state += f"Processing: {streaming_service.is_processing}\n"
-                state += f"History: {len(streaming_service.conversation_history)} messages"
+                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']}"
             else:
                 status = "❌ Không thể khởi động VAD"
-                state = "Lỗi khởi động"
+                state_text = "Lỗi khởi động"
             
-            return status, state
+            return status, state_text
         
         def stop_vad():
             """Dừng VAD"""
             streaming_service.stop_listening()
             is_vad_active.value = False
-            state = f"VAD: Đã dừng\n"
-            state += f"History: {len(streaming_service.conversation_history)} messages"
-            return "🛑 VAD đã dừng", state
+            state_info = streaming_service.get_conversation_state()
+            state_text = f"VAD: Đã dừng\nHistory: {state_info['history_length']} messages"
+            return "🛑 VAD đã dừng", state_text
         
         def process_microphone(audio_data):
             """Xử lý microphone input"""
@@ -302,47 +297,39 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
             
             try:
                 result = streaming_service.process_streaming_audio(audio_data)
-                state = f"Manual mode\nHistory: {len(streaming_service.conversation_history)} messages"
-                return result['transcription'], result['response'], result['tts_audio'], state
+                state_info = streaming_service.get_conversation_state()
+                state_text = f"Manual mode\nHistory: {state_info['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 clear_chat():
             """Xóa hội thoại"""
             streaming_service.clear_conversation()
-            state = f"Đã xóa hội thoại\nHistory: 0 messages"
-            return "", "", None, state
+            state_info = streaming_service.get_conversation_state()
+            state_text = f"Đã xóa hội thoại\nHistory: {state_info['history_length']} messages"
+            return "", "", None, state_text
         
-        def check_vad_auto():
-            """Tự động kiểm tra VAD results (nếu có)"""
-            if is_vad_active.value and last_vad_result.value:
-                result = last_vad_result.value
-                last_vad_result.value = None  # Reset sau khi dùng
-                state = f"VAD: Đã xử lý\nHistory: {len(streaming_service.conversation_history)} messages"
-                return result['transcription'], result['response'], result['tts_audio'], state
-            return gr.skip(), gr.skip(), gr.skip(), gr.skip()
         def refresh_latency():
             """Làm mới latency metrics"""
             stats = streaming_service.get_latency_stats()
             return stats
         
         def update_state_info():
-            """Cập nhật thông tin trạng thái với latency"""
+            """Cập nhật thông tin trạng thái"""
             state = streaming_service.get_conversation_state()
             
-            # Format latency info
-            latency_info = state.get('latency_stats', {})
             formatted_state = f"VAD: {'Đang chạy' if state['is_listening'] else 'Dừng'}\n"
-            formatted_state += f"Processing: {state['is_processing']}\n"
-            formatted_state += f"History: {state['history_length']} messages\n"
             formatted_state += f"Queue: {state['queue_size']}\n"
+            formatted_state += f"History: {state['history_length']} messages\n"
+            formatted_state += f"Threads: {state['worker_threads']}\n"
+            formatted_state += f"Last: {state['last_update']}"
             
-            if latency_info.get('total', {}).get('count', 0) > 0:
-                total_avg = latency_info['total']['recent_avg']
-                formatted_state += f"Avg Latency: {total_avg:.2f}s"
+            latency_info = streaming_service.get_latency_stats()
             
             return formatted_state, latency_info
-        # Event handlers đơn giản
+
+        # Event handlers
         start_btn.click(start_vad, outputs=[vad_status, state_info])
         stop_btn.click(stop_vad, outputs=[vad_status, state_info])
         
@@ -357,14 +344,13 @@ def create_streaming_voice_tab(streaming_service: StreamingVoiceService):
             outputs=[transcription_box, response_box, audio_output, state_info]
         )
 
-        # Thêm event cho refresh latency
         refresh_latency_btn.click(
             refresh_latency,
             outputs=[latency_display]
         )
         
-        # Tự động cập nhật latency mỗi 5 giây
-        gr.Timer(5.0).tick(
+        # Tự động cập nhật mỗi 3 giây
+        gr.Timer(3.0).tick(
             fn=update_state_info,
             outputs=[state_info, latency_display]
         )