Update app.py

app.py

@@ -25,6 +25,59 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 index = None
 metadata = None
 
+# 新增全局变量
+last_updated = 0
+index_refresh_interval = 300  # 5分钟刷新一次
+
+# 新增索引刷新函数
+def refresh_index():
+    global index, metadata, last_updated
+    
+    while True:
+        try:
+            # 检查是否有更新
+            current_time = time.time()
+            if current_time - last_updated > index_refresh_interval:
+                print("🔄 检查索引更新...")
+                
+                # 获取最新元数据
+                METADATA_PATH = hf_hub_download(
+                    repo_id="GOGO198/GOGO_rag_index",
+                    filename="metadata.csv",
+                    cache_dir=CACHE_DIR,
+                    token=os.getenv("HF_TOKEN"),
+                    force_download=True  # 强制更新
+                )
+                
+                # 检查文件修改时间
+                file_mtime = os.path.getmtime(METADATA_PATH)
+                if file_mtime > last_updated:
+                    print("📥 检测到新索引，重新加载...")
+                    
+                    # 重新加载索引
+                    INDEX_PATH = hf_hub_download(
+                        repo_id="GOGO198/GOGO_rag_index",
+                        filename="faiss_index.bin",
+                        cache_dir=CACHE_DIR,
+                        token=os.getenv("HF_TOKEN"),
+                        force_download=True
+                    )
+                    new_index = faiss.read_index(INDEX_PATH)
+                    new_metadata = pd.read_csv(METADATA_PATH)
+                    
+                    # 原子替换
+                    index = new_index
+                    metadata = new_metadata
+                    last_updated = file_mtime
+                    
+                    print(f"✅ 索引更新完成 | 记录数: {len(metadata)}")
+        
+        except Exception as e:
+            print(f"索引更新失败: {str(e)}")
+        
+        # 每30秒检查一次
+        time.sleep(30)
+
 def load_resources():
     """加载所有必要资源（768维专用）"""
     global index, metadata
@@ -77,27 +130,45 @@ def load_resources():
                 print(f"❌ 元数据加载失败: {str(e)}")
                 raise
 
+# 启动索引刷新线程
+threading.Thread(target=refresh_index, daemon=True).start()
+
 # 确保资源在API调用前加载
 load_resources()
 
 def predict(vector):
     try:
+        print(f"接收向量: {vector[:3]}... (长度: {len(vector)})")
+        
         # 确保向量格式正确
         query_vector = np.array(vector).astype('float32').reshape(1, -1)
         
-        # FAISS搜索
-        D, I = index.search(query_vector, k=3)
+        # 动态结果数量 (最大不超过总文档数)
+        k = min(3, index.ntotal)
+        if k == 0:
+            return {
+                "status": "success",
+                "results": [],
+                "message": "索引为空"
+            }
+        
+        print(f"执行FAISS搜索 | k={k}")
+        D, I = index.search(query_vector, k=k)
+        
+        # 打印搜索结果
+        print(f"搜索结果索引: {I[0]}")
+        print(f"搜索距离: {D[0]}")
         
-        # 构建结果 - 使用安全的列名访问
+        # 构建结果
         results = []
-        for i in range(3):
+        for i in range(k):
             try:
                 idx = I[0][i]
                 result = {
                     "source": metadata.iloc[idx]["source"],
-                    # 安全访问content字段
                     "content": metadata.iloc[idx].get("content", ""),
-                    "confidence": float(1/(1+D[0][i]))
+                    "confidence": float(1/(1+D[0][i])),
+                    "distance": float(D[0][i])
                 }
                 results.append(result)
             except Exception as e:
@@ -165,6 +236,6 @@ if __name__ == "__main__":
     print(f"索引维度: {index.d}")
     print(f"元数据记录数: {len(metadata)}")
     print("="*50)
-    
+
     # 只启动FastAPI服务
     uvicorn.run(app, host="0.0.0.0", port=7860)