#!/bin/bash
# Python依赖安装脚本 (Linux/macOS)

echo "========================================"
echo "Python依赖安装脚本"
echo "========================================"
echo ""

# 检查Python是否安装
if ! command -v python3 &> /dev/null; then
    echo "错误: 未找到Python3，请先安装Python"
    exit 1
fi

echo "检测到Python版本:"
python3 --version
echo ""

# 升级pip
echo "正在升级pip..."
python3 -m pip install --upgrade pip
echo ""

# 选择安装方式
echo "请选择安装方式:"
echo "1. 安装所有依赖 (包括二维码功能)"
echo "2. 只安装核心依赖 (PyMuPDF)"
echo ""
read -p "请输入选择 (1 或 2): " choice

if [ "$choice" = "1" ]; then
    echo ""
    echo "正在安装所有依赖..."
    pip3 install -r requirements.txt
    echo ""
    echo "注意: 二维码功能需要安装zbar库"
    if [[ "$OSTYPE" == "darwin"* ]]; then
        echo "macOS: brew install zbar"
    else
        echo "Linux: sudo apt-get install libzbar0"
    fi
elif [ "$choice" = "2" ]; then
    echo ""
    echo "正在安装核心依赖..."
    pip3 install PyMuPDF
else
    echo "无效的选择"
    exit 1
fi

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

# 验证安装
echo "验证安装..."
python3 -c "import fitz; print('PyMuPDF version:', fitz.version)"

echo ""