Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 import json
 import shutil
+import html
 from datetime import datetime
 from retriever import retriever, reload_retriever
 from generator import answer_query
@@ -11,7 +12,6 @@ from langchain_community.document_loaders import (
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_huggingface import HuggingFaceEmbeddings
 from langchain_community.vectorstores import FAISS
-import html
 
 # Đường dẫn file CSS
 CUSTOM_CSS_PATH = "gradio_theme.css"
@@ -54,7 +54,7 @@ def process_document(file):
         elif file_path.endswith(".csv"):
             loader = CSVLoader(file_path)
         elif file_path.endswith(".txt"):
-            loader = TextLoader(file_path, autodetect_encoding=True)   # <== fix lỗi txt
+            loader = TextLoader(file_path, autodetect_encoding=True)
         elif file_path.endswith(".docx") or file_path.endswith(".doc"):
             loader = UnstructuredWordDocumentLoader(file_path)
         else:
@@ -90,8 +90,9 @@ def delete_file(filename):
 def clear_inputs():
     return "", ""
 
-def query_function(question, model_choice, temperature, include_sources):
-    answer, docs = answer_query(question, model=model_choice, temperature=temperature)
+def query_function(question, temperature, include_sources):
+    fixed_model = "sentence-transformers/all-MiniLM-L6-v2"
+    answer, docs = answer_query(question, model=fixed_model, temperature=temperature)
     answer = html.escape(answer)
 
     if include_sources and docs:
@@ -109,38 +110,47 @@ def query_function(question, model_choice, temperature, include_sources):
             answer += sources_text
     return answer
 
-# Giao diện Gradio
+# Tạo giao diện Gradio
 with gr.Blocks(css=CUSTOM_CSS_PATH) as demo:
     with gr.Row():
         with gr.Column(scale=5):
             gr.Markdown("## 🔍 RAGFlow Enterprise Search\nTìm kiếm thông minh từ tài liệu nội bộ", elem_classes="container-box")
 
     with gr.Tabs():
+        # Tab Tìm kiếm
         with gr.TabItem("🔍 Tìm kiếm"):
             with gr.Column(elem_classes="container-box"):
-                question = gr.Textbox(lines=3, label="Câu hỏi")
-                with gr.Row():                    
-                    temperature = gr.Slider(0, 1, value=0.2, step=0.1, label="Temperature")
+                question       = gr.Textbox(lines=3, label="Câu hỏi")
+                with gr.Row():
+                    temperature     = gr.Slider(0, 1, value=0.2, step=0.1, label="Temperature")
                     include_sources = gr.Checkbox(label="Hiển thị nguồn", value=True)
                 with gr.Row():
                     search_btn = gr.Button("🔍 Tìm kiếm", variant="primary", elem_classes="button-primary")
-                    clear_btn = gr.Button("🗑️ Xóa", variant="secondary", elem_classes="button-secondary")
-                output = gr.Markdown(elem_classes="output-box")  # Hiển thị kết quả trong khung đẹp
+                    clear_btn  = gr.Button("🗑️ Xóa", variant="secondary", elem_classes="button-secondary")
+                output = gr.Markdown(elem_classes="output-box")
 
-            search_btn.click(query_function, inputs=[question, model_choice, temperature, include_sources], outputs=[output])
-            clear_btn.click(clear_inputs, outputs=[question, output])
+            search_btn.click(query_function,
+                             inputs=[question, temperature, include_sources],
+                             outputs=[output])
+            clear_btn.click(clear_inputs,
+                            outputs=[question, output])
 
+        # Tab Quản lý tài liệu
         with gr.TabItem("📚 Quản lý tài liệu"):
             with gr.Column(elem_classes="container-box"):
                 upload_file = gr.File(label="Tải lên tài liệu", file_types=[".pdf", ".docx", ".doc", ".csv", ".txt"])
-                upload_btn = gr.Button("📄 Tải lên và xử lý", variant="primary")
+                upload_btn  = gr.Button("📄 Tải lên và xử lý", variant="primary")
                 upload_status = gr.Textbox(label="Trạng thái", lines=3, interactive=False)
             uploaded_files_list = gr.Markdown(value=update_uploaded_files(), elem_classes="scroll-box")
             with gr.Column(elem_classes="container-box"):
                 delete_filename = gr.Textbox(label="Tên file muốn xóa")
-                delete_btn = gr.Button("🗑️ Xóa tài liệu", variant="secondary")
-
-            upload_btn.click(process_document, inputs=[upload_file], outputs=[upload_status, uploaded_files_list])
-            delete_btn.click(delete_file, inputs=[delete_filename], outputs=[uploaded_files_list])
+                delete_btn      = gr.Button("🗑️ Xóa tài liệu", variant="secondary")
+
+            upload_btn.click(process_document,
+                             inputs=[upload_file],
+                             outputs=[upload_status, uploaded_files_list])
+            delete_btn.click(delete_file,
+                             inputs=[delete_filename],
+                             outputs=[uploaded_files_list])
 
     demo.launch(share=True)