Update app.py
Browse files
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维专用)"""
|