Update app.py

app.py

@@ -73,14 +73,20 @@ def predict(vector):
         start_time = time.time()
         load_resources()  # 确保资源已加载
         
-        # 转换为numpy数组并验证维度
-        vector = np.array(vector, dtype=np.float32)
-        if vector.ndim == 1:  # 自动升维
-            vector = vector.reshape(1, -1)
+        # 改进维度处理
+        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)
         
@@ -95,7 +101,11 @@ def predict(vector):
         return response
         
     except Exception as e:
-        return f"处理错误: {str(e)}"
+        # 添加详细错误日志
+        import traceback
+        error_msg = f"处理错误: {str(e)}\n{traceback.format_exc()}"
+        print(error_msg)
+        return error_msg
 
 # 创建简化接口
 with gr.Blocks() as demo:
@@ -134,6 +144,11 @@ if __name__ == "__main__":
     except:
         pass
 
+    print("="*50)
+    print("Space启动完成 | 准备接收请求")
+    print(f"索引维度: {index.d if index else '未加载'}")
+    print(f"元数据记录: {len(metadata) if metadata is not None else 0}")
+    print("="*50)
     
     demo.launch(
         server_name="0.0.0.0",