pyproject.toml

[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"

[project]
name = "aTrain"
description = "aTrain is a tool for automatically transcribing speech recordings utilizing state-of-the-art machine learning models without uploading any data."

authors = [
  { name = "Armin Haberl", email = "armin.haberl@uni-graz.at" },
  { name = "Jürgen Fleiß", email = "juergen.fleiss@uni-graz.at" },
  { name = "Dominik Kowald", email = "dkowald@know-center.at" },
  { name = "Stefan Thalmann", email = "stefan.thalmann@uni-graz.at" },
]

dynamic = ["version"]
readme = "README.md"
license = { file = "LICENSE" }
requires-python = ">=3.11"

dependencies = [
  "aTrain_core@git+https://github.com/JuergenFleiss/atrain_core.git@develop",
  "wakepy==0.7.2",
  "show-in-file-manager==1.1.4",
  "typer==0.16.0",
  "nicegui==2.21.1",
  "pywebview==6.1",
  "pywebview[GTK]; sys_platform == 'linux'",
  "dask; sys_platform == 'linux'",
]

[tool.setuptools.packages.find]
where = ["."]
include = ["aTrain*"]

[project.scripts]
aTrain = "aTrain.app:cli"

[tool.setuptools.dynamic]
version = { attr = "aTrain.version.__version__" }

[tool.uv]
# torch+cu128 wheels live only on download.pytorch.org; PyPI rejects
# local-version tags (+...), so cross-index name collision is impossible.
index-strategy = "unsafe-best-match"

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

[tool.uv.sources]
torch       = [{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" }]
torchaudio  = [{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" }]
torchvision = [{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" }]

[dependency-groups]
dev = [
  "ruff>=0.15.0",
  "bandit[toml]>=1.9.0",
  "pip-audit>=2.10.0",
]

[tool.bandit]
# Only the aTrain package is scanned (CI runs `bandit -r aTrain`); these
# excludes keep ad-hoc local runs from descending into build/venv/model dirs.
exclude_dirs = [".venv", "build", "dist", "aTrain.egg-info", "aTrain/required_models"]