Update app.py

app.py

@@ -12,13 +12,12 @@ os.environ["XDG_CACHE_HOME"] = "/tmp/.cache"
 os.environ["HOME"] = "/tmp"
 for p in ["/tmp/hf_home","/tmp/hf_cache","/tmp/hf_datasets","/tmp/.cache"]:
     os.makedirs(p, exist_ok=True)
-# Xóa cache cũ nếu có
 shutil.rmtree("/.cache", ignore_errors=True)
 
 # =================== #
 #   Import thư viện   #
 # =================== #
-import time, hashlib, gzip, pickle, json, traceback, re
+import gzip, pickle, json, re
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 import numpy as np
@@ -75,16 +74,12 @@ except Exception as e:
 # ================ #
 faiss_index = faiss.read_index(FAISS_PATH)
 with gzip.open(BM25_PATH,"rb") as f: bm25 = pickle.load(f)
-with gzip.open(METAS_PATH,"rb") as f:
-    metas = pickle.load(f)
+with gzip.open(METAS_PATH,"rb") as f: metas = pickle.load(f)
 if isinstance(metas,dict) and "corpus" in metas:
     corpus = metas["corpus"]
 else:
     corpus = metas
 
-# Lưu list key để tránh tạo lại nhiều lần
-meta_keys = list(range(len(corpus)))
-
 # Load embedding model
 device = os.environ.get("DEVICE","cpu")
 embedding_model = SentenceTransformer(EMB_MODEL, device=device)
@@ -107,8 +102,6 @@ if API_KEY:
     except Exception as e:
         print("Warning: cannot init GenAI:", e)
 
-answer_cache = TTLCache(maxsize=CACHE_MAX, ttl=CACHE_TTL)
-
 # =================== #
 # Utility / Retrieve  #
 # =================== #
@@ -117,7 +110,6 @@ def minmax_scale(arr):
     return np.zeros_like(arr) if len(arr)==0 or np.max(arr)==np.min(arr) else (arr-np.min(arr))/(np.max(arr)-np.min(arr))
 
 def classify_followup(text:str)->int:
-    # như code gốc, bỏ bớt regex nặng để nhanh hơn
     t=text.lower().strip()
     if len(t.split())<=4: return 0
     if re.search(r"\b(nó|cái này|thế thì|vậy thì)\b",t): return 0
@@ -152,12 +144,13 @@ def get_full_procedure_text_by_parent(pid):
     field_map={"ten_thu_tuc":"Tên thủ tục","cach_thuc_thuc_hien":"Cách thức thực hiện","thanh_phan_ho_so":"Thành phần hồ sơ","trinh_tu_thuc_hien":"Trình tự thực hiện","co_quan_thuc_hien":"Cơ quan thực hiện","yeu_cau_dieu_kien":"Yêu cầu, điều kiện","nguon":"Nguồn"}
     return "\n\n".join([f"{field_map[k]}:\n{v}" for k,v in rec.items() if k in field_map and v])
 
-# ================ #
-# Flask endpoints  #
-# ================ #
+# =================== #
+# Flask endpoints     #
+# =================== #
 app=Flask(__name__)
 CORS(app)
 chat_histories={}
+answer_cache = TTLCache(maxsize=CACHE_MAX, ttl=CACHE_TTL)
 
 @app.route("/health")
 def health(): return {"status":"ok"}