Update ui/tabs.py

ui/tabs.py

@@ -51,7 +51,7 @@ def create_rag_tab(rag_system: EnhancedRAGSystem, wikipedia_processor: Wikipedia
             
             gr.Markdown("### 📊 Thống kê Database")
             stats_btn = gr.Button("📊 Database Stats", variant="secondary")
-            stats_display = gr.Textbox(label="Thống kê", interactive=False)
+            stats_display = gr.Textbox(label="Thống kê", interactive=False, lines=5)
             
             gr.Markdown("### 🔍 Tìm kiếm Database")
             search_query = gr.Textbox(
@@ -78,14 +78,20 @@ def create_rag_tab(rag_system: EnhancedRAGSystem, wikipedia_processor: Wikipedia
             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['count']}"
+            return f"✅ Đã thêm {len(documents)} documents Wikipedia vào RAG database. Tổng số documents: {stats['total_documents']}"
             
         except Exception as e:
             return f"❌ Lỗi xử lý file Wikipedia: {str(e)}"
     
+    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)
+    
     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(rag_system.semantic_search, inputs=[search_query], outputs=[rag_results])
+    search_btn.click(search_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()