#!/bin/bash
# 下载并安装 ONNX Runtime Linux x64

set -e

VERSION="1.16.1"
URL="https://github.com/microsoft/onnxruntime/releases/download/v${VERSION}/onnxruntime-linux-x64-${VERSION}.tgz"
TARGET_DIR="/home/ubuntu/workspace/cherry-studio/enginneer/impress_asr_input_rust/lib"

echo "下载 ONNX Runtime ${VERSION}..."

mkdir -p "$TARGET_DIR"
cd "$TARGET_DIR"

# 使用代理下载
export http_proxy="http://127.0.0.1:7897"
export https_proxy="http://127.0.0.1:7897"

curl -L -o onnxruntime.tgz "$URL"

echo "解压..."
tar xzf onnxruntime.tgz

echo "设置环境变量..."
echo "export ONNXRUNTIME_LIB=${TARGET_DIR}/onnxruntime-linux-x64-${VERSION}/lib" >> ~/.bashrc
echo "export LD_LIBRARY_PATH=${TARGET_DIR}/onnxruntime-linux-x64-${VERSION}/lib:\$LD_LIBRARY_PATH" >> ~/.bashrc

echo "完成！请运行：source ~/.bashrc"