[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"
regex = "^2024.11.6"
numpy = "1.23.5"
tqdm = "^4.67.1"
requests = ">=2.31.0,<2.32.0"
six = "^1.16.0"
wget = "^3.2"
pandas = "^2.2.3"

[tool.poetry.group.ml.dependencies]
torch = "2.3.1"
transformers = "4.44.2"
libf0 = "^1.0.2"
torchcrepe = "0.0.23"
torchfcpe = "^0.0.4"
einops = "^0.8.0"
torchaudio = "2.3.1"
scikit-learn = "^1.5.2"

[tool.poetry.group.cli.dependencies]
click = "^8.1.7"
click-help-colors = "^0.9.4"

[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"

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

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


[tool.poetry.group.ui.dependencies]
matplotlib = "3.7.2"
tensorboard = "^2.18.0"


[tool.poetry.group.audio.dependencies]
librosa = "0.9.2"
pydub = "^0.25.1"
resampy = "^0.4.3"
pedalboard = "^0.9.16"
scipy = "1.11.1"
faiss-cpu = "1.7.3"
noisereduce = "^3.0.3"
stftpitchshift = "^2.0"
versatile-audio-upscaler = "^0.0.2"


[tool.poetry.group.misc.dependencies]
beautifulsoup4 = "^4.12.3"
pypresence = "^4.3.0"
flask = "^3.1.0"
local-attention = "^1.9.15"
edge-tts = "6.1.9"

[tool.ruff]
line-length = 132

[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"

#1181 errors
[[tool.mypy.overrides]]
module = [
    "core.*",
    "rvc.infer.infer",
    "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.preprocess.preprocess",
    "rvc.train.preprocess.slicer",
    "rvc.train.process.extract_small_model",
    "tabs.extra.f0_extractor.f0_extractor",
]
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",
    "pydub",
    "pypresence",
    "resampy",
    "scipy.*",
    "sklearn.*",
    "soundfile",
    "stftpitchshift",
    "tensorboard",
    "torchaudio.*",
    "torchcrepe",
    "torchfcpe",
    "transformers",
    "wget",
]
ignore_missing_imports = true

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