Update app.py

app.py

@@ -164,18 +164,36 @@ def predict(vector):
         for i in range(k):
             try:
                 idx = I[0][i]
+                
+                # 安全处理距离值
+                distance = D[0][i]
+                if not np.isfinite(distance) or distance < 0:
+                    print(f"警告: 无效距离值 {distance}，使用默认值0")
+                    distance = 0.0
+                
+                # 安全处理置信度值
+                try:
+                    confidence = 1 / (1 + distance)
+                except ZeroDivisionError:
+                    confidence = 1.0
+                
+                if not np.isfinite(confidence) or confidence < 0:
+                    print(f"警告: 无效置信度 {confidence}，使用默认值0.5")
+                    confidence = 0.5
+                
                 result = {
                     "source": metadata.iloc[idx]["source"],
                     "content": metadata.iloc[idx].get("content", ""),
-                    "confidence": float(1/(1+D[0][i])),
-                    "distance": float(D[0][i])
+                    "confidence": float(confidence),
+                    "distance": float(distance)
                 }
                 results.append(result)
             except Exception as e:
                 print(f"结果处理错误: {str(e)}")
                 results.append({
                     "error": f"结果 {i+1}: 数据获取失败",
-                    "confidence": 0.0
+                    "confidence": 0.5,
+                    "distance": 0.0
                 })
 
         return {
@@ -183,13 +201,11 @@ def predict(vector):
             "results": results
         }
     except Exception as e:
-        return JSONResponse(
-            status_code=500,
-            content={
-                "status": "error",
-                "message": f"服务器内部错误: {str(e)}"
-            }
-        )
+        # 返回错误响应
+        return {
+            "status": "error",
+            "message": f"服务器内部错误: {str(e)}"
+        }
         
 # 创建FastAPI应用
 app = FastAPI()