#!/usr/bin/env bash
set -euo pipefail

echo "=== 安装 Python 依赖 ==="
pip install numpy>=1.24 onnxruntime>=1.16

echo "=== 尝试安装 axengine（NPU 推理；仅 AX650 板端可用）==="
if python3 -c "import axengine" 2>/dev/null; then
    echo "axengine 已就绪"
else
    if pip install axengine 2>/dev/null; then
        echo "axengine 安装成功"
    else
        echo "⚠ 本机无法安装 axengine（通常因为不是 AX 板端环境）。"
        echo "  板端请先安装匹配的 axengine wheel，例如："
        echo "  pip3 install axengine-<版本>-py3-none-any.whl"
    fi
fi

echo "=== C++ SDK（可选）==="
echo "  需要 AX650 BSP SDK（含 ax_engine 头文件与库），然后："
echo "  mkdir -p cpp/build && cd cpp/build"
echo "  cmake .. -DAX650_SDK_ROOT=/path/to/bsp/out && make -j$(nproc)"

echo "✅ 环境准备完成"