Update ui/tabs.py

ui/tabs.py

@@ -13,6 +13,8 @@ from services.stream_object_detection_service import  StreamObjectDetection
 from services.voice_coding_service import  VoiceCodingService
 from services.sambanova_voice_service import  SambanovaVoiceService
 from services.gemini_realtime_service import GeminiRealtimeService
+from core.conversational_agent import ConversationalAgent
+
 from core.rag_system import EnhancedRAGSystem
 from core.tts_service import EnhancedTTSService
 from core.wikipedia_processor import WikipediaProcessor
@@ -26,7 +28,7 @@ def create_all_tabs(audio_service: AudioService, chat_service: ChatService,
                    voice_coding_service: VoiceCodingService,
                    sambanova_voice_service: SambanovaVoiceService
                    ):
-    
+    multilingual_manager = rag_system.multilingual_manager if hasattr(rag_system, 'multilingual_manager') else None
     with gr.Tab("🎙️ Streaming Voice "):
         create_streaming_voice_tab(streaming_voice_service)
     with gr.Tab("OpenAI Realtime"):
@@ -48,7 +50,11 @@ def create_all_tabs(audio_service: AudioService, chat_service: ChatService,
     
     with gr.Tab("📚 RAG Wikipedia"):
         create_rag_tab(rag_system, wikipedia_processor)
-    
+    with gr.Tab("🌐 Language Info"):
+        if multilingual_manager:
+            create_language_info_tab(multilingual_manager)
+        else:
+            gr.Markdown("⚠️ Không có multilingual manager")
     with gr.Tab("🔊 Text-to-Speech"):
         create_tts_tab(tts_service)
     
@@ -276,6 +282,203 @@ def create_gemini_realtime_tab():
     return gemini_tab
 
 # FastAPI routes cho Gemini (tùy chọn)
+def create_cag_tab(rag_system: EnhancedRAGSystem, multilingual_manager: MultilingualManager):
+    """Tạo tab CAG (Conversational Agent Grounding)"""
+    
+    # Khởi tạo conversational agent
+    conversational_agent = ConversationalAgent(rag_system, multilingual_manager)
+    
+    with gr.Blocks() as cag_tab:
+        gr.Markdown("## 🤖 Chatbot với Tri thức (CAG)")
+        gr.Markdown("Chatbot thông minh sử dụng thông tin từ cơ sở tri thức RAG để trả lời chính xác")
+        
+        with gr.Row():
+            with gr.Column(scale=2):
+                # Chat interface
+                chatbot = gr.Chatbot(
+                    label="🤖 CAG Chatbot",
+                    height=400,
+                    show_copy_button=True,
+                    bubble_full_width=False,
+                    avatar_images=(
+                        "https://cdn-icons-png.flaticon.com/512/1077/1077114.png",  # User
+                        "https://cdn-icons-png.flaticon.com/512/4712/4712027.png"   # Assistant
+                    )
+                )
+                
+                with gr.Row():
+                    msg = gr.Textbox(
+                        label="Nhập câu hỏi",
+                        placeholder="Hỏi tôi về bất kỳ điều gì trong cơ sở tri thức...",
+                        scale=4,
+                        container=False
+                    )
+                    submit_btn = gr.Button("Gửi", variant="primary", scale=1)
+                    clear_btn = gr.Button("Xóa", variant="secondary", scale=1)
+                
+                # Additional info
+                with gr.Row():
+                    lang_display = gr.Textbox(
+                        label="🌐 Ngôn ngữ phát hiện",
+                        interactive=False,
+                        scale=2
+                    )
+                    conv_stats_btn = gr.Button("📊 Thống kê", variant="secondary", scale=1)
+                
+                # Knowledge base stats
+                with gr.Accordion("📊 Cơ sở tri thức", open=False):
+                    kb_stats = gr.Textbox(
+                        label="Thống kê",
+                        interactive=False,
+                        lines=4
+                    )
+                    refresh_kb_btn = gr.Button("🔄 Cập nhật", variant="secondary", size="sm")
+            
+            with gr.Column(scale=1):
+                gr.Markdown("### 📚 Nguồn thông tin")
+                
+                # Knowledge sources from RAG
+                sources_display = gr.JSON(
+                    label="Kết quả tìm kiếm từ RAG",
+                    show_label=True,
+                    container=True
+                )
+                
+                # Conversation statistics
+                conv_stats_display = gr.JSON(
+                    label="📈 Thống kê hội thoại",
+                    show_label=True,
+                    container=True
+                )
+                
+                # Language info
+                with gr.Accordion("🌐 Thông tin ngôn ngữ", open=False):
+                    lang_info_btn = gr.Button("ℹ️ Xem ngôn ngữ", variant="secondary", size="sm")
+                    lang_info_display = gr.JSON(
+                        label="Ngôn ngữ hỗ trợ",
+                        container=True
+                    )
+                
+                # Instructions
+                with gr.Accordion("ℹ️ Hướng dẫn", open=False):
+                    gr.Markdown("""
+                    **Cách sử dụng CAG:**
+                    1. Nhập câu hỏi vào ô chat
+                    2. Bot tự động tìm trong cơ sở tri thức
+                    3. Trả lời dựa trên thông tin tìm thấy
+                    4. Xem nguồn thông tin bên phải
+                    
+                    **Đặc điểm:**
+                    - 🤖 Grounding từ RAG
+                    - 🔍 Tự động tìm kiếm
+                    - 🌐 Đa ngôn ngữ
+                    - 📊 Theo dõi nguồn
+                    """)
+        
+        def respond(message: str, chat_history: List[Tuple[str, str]]) -> Tuple[str, List, List, str]:
+            """Xử lý phản hồi chatbot"""
+            if not message.strip():
+                return "", chat_history, [], "⏳ Chờ câu hỏi..."
+            
+            try:
+                # Gọi conversational agent
+                response, search_results = conversational_agent.process_query(message, chat_history)
+                
+                # Cập nhật chat history
+                chat_history.append((message, response))
+                
+                # Phát hiện ngôn ngữ
+                language = multilingual_manager.detect_language(message)
+                lang_info = multilingual_manager.get_language_info(language)
+                lang_name = lang_info.get('name', language)
+                
+                return "", chat_history, search_results, f"🌐 {lang_name}"
+                
+            except Exception as e:
+                error_response = f"Xin lỗi, đã xảy ra lỗi: {str(e)}"
+                chat_history.append((message, error_response))
+                return "", chat_history, [{"error": str(e)}], "❌ Lỗi"
+        
+        def get_conversation_stats():
+            """Lấy thống kê hội thoại"""
+            return conversational_agent.get_conversation_stats()
+        
+        def get_kb_stats():
+            """Lấy thống kê knowledge base"""
+            stats = rag_system.get_collection_stats()
+            lang_dist = stats['language_distribution']
+            lang_dist_str = ", ".join([f"{k}: {v}" for k, v in lang_dist.items()])
+            
+            return f"""
+📊 THỐNG KÊ CƠ SỞ TRI THỨC:
+• 📄 Tổng documents: {stats['total_documents']}
+• 🔤 Embeddings: {stats['embedding_count']}
+• 🌐 Ngôn ngữ: {lang_dist_str}
+• 📐 Dimension: {stats['embedding_dimension']}
+• 🏷️ Trạng thái: {stats['status']}
+            """
+        
+        def get_language_info():
+            """Lấy thông tin ngôn ngữ hỗ trợ"""
+            return multilingual_manager.get_supported_languages()
+        
+        def clear_chat():
+            """Xóa chat"""
+            conversational_agent.clear_conversation_history()
+            return [], [], "🧹 Đã xóa"
+        
+        def refresh_all():
+            """Làm mới tất cả"""
+            kb_stats_text = get_kb_stats()
+            conv_stats = get_conversation_stats()
+            lang_info = get_language_info()
+            return kb_stats_text, conv_stats, lang_info
+        
+        # Event handlers
+        submit_btn.click(
+            respond, 
+            inputs=[msg, chatbot], 
+            outputs=[msg, chatbot, sources_display, lang_display]
+        )
+        
+        msg.submit(
+            respond,
+            inputs=[msg, chatbot],
+            outputs=[msg, chatbot, sources_display, lang_display]
+        )
+        
+        clear_btn.click(
+            clear_chat,
+            inputs=[],
+            outputs=[chatbot, sources_display, lang_display]
+        )
+        
+        conv_stats_btn.click(
+            get_conversation_stats,
+            inputs=[],
+            outputs=[conv_stats_display]
+        )
+        
+        refresh_kb_btn.click(
+            get_kb_stats,
+            inputs=[],
+            outputs=[kb_stats]
+        )
+        
+        lang_info_btn.click(
+            get_language_info,
+            inputs=[],
+            outputs=[lang_info_display]
+        )
+        
+        # Initialize on load
+        cag_tab.load(
+            refresh_all,
+            inputs=[],
+            outputs=[kb_stats, conv_stats_display, lang_info_display]
+        )
+    
+    return cag_tab
 def setup_gemini_routes(app):
     """Thiết lập routes FastAPI cho Gemini"""