GOGO198 commited on
Commit
c9bb56d
·
verified ·
1 Parent(s): 7de46e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -36,26 +36,23 @@ def refresh_index():
36
 
37
  while True:
38
  try:
39
- # 检查是否有更新
40
  current_time = time.time()
41
  if current_time - last_updated > index_refresh_interval:
42
  print("🔄 检查索引更新...")
43
 
44
- # 获取最新元数据
45
  METADATA_PATH = hf_hub_download(
46
  repo_id="GOGO198/GOGO_rag_index",
47
  filename="metadata.csv",
48
  cache_dir=CACHE_DIR,
49
  token=os.getenv("HF_TOKEN"),
50
- force_download=True # 强制更新
51
  )
52
 
53
- # 检查文件修改时间
54
  file_mtime = os.path.getmtime(METADATA_PATH)
55
  if file_mtime > last_updated:
56
  print("📥 检测到新索引,重新加载...")
57
 
58
- # 重新加载索引
59
  INDEX_PATH = hf_hub_download(
60
  repo_id="GOGO198/GOGO_rag_index",
61
  filename="faiss_index.bin",
@@ -66,7 +63,6 @@ def refresh_index():
66
  new_index = faiss.read_index(INDEX_PATH)
67
  new_metadata = pd.read_csv(METADATA_PATH)
68
 
69
- # 原子替换
70
  index = new_index
71
  metadata = new_metadata
72
  last_updated = file_mtime
@@ -76,8 +72,7 @@ def refresh_index():
76
  except Exception as e:
77
  print(f"索引更新失败: {str(e)}")
78
 
79
- # 每30秒检查一次
80
- time.sleep(30)
81
 
82
  def load_resources():
83
  """加载所有必要资源(768维专用)"""
 
36
 
37
  while True:
38
  try:
 
39
  current_time = time.time()
40
  if current_time - last_updated > index_refresh_interval:
41
  print("🔄 检查索引更新...")
42
 
43
+ # 强制下载最新文件
44
  METADATA_PATH = hf_hub_download(
45
  repo_id="GOGO198/GOGO_rag_index",
46
  filename="metadata.csv",
47
  cache_dir=CACHE_DIR,
48
  token=os.getenv("HF_TOKEN"),
49
+ force_download=True
50
  )
51
 
 
52
  file_mtime = os.path.getmtime(METADATA_PATH)
53
  if file_mtime > last_updated:
54
  print("📥 检测到新索引,重新加载...")
55
 
 
56
  INDEX_PATH = hf_hub_download(
57
  repo_id="GOGO198/GOGO_rag_index",
58
  filename="faiss_index.bin",
 
63
  new_index = faiss.read_index(INDEX_PATH)
64
  new_metadata = pd.read_csv(METADATA_PATH)
65
 
 
66
  index = new_index
67
  metadata = new_metadata
68
  last_updated = file_mtime
 
72
  except Exception as e:
73
  print(f"索引更新失败: {str(e)}")
74
 
75
+ time.sleep(30) # 每30秒检查
 
76
 
77
  def load_resources():
78
  """加载所有必要资源(768维专用)"""