#!/bin/bash
# Mac 环境设置脚本

echo "================================"
echo "Mac 环境设置 - Qt 版本"
echo "================================"

# 检查 Python 版本
python_version=$(python3 --version 2>&1)
echo "Python 版本: $python_version"

# 创建虚拟环境
if [ ! -d "venv" ]; then
    echo "创建虚拟环境..."
    python3 -m venv venv
fi

# 激活虚拟环境
echo "激活虚拟环境..."
source venv/bin/activate

# 升级 pip
echo "升级 pip..."
pip install --upgrade pip

# 卸载可能冲突的旧版本
echo "清理旧版本..."
pip uninstall -y google-genai Pillow PyQt5 pyinstaller 2>/dev/null

# 安装依赖
echo "================================"
echo "安装依赖..."
echo "================================"

# 如果有锁定版本文件，优先使用
if [ -f "requirements-lock.txt" ]; then
    echo "使用锁定版本 (requirements-lock.txt)..."
    pip install -r requirements-lock.txt
else
    echo "使用范围版本 (requirements.txt)..."
    pip install -r requirements.txt
fi

# 验证安装
echo "================================"
echo "验证安装..."
echo "================================"

pip list | grep -E "google-genai|Pillow|PyQt5|pyinstaller"

echo "================================"
echo "设置完成!"
echo "================================"
echo ""
echo "运行应用: python3 image_generator_qt.py"
echo ""