[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "captchbreaker"
version = "0.1.0"
description = "验证码识别多模型系统 - 调度模型 + 多专家模型两级架构"
requires-python = ">=3.10,<3.13"
dependencies = [
    "torch==2.5.1",
    "torchvision==0.20.1",
    "onnx>=1.14.0",
    "onnxscript>=0.6.0",
    "onnxruntime>=1.15.0,<1.24.0",
    "pillow>=10.0.0",
    "numpy>=1.24.0",
    "tqdm>=4.65.0",
]

[project.optional-dependencies]
server = [
    "fastapi>=0.100.0",
    "uvicorn>=0.23.0",
    "python-multipart>=0.0.6",
]
cv = [
    "opencv-python>=4.8.0",
]
dev = [
    "pytest>=7.0.0",
]

[project.scripts]
captcha = "cli:main"

[tool.uv.sources]
torch = [
    { index = "pytorch-cu121", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" },
]
torchvision = [
    { index = "pytorch-cu121", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" },
]

[[tool.uv.index]]
name = "pytorch-cu121"
url = "https://download.pytorch.org/whl/cu121"
explicit = true

[tool.setuptools]
py-modules = ["cli", "config", "server"]

[tool.setuptools.packages.find]
include = [
    "generators",
    "generators.*",
    "models",
    "models.*",
    "training",
    "training.*",
    "inference",
    "inference.*",
    "solvers",
    "solvers.*",
    "utils",
    "utils.*",
]