[tool.poetry]
name = "tts-service"
version = "0.1.0"
description = ""
authors = ["Jesus Lopez <jesus@jesusla.com>"]
readme = "README.md"
packages = [
    { include = "assets", from = "." },
    { include = "core", from = "." },
    { include = "rvc", from = "." },
    { include = "tabs", from = "." },
    { include = "tts_service", from = "." },
]


[tool.poetry.scripts]
tts-service = "tts_service.cli:main"


[tool.poetry.dependencies]
python = "~3.10"
beautifulsoup4 = "^4.12.3"
boto3 = "^1.35.70"
click = "^8.1.7"
click-help-colors = "^0.9.4"
edge-tts = "6.1.9"
einops = "^0.8.0"
faiss-cpu = "1.7.3"
gradio = "4.43.0"
librosa = "0.9.2"
local-attention = "^1.9.15"
matplotlib = "3.7.2"
noisereduce = "^3.0.3"
numpy = "1.23.5"
pandas = "^2.2.3"
pedalboard = "^0.9.16"
pydantic = "^2.10.2"
python-dotenv = "^1.0.1"
requests = ">=2.31.0,<2.32.0"
scikit-learn = "^1.5.2"
scipy = "1.11.1"
six = "^1.16.0"
soundfile = "^0.12.1"
stftpitchshift = "^2.0"
torch = "2.3.1"
torchaudio = "2.3.1"
torchcrepe = "0.0.23"
tqdm = "^4.67.1"
transformers = "4.44.2"
versatile-audio-upscaler = "^0.0.2"
wget = "^3.2"

[tool.poetry.group.ci.dependencies]
gradio = "4.43.0"
huggingface-hub = "^0.26.2"

[tool.poetry.group.dev.dependencies]
debugpy = "^1.8.9"
ipykernel = "^6.29.5"
pipdeptree = "^2.23.4"
pre-commit = "^4.0.1"
ruff = "^0.7.4"
vulture = "^2.13"
deptry = "^0.21.1"

[tool.poetry.group.test.dependencies]
pytest = "^8.3.3"
pytest-cov = "^6.0.0"

[tool.poetry.group.typing.dependencies]
mypy = "^1.13.0"
pandas-stubs = "^2.2.3.241009"
types-beautifulsoup4 = "^4.12.0.20241020"
types-boto3 = "^1.0.2"
types-regex = "^2024.11.6.20241108"
types-requests = "^2.32.0.20241016"
types-six = "^1.16.21.20241105"


[tool.ruff]
line-length = 132
include = ["tts-service/**/*.py"]

[tool.ruff.lint]
select = [
    # pycodestyle
    "E",
    # Pyflakes
    "F",
    # pyupgrade
    "UP",
    # flake8-bugbear
    "B",
    # flake8-simplify
    "SIM",
    # isort
    "I",
]


[tool.mypy]
packages = "assets,core,rvc,tabs,tts_service,tests"
check_untyped_defs = true
explicit_package_bases = true
namespace_packages = true

[[tool.mypy.overrides]]
module = [
    "rvc.lib.algorithm.generators",
    "rvc.lib.algorithm.residuals",
    "rvc.lib.predictors.RMVPE",
    "rvc.lib.tools.gdown",
    "rvc.lib.tools.model_download",
    "rvc.train.losses",
    "rvc.train.process.extract_model",
    "rvc.train.process.model_blender",
]
check_untyped_defs = false

[[tool.mypy.overrides]]
module = [
    "rvc.infer.pipeline",
    "rvc.lib.algorithm.attentions",
    "rvc.lib.algorithm.commons",
    "rvc.lib.algorithm.synthesizers",
    "rvc.lib.predictors.FCPE",
    "rvc.lib.zluda",
    "rvc.train.train",
    "rvc.train.data_utils",
    "rvc.train.extract.extract",
    "rvc.train.mel_processing",
    "rvc.train.preprocess.preprocess",
    "rvc.train.preprocess.slicer",
    "rvc.train.process.extract_small_model",
    "tabs.extra.f0_extractor.f0_extractor",
    "tts_service.whitelist",
]
ignore_errors = true

[[tool.mypy.overrides]]
module = [
    "audio_upscaler",
    "edge_tts",
    "faiss",
    "gradio",
    "gradio.themes.base",
    "gradio.themes.utils",
    "libf0",
    "librosa.*",
    "local_attention",
    "matplotlib.*",
    "noisereduce",
    "pedalboard_native",
    "pydub",
    "pypresence",
    "resampy",
    "scipy.*",
    "sklearn.*",
    "soundfile",
    "stftpitchshift",
    "tensorboard",
    "torchaudio.*",
    "torchcrepe",
    "torchfcpe",
    "tqdm",
    "transformers",
    "wget",
]
ignore_missing_imports = true

[[tool.mypy.overrides]]
module = [
    "rvc.lib.tools.prerequisites_download",
    "tts_service.cli",
    "tts_service.utils",
]
warn_unused_configs = true
disallow_any_generics = true
disallow_subclassing_any = true
disallow_untyped_calls = true
disallow_untyped_defs = true
disallow_incomplete_defs = true
check_untyped_defs = true
disallow_untyped_decorators = true
warn_unused_ignores = true
warn_return_any = true
no_implicit_reexport = true

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"