Update app.py

app.py

@@ -105,8 +105,10 @@ def predict(vector):
             return {"error": error_msg}
 
         # 添加实际处理逻辑
-        vector_array = np.array(vector, dtype=np.float32)
-        D, I = index.search(vector_array, k=3)
+        #vector_array = np.array(vector, dtype=np.float32)
+        # 转换格式
+        query_vector = np.array(vector).astype('float32').reshape(1, -1)
+        D, I = index.search(query_vector, k=3)
 
         results = []
         for i in range(3):
@@ -125,8 +127,11 @@ def predict(vector):
         
         # 返回标准化的JSON格式
         return {
-            "status": "success",
-            "results": results
+            # "status": "success",
+            # "results": results
+            "source": [metadata.iloc[i]["source"] for i in I[0]],
+            "content": [metadata.iloc[i]["content"] for i in I[0]],
+            "confidence": [float(1/(1+d)) for d in D[0]]
         }
         
     except Exception as e: