#!/bin/bash

# 内容识别脚本依赖安装脚本

echo "开始安装内容识别脚本的依赖包..."

# 检查Python环境
if ! command -v python3 &> /dev/null; then
    echo "错误: 未找到 python3 命令"
    exit 1
fi

# 检查pip
if ! command -v pip3 &> /dev/null; then
    echo "错误: 未找到 pip3 命令"
    exit 1
fi

echo "Python版本: $(python3 --version)"
echo "Pip版本: $(pip3 --version)"

# 安装基础依赖
echo "安装基础依赖..."
pip3 install python-dotenv pymysql

# 安装HTTP请求库（包含SOCKS支持）
echo "安装HTTP请求库..."
pip3 install "httpx[socks]" requests

# 安装AI API客户端
echo "安装AI API客户端..."
pip3 install google-genai

# 安装飞书API客户端
echo "安装飞书API客户端..."
pip3 install lark-oapi

# 安装日志工具
echo "安装日志工具..."
pip3 install loguru

echo ""
echo "依赖安装完成！"
echo ""
echo "如果遇到权限问题，请使用:"
echo "  sudo pip3 install -r requirements.txt"
echo ""
echo "或者使用虚拟环境:"
echo "  python3 -m venv venv"
echo "  source venv/bin/activate"
echo "  pip install -r requirements.txt"