Update app.py

app.py

@@ -74,35 +74,17 @@ def load_resources():
 def predict(vector):
     """处理768维向量输入并返回答案"""
     try:
-        start_time = time.time()
-        load_resources()  # 确保资源已加载
+        # 确保输入数据完整
+        if len(vector[0]) != 768:
+            return "维度错误：需要768维向量"
         
-        # 改进维度处理
-        if isinstance(vector, list) and isinstance(vector[0], list):
-            # 已经是二维数组
-            vector = np.array(vector, dtype=np.float32)
-        else:
-            # 一维数组处理
-            vector = np.array([vector], dtype=np.float32)
-            
-        if vector.shape[1] != 768:
-            return f"维度错误：预期768维，收到{vector.shape[1]}维"
-
-        # 添加处理日志
-        print(f"接收请求 | 维度: {vector.shape}")
-            
-        # FAISS 搜索
-        D, I = index.search(vector, k=3)
-        
-        # 获取最相关结果
-        results = []
-        for i in range(3):
-            result = metadata.iloc[I[0][i]]
-            results.append(f"相关结果 {i+1}: {result['source']} | 相似度: {1/(1+D[0][i]):.2f}")
-        
-        response = "\n".join(results)
-        print(f"处理时间: {time.time() - start_time:.2f}秒")
-        return response
+        # 添加实际处理逻辑
+        D, I = index.search(np.array(vector, dtype=np.float32), k=3)
+        results = [
+            f"匹配结果 {i+1}: {metadata.iloc[I[0][i]]['source']} | 置信度: {1/(1+D[0][i]):.2f}"
+            for i in range(3)
+        ]
+        return "\n".join(results)
         
     except Exception as e:
         # 添加详细错误日志