Update app.py

app.py

@@ -56,9 +56,11 @@ FORCE_REBUILD_DB = False
 MAX_HISTORY_TURNS = 4
 
 # ---------- CẤU HÌNH DATASET PERSISTENT (MIỄN PHÍ) ----------
-HF_USERNAME = "PBThuong96"          # 🔴 THAY BẰNG TÊN CỦA BẠN
+# 🔴 THAY YOUR-USERNAME BẰNG TÊN THẬT CỦA BẠN (VÍ DỤ: PBThuong96)
+HF_USERNAME = "your-username"
 DATASET_NAME = "deepmed-db"
-DATASET_REPO = f"datasets/{HF_USERNAME}/{DATASET_NAME}"
+# ⚠️ QUAN TRỌNG: Không được thêm "datasets/" vào đầu, chỉ cần "username/dataset-name"
+DATASET_REPO = f"{HF_USERNAME}/{DATASET_NAME}"
 
 # ---------- XÁC THỰC HUGGING FACE ----------
 if HF_TOKEN:
@@ -78,6 +80,7 @@ def download_persistent_data():
             logging.info("📥 Đang tải Chroma DB từ Dataset...")
             snapshot_download(
                 repo_id=DATASET_REPO,
+                repo_type="dataset",  # bắt buộc khi dùng dataset
                 allow_patterns=f"{DB_PATH}/*",
                 local_dir=".",
                 local_dir_use_symlinks=False,
@@ -92,6 +95,7 @@ def download_persistent_data():
             logging.info("📥 Đang tải splits_cache.pkl...")
             snapshot_download(
                 repo_id=DATASET_REPO,
+                repo_type="dataset",
                 allow_patterns=SPLITS_CACHE,
                 local_dir=".",
                 local_dir_use_symlinks=False,
@@ -280,7 +284,7 @@ def get_retrievers():
         model_name="BAAI/bge-reranker-v2-m3",
         model_kwargs={'device': 'cpu', 'low_cpu_mem_usage': True}
     )
-    compressor = CrossEncoderReranker(model=reranker_model, top_n=5)
+    compressor = CrossEncoderReranker(model=reranker_model, top_n=3)
     deep_retriever = ContextualCompressionRetriever(
         base_compressor=compressor,
         base_retriever=ensemble_deep
@@ -303,7 +307,7 @@ class DeepMedBot:
         try:
             self.fast_retriever, self.deep_retriever = get_retrievers()
             self.llm = ChatGoogleGenerativeAI(
-                model="gemini-2.5-flash",   # ổn định và nhanh
+                model="gemini-1.5-flash",   # ổn định và nhanh
                 temperature=0.2,
                 google_api_key=GOOGLE_API_KEY,
                 convert_system_message_to_human=True