#!/bin/bash

echo "批量图片背景移除工具 - 安装和运行脚本"
echo "======================================"

# 检查Python是否已安装
if ! command -v python3 &> /dev/null; then
    echo "错误: 未检测到Python。请安装Python 3.8或更高版本后再运行此脚本。"
    exit 1
fi

# 检查虚拟环境是否已存在
if [ ! -d "venv" ]; then
    echo "创建虚拟环境..."
    python3 -m venv venv
    if [ $? -ne 0 ]; then
        echo "错误: 创建虚拟环境失败。"
        exit 1
    fi
fi

# 激活虚拟环境并安装依赖
echo "激活虚拟环境并安装依赖..."
source venv/bin/activate
pip install -r requirements.txt
if [ $? -ne 0 ]; then
    echo "错误: 安装依赖失败。"
    exit 1
fi

# 检查NumPy版本并降级（如果需要）
numpy_version=$(python -c "import numpy as np; print(np.__version__)")
echo "当前NumPy版本: $numpy_version"

if [[ $numpy_version == 2.* ]]; then
    echo "检测到NumPy 2.x版本，这可能与onnxruntime不兼容"
    echo "正在降级NumPy到1.24.3..."
    pip install numpy==1.24.3
    if [ $? -ne 0 ]; then
        echo "警告: NumPy降级失败，程序可能无法正常运行。"
    else
        echo "NumPy已成功降级到1.24.3。"
    fi
fi

# 确保使用CPU版本的onnxruntime
pip uninstall -y onnxruntime-gpu
pip install onnxruntime==1.16.3

echo ""
echo "安装完成！"
echo ""

# 运行程序
echo "启动批量图片背景移除工具..."
python run.py

echo ""
echo "感谢使用！" 