Update ui/tabs.py

ui/tabs.py

@@ -36,62 +36,155 @@ def create_all_tabs(audio_service: AudioService, chat_service: ChatService,
     with gr.Tab("🌐 Language Info"):  # NEW TAB
         create_language_info_tab(rag_system.multilingual_manager)
 def create_rag_tab(rag_system: EnhancedRAGSystem, wikipedia_processor: WikipediaProcessor):
-    gr.Markdown("## Upload data có sẵn")
+    """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")
+            gr.Markdown("### 📤 Upload Dữ Liệu")
             file_upload = gr.File(
-                label="Tải lên file",
+                label="Tải lên file (TXT, CSV, JSON)",
                 file_types=['.txt', '.csv', '.json'],
-                file_count="single"
+                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)
+            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=5)
+            stats_display = gr.Textbox(
+                label="Thống kê", 
+                interactive=False,
+                lines=6
+            )
             
-            gr.Markdown("### 🔍 Tìm kiếm Database")
+        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 để tìm kiếm..."
+                placeholder="Nhập từ khóa để kiểm tra dữ liệu..."
             )
             search_btn = gr.Button("🔍 Tìm kiếm", variant="secondary")
-        
-        with gr.Column(scale=2):
-            gr.Markdown("### 📋 Kết quả tìm kiếm RAG")
-            rag_results = gr.JSON(label="Tài liệu tham khảo tìm được")
+            rag_results = gr.JSON(
+                label="Kết quả tìm kiếm",
+                show_label=True
+            )
     
     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 "❌ Vui lòng chọn file để upload"
         
         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 tìm thấy dữ liệu nào trong file."
+                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']
             
-            metadatas = [{"source": "wikipedia", "type": "knowledge", "file": file.name, "language": "vi"} for _ in documents]
             rag_system.add_documents(documents, metadatas)
             
-            stats = rag_system.get_collection_stats()
-            return f"✅ Đã thêm {len(documents)} documents Wikipedia vào RAG database. Tổng số documents: {stats['total_documents']}"
+            # 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:
-            return f"❌ Lỗi xử lý file Wikipedia: {str(e)}"
+            error_msg = f"❌ LỖI UPLOAD: {str(e)}"
+            print(f"UPLOAD ERROR: {traceback.format_exc()}")
+            return error_msg
     
-    def search_database(query):
-        """Hàm wrapper để tìm kiếm với top_k mặc định"""
-        if not query or query.strip() == "":
-            return []
-        return rag_system.semantic_search(query, top_k=3)
+    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"}]
+        
+        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
+                })
+            
+            return formatted_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(rag_system.get_collection_stats, inputs=[], outputs=[stats_display])
-    search_btn.click(search_database, inputs=[search_query], outputs=[rag_results])
+    stats_btn.click(get_rag_stats, inputs=[], outputs=[stats_display])
+    search_btn.click(search_rag_database, inputs=[search_query], outputs=[rag_results])
 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()