Update app.py

app.py

@@ -84,11 +84,15 @@ from functools import lru_cache
 @lru_cache(maxsize=100)
 def predict(vector):
     """处理768维向量输入并返回答案"""
+    start_time = time.time()
     print(f"输入向量维度: {np.array(vector).shape}")
+    
     try:
+        # 验证输入格式
         if not isinstance(vector, list) or len(vector) == 0:
-            print("错误：输入格式无效")
-            return "错误：输入格式无效"
+            error_msg = "错误：输入格式无效"
+            print(error_msg)
+            return error_msg
 
         if len(vector) != 1 or len(vector[0]) != 768:
             error_msg = f"错误：需要1x768的二维数组，收到{len(vector)}x{len(vector[0]) if vector else 0}"
@@ -96,20 +100,27 @@ def predict(vector):
             return error_msg
         
         # 添加实际处理逻辑
-        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)
-        ]
+        vector_array = np.array(vector, dtype=np.float32)
+        D, I = index.search(vector_array, k=3)
+        
+        results = []
+        for i in range(3):
+            try:
+                result = metadata.iloc[I[0][i]]
+                confidence = 1/(1+D[0][i])
+                results.append(f"匹配结果 {i+1}: {result['source']} | 置信度: {confidence:.2f}")
+            except Exception as e:
+                print(f"结果处理错误: {str(e)}")
+                results.append(f"结果 {i+1}: 数据获取失败")
+        
+        print(f"处理完成 | 耗时: {time.time()-start_time:.2f}秒")
         return "\n".join(results)
         
     except Exception as e:
-        # 添加详细错误日志
-        error_msg = "处理错误，请重试或联系管理员"
-        # import traceback
-        # error_msg = f"处理错误: {str(e)}\n{traceback.format_exc()}"
+        import traceback
+        error_msg = f"处理错误: {str(e)}\n{traceback.format_exc()}"
         print(error_msg)
-        return error_msg
+        return "处理错误，请重试或联系管理员"
 
 # 创建简化接口
 with gr.Blocks() as demo: