Update ui/tabs.py

ui/tabs.py

@@ -37,154 +37,150 @@ def create_all_tabs(audio_service: AudioService, chat_service: ChatService,
         create_language_info_tab(rag_system.multilingual_manager)
 def create_rag_tab(rag_system: EnhancedRAGSystem, wikipedia_processor: WikipediaProcessor):
     """Tạo tab RAG với debug chi tiết"""
-    gr.Markdown("## 📚 Upload Dữ Liệu Wikipedia")
     
-    with gr.Row():
-        with gr.Column(scale=1):
-            gr.Markdown("### 📤 Upload Dữ Liệu")
-            file_upload = gr.File(
-                label="Tải lên file (TXT, CSV, JSON)",
-                file_types=['.txt', '.csv', '.json'],
-                file_count="single",
-                type="filepath"
-            )
-            upload_btn = gr.Button("📤 Upload Data", variant="primary")
-            upload_status = gr.Textbox(
-                label="Trạng thái Upload",
-                interactive=False,
-                lines=5
-            )
-            
-            gr.Markdown("### 📊 Thống kê Database")
-            stats_btn = gr.Button("📊 Database Stats", variant="secondary")
-            stats_display = gr.Textbox(
-                label="Thống kê", 
-                interactive=False,
-                lines=6
-            )
-            
-        with gr.Column(scale=2):
-            gr.Markdown("### 🔍 Tìm kiếm & Kiểm tra")
-            search_query = gr.Textbox(
-                label="Tìm kiếm trong database",
-                placeholder="Nhập từ khóa để kiểm tra dữ liệu..."
-            )
-            search_btn = gr.Button("🔍 Tìm kiếm", variant="secondary")
-            rag_results = gr.JSON(
-                label="Kết quả tìm kiếm",
-                show_label=True
-            )
+    # Initialize systems if not provided
+    if rag_system is None:
+        rag_system = EnhancedRAGSystem()
+    if wikipedia_processor is None:
+        wikipedia_processor = WikipediaProcessor()
     
-    def upload_wikipedia_file(file):
-        """Xử lý upload file với debug đầy đủ"""
-        if file is None:
-            return "❌ Vui lòng chọn file để upload"
+    with gr.Blocks() as rag_tab:
+        gr.Markdown("## 📚 Upload Dữ Liệu Wikipedia")
         
-        try:
-            print(f"🔄 Bắt đầu upload file: {file.name}")
-            
-            # Kiểm tra file tồn tại
-            if not os.path.exists(file.name):
-                return f"❌ File không tồn tại: {file.name}"
-            
-            # Xử lý file
-            documents = wikipedia_processor.process_uploaded_file(file.name)
-            
-            if not documents:
-                return "❌ Không thể trích xuất dữ liệu từ file. File có thể trống hoặc định dạng không đúng."
-            
-            print(f"✅ Đã xử lý {len(documents)} documents")
-            
-            # Tạo metadata
-            metadatas = []
-            for i, doc in enumerate(documents):
-                metadata = {
-                    "source": "uploaded_file",
-                    "type": "knowledge", 
-                    "file_name": os.path.basename(file.name),
-                    "language": "vi",
-                    "doc_id": i,
-                    "length": len(doc)
-                }
-                metadatas.append(metadata)
-            
-            # Thêm vào RAG system
-            old_stats = rag_system.get_collection_stats()
-            old_count = old_stats['total_documents']
-            
-            rag_system.add_documents(documents, metadatas)
-            
-            # Lấy thống kê mới
-            new_stats = rag_system.get_collection_stats()
-            new_count = new_stats['total_documents']
-            
-            success_msg = f"""
-✅ UPLOAD THÀNH CÔNG!
-
-📁 File: {os.path.basename(file.name)}
-📄 Documents xử lý: {len(documents)}
-📊 Documents thêm vào: {new_count - old_count}
-🏷️ Tổng documents: {new_count}
-🔤 Embeddings: {new_stats['embedding_count']}
-🌐 Ngôn ngữ: {new_stats['language_distribution']}
-
-💡 Bạn có thể tìm kiếm ngay để kiểm tra dữ liệu!
-"""
-            return success_msg
-            
-        except Exception as e:
-            error_msg = f"❌ LỖI UPLOAD: {str(e)}"
-            print(f"UPLOAD ERROR: {traceback.format_exc()}")
-            return error_msg
-    
-    def get_rag_stats():
-        """Lấy thống kê chi tiết"""
-        try:
-            stats = rag_system.get_collection_stats()
-            return f"""
-📊 THỐNG KÊ RAG DATABASE:
-
-• 📄 Tổng documents: {stats['total_documents']}
-• 🔤 Số embeddings: {stats['embedding_count']}
-• 📐 Dimension: {stats['embedding_dimension']}
-• 🌐 Phân bố ngôn ngữ: {stats['language_distribution']}
-• ✅ Trạng thái: {stats['status']}
-• 🏷️ Tên: {stats['name']}
-
-💡 Embeddings: {'Có' if stats['has_embeddings'] else 'Không'}
-"""
-        except Exception as e:
-            return f"❌ Lỗi lấy thống kê: {str(e)}"
-    
-    def search_rag_database(query):
-        """Tìm kiếm để kiểm tra dữ liệu"""
-        if not query.strip():
-            return [{"message": "Nhập từ khóa để tìm kiếm"}]
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown("### 📤 Upload Dữ Liệu")
+                file_upload = gr.File(
+                    label="Tải lên file (TXT, CSV, JSON)",
+                    file_types=['.txt', '.csv', '.json'],
+                    file_count="single"
+                )
+                upload_btn = gr.Button("📤 Upload Data", variant="primary")
+                upload_status = gr.Textbox(
+                    label="Trạng thái Upload",
+                    interactive=False,
+                    lines=5
+                )
+                
+                gr.Markdown("### 📊 Thống kê Database")
+                stats_btn = gr.Button("📊 Database Stats", variant="secondary")
+                stats_display = gr.Textbox(
+                    label="Thống kê", 
+                    interactive=False,
+                    lines=6
+                )
+                
+            with gr.Column(scale=2):
+                gr.Markdown("### 🔍 Tìm kiếm & Kiểm tra")
+                search_query = gr.Textbox(
+                    label="Tìm kiếm trong database",
+                    placeholder="Nhập từ khóa để kiểm tra dữ liệu..."
+                )
+                search_btn = gr.Button("🔍 Tìm kiếm", variant="secondary")
+                rag_results = gr.JSON(
+                    label="Kết quả tìm kiếm",
+                    show_label=True
+                )
         
-        try:
-            results = rag_system.semantic_search(query, top_k=3)
-            
-            if not results:
-                return [{"message": "Không tìm thấy kết quả nào", "query": query}]
-            
-            formatted_results = []
-            for i, result in enumerate(results):
-                formatted_results.append({
-                    'id': result.id,
-                    'text': result.text[:150] + "..." if len(result.text) > 150 else result.text,
-                    'similarity': round(result.similarity, 3),
-                    'metadata': result.metadata
-                })
+        def upload_wikipedia_file(file):
+            """Xử lý upload file với debug đầy đủ"""
+            if file is None:
+                return "❌ Vui lòng chọn file để upload"
             
-            return formatted_results
+            try:
+                print(f"🔄 Bắt đầu upload file: {file.name}")
+                
+                # Kiểm tra file tồn tại
+                if not os.path.exists(file.name):
+                    return f"❌ File không tồn tại: {file.name}"
+                
+                # Xử lý file
+                documents = wikipedia_processor.process_uploaded_file(file.name)
+                
+                if not documents:
+                    return "❌ Không thể trích xuất dữ liệu từ file. File có thể trống hoặc định dạng không đúng."
+                
+                print(f"✅ Đã xử lý {len(documents)} documents")
+                
+                # Tạo metadata
+                metadatas = []
+                for i, doc in enumerate(documents):
+                    metadata = {
+                        "source": "uploaded_file",
+                        "type": "knowledge", 
+                        "file_name": os.path.basename(file.name),
+                        "language": "vi",
+                        "doc_id": i,
+                        "length": len(doc)
+                    }
+                    metadatas.append(metadata)
+                
+                # Thêm vào RAG system
+                old_stats = rag_system.get_collection_stats()
+                old_count = old_stats['total_documents']
+                
+                rag_system.add_documents(documents, metadatas)
+                
+                # Lấy thống kê mới
+                new_stats = rag_system.get_collection_stats()
+                new_count = new_stats['total_documents']
+                
+                success_msg = f"""
+    ✅ UPLOAD THÀNH CÔNG!
+    📁 File: {os.path.basename(file.name)}
+    📄 Documents xử lý: {len(documents)}
+    📊 Documents thêm vào: {new_count - old_count}
+    🏷️ Tổng documents: {new_count}
+    🔤 Embeddings: {new_stats['embedding_count']}
+    🌐 Ngôn ngữ: {new_stats['language_distribution']}
+    💡 Bạn có thể tìm kiếm ngay để kiểm tra dữ liệu!
+    """
+                return success_msg
+                
+            except Exception as e:
+                error_msg = f"❌ LỖI UPLOAD: {str(e)}"
+                print(f"UPLOAD ERROR: {traceback.format_exc()}")
+                return error_msg
+        
+        def get_rag_stats():
+            """Lấy thống kê chi tiết"""
+            try:
+                stats = rag_system.get_collection_stats()
+                return f"""
+    📊 THỐNG KÊ RAG DATABASE:
+    • 📄 Tổng documents: {stats['total_documents']}
+    • 🔤 Số embeddings: {stats['embedding_count']}
+    • 📐 Dimension: {stats['embedding_dimension']}
+    • 🌐 Phân bố ngôn ngữ: {stats['language_distribution']}
+    • ✅ Trạng thái: {stats['status']}
+    • 🏷️ Tên: {stats['name']}
+    💡 Embeddings: {'Có' if stats['has_embeddings'] else 'Không'}
+    """
+            except Exception as e:
+                return f"❌ Lỗi lấy thống kê: {str(e)}"
+        
+        def search_rag_database(query):
+            """Tìm kiếm để kiểm tra dữ liệu"""
+            if not query.strip():
+                return [{"message": "Nhập từ khóa để tìm kiếm"}]
             
-        except Exception as e:
-            return [{"error": f"Lỗi tìm kiếm: {str(e)}"}]
+            try:
+                results = rag_system.semantic_search(query, top_k=3)
+                
+                if not results:
+                    return [{"message": "Không tìm thấy kết quả nào", "query": query}]
+                
+                return results
+                
+            except Exception as e:
+                return [{"error": f"Lỗi tìm kiếm: {str(e)}"}]
+        
+        # Event handlers
+        upload_btn.click(upload_wikipedia_file, inputs=[file_upload], outputs=[upload_status])
+        stats_btn.click(get_rag_stats, inputs=[], outputs=[stats_display])
+        search_btn.click(search_rag_database, inputs=[search_query], outputs=[rag_results])
     
-    # Event handlers
-    upload_btn.click(upload_wikipedia_file, inputs=[file_upload], outputs=[upload_status])
-    stats_btn.click(get_rag_stats, inputs=[], outputs=[stats_display])
-    search_btn.click(search_rag_database, inputs=[search_query], outputs=[rag_results])
+    return rag_tab
 def create_audio_tab(audio_service: AudioService):
     gr.Markdown("## Nói chuyện với AI (Đa ngôn ngữ)")
     audio_input, transcription_output, response_output, tts_audio_output, process_button = create_audio_components()