[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "coqui-tts"
version = "0.28.0.dev0"
requires-python = ">=3.10, <3.15"
dependencies = [
    # Core
    "numpy>=1.26.0",
    "scipy>=1.13.0",
    "soundfile>=0.12.0",
    "librosa>=0.11.0",
    "numba>=0.58.0",
    "inflect>=5.6.0",
    "tqdm>=4.64.1",
    "anyascii>=0.3.0",
    "pyyaml>=6.0",
    "fsspec[http]>=2023.6.0",
    "packaging>=23.1",
    "typing_extensions>=4.10",
    # Inference
    "pysbd>=0.3.4",
    # Training
    "matplotlib>=3.8.4",
    # Coqui stack
    "coqui-tts-trainer>=0.3.0,<0.4.0",
    "coqpit-config>=0.2.3,<0.3.0",
    "monotonic-alignment-search>=0.1.0",
    "ko-speech-tools>=0.1.0",
    # Tortoise
    "einops>=0.6.0",
    "transformers>=4.57",
    "huggingface-hub>=0.36.0",
    # XTTS
    "num2words>=0.5.14",
]

[project.optional-dependencies]
cpu = [
    "torch>=2.2",
    "torchaudio>=2.2.0",
]
cuda = [
    "torch>=2.2",
    "torchaudio>=2.2.0",
]
# For running the TTS server
server = ["flask>=3.0.0"]

[tool.uv]
required-environments = [
    "sys_platform == 'linux' and platform_machine == 'x86_64'"
]
conflicts = [
  [
    { extra = "cpu" },
    { extra = "cuda" },
  ],
]

[tool.uv.sources]
torch = [
  { index = "pytorch-cpu", extra = "cpu" },
  { index = "pytorch-cuda", extra = "cuda" },
]
torchaudio = [
  { index = "pytorch-cpu", extra = "cpu" },
  { index = "pytorch-cuda", extra = "cuda" },
]

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

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

[tool.hatch.build.targets.wheel]
packages = ["TTS"]