# AIToolkit 开发环境依赖
# 使用此文件来设置一个统一的开发和测试环境

# 安装核心包本身（以可编辑模式）
# 这将同时安装 setup.py 中定义的核心依赖和可选依赖
# -e .[all,dev]

# 核心依赖（供参考，由setup.py管理）
# mediapipe>=0.10.0
# opencv-python>=4.8.0
# numpy>=1.21.0,<1.25.0
# Pillow>=9.0.0
# protobuf>=3.20.0,<5.0.0

# 为了方便，可以直接安装所有依赖项
-e .[all,dev]

# 增强OCR支持
cnocr>=2.2.0  # 中文OCR，会自动安装相关依赖

# 深度学习依赖（可选）
torch>=1.11.0; platform_system != "Windows" or python_version >= "3.8"
torchvision>=0.12.0; platform_system != "Windows" or python_version >= "3.8"

# 模型训练依赖（简化版）
tensorflow>=2.12.0,<2.15.0
scikit-learn>=1.2.0
matplotlib>=3.5.0
pandas>=1.5.0
tqdm>=4.64.0

# 兼容性修复
protobuf>=3.20.0,<5.0.0  # 更宽松的protobuf版本

# 可选的模型制作器（注释掉减少冲突）
# mediapipe-model-maker>=0.10.7

# WebRTC相关依赖（可选）
# aiortc>=1.3.2
# websockets>=10.3
# av>=9.2.0 