from pathlib import Path

from setuptools import find_packages, setup

README_TEXT = (Path(__file__).parent / "README.md").read_text(encoding="utf-8")

MAINTAINER = "Konstantin Verner"
MAINTAINER_EMAIL = "konst.verner@gmail.com"
REQUIRED_PKGS = ["accelerate==0.21.0",
                 "aiohttp==3.8.4",
                 "aiosignal==1.3.1",
                 "appdirs==1.4.4",
                 "async-timeout==4.0.2",
                 "attrs==23.1.0",
                 "audioread==3.0.0",
                 "certifi==2023.5.7",
                 "cffi==1.15.1",
                 "charset-normalizer==3.2.0",
                 "colorama==0.4.6",
                 "datasets==2.13.1",
                 "decorator>=4.0.2",
                 "dill==0.3.6",
                 "filelock==3.12.2",
                 "frozenlist==1.4.0",
                 "fsspec==2023.6.0",
                 "huggingface-hub==0.16.4",
                 "HyperPyYAML==1.2.1",
                 "idna==3.4",
                 "Jinja2==3.1.2",
                 "joblib==1.3.1",
                 "lazy_loader==0.3",
                 "librosa==0.10.0.post2",
                 "llvmlite==0.40.1",
                 "MarkupSafe==2.1.3",
                 "mpmath==1.3.0",
                 "msgpack==1.0.5",
                 "multidict==6.0.4",
                 "multiprocess==0.70.14",
                 "networkx==3.1",
                 "numba==0.57.1",
                 "numpy>=1.22",
                 "packaging==23.1",
                 "pandas>=1.5.3",
                 "pooch==1.6.0",
                 "psutil==5.9.5",
                 "pyarrow>=3.0.0",
                 "pycparser==2.21",
                 "python-dateutil==2.8.2",
                 "pytz==2023.3",
                 "PyYAML==6.0",
                 "ruamel.yaml==0.17.28",
                 "ruamel.yaml.clib==0.2.7",
                 "safetensors==0.3.1",
                 "scikit-learn==1.3.0",
                 "scipy==1.11.1",
                 "sentencepiece==0.1.99",
                 "six==1.16.0",
                 "soundfile==0.12.1",
                 "soxr==0.3.5",
                 "speechbrain==0.5.14",
                 "sympy==1.12",
                 "threadpoolctl==3.2.0",
                 "tokenizers==0.13.3",
                 "torch==2.0.1",
                 "torchaudio==2.0.2",
                 "tqdm==4.65.0",
                 "transformers==4.30.2",
                 "typing_extensions==4.7.1",
                 "tzdata==2023.3",
                 "urllib3==2.0.3",
                 "xxhash==3.2.0",
                 "yarl==1.9.2"]

print(find_packages("src"))

setup(
    name="deep_voice_cloning",
    version="0.1.0",
    description="Few-Shot Voice Cloning",
    long_description=README_TEXT,
    long_description_content_type="text/markdown",
    maintainer=MAINTAINER,
    maintainer_email=MAINTAINER_EMAIL,
    url="",
    download_url="",
    license="MIT",
    package_dir={"": "src"},
    packages=find_packages("src"),
    include_package_data=True,
    package_data={"": ["*.json"]},
    install_requires=REQUIRED_PKGS,
    classifiers=[
        "Development Status :: 1 - Planning",
        "Intended Audience :: Developers",
        "Intended Audience :: Education",
        "Intended Audience :: Science/Research",
        "License :: OSI Approved :: MIT",
        "Operating System :: OS Independent",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
    ],
    keywords="asr, machine learning, fewshot learning, transformers",
    zip_safe=False,  # Required for mypy to find the py.typed file
)