from pathlib import Path
from setuptools import setup, find_packages

NAME = "soundsauce"
DESCRIPTION = ""
URL = ""
EMAIL = ""
AUTHOR = ""
REQUIRES_PYTHON = ">=3.9.0"
VERSION = "0.0.1"

HERE = Path(__file__).parent

try:
    with open(HERE / "README.md", encoding="utf-8") as f:
        long_description = "\n" + f.read()
except FileNotFoundError:
    long_description = DESCRIPTION

setup(
    name=NAME,
    version=VERSION,
    description=DESCRIPTION,
    long_description=long_description,
    long_description_content_type="text/markdown",
    author=AUTHOR,
    author_email=EMAIL,
    python_requires=REQUIRES_PYTHON,
    url=URL,
    packages=find_packages(),
    install_requires=[
        "torch>=2.0",
        "torchaudio>=2.0.1",
        "soundfile",
        "flask",
        "flask-socketio",
        "audiocraft@git+https://github.com/facebookresearch/audiocraft",
        "gradio",
        "gradio_client",
        "pretty_midi"
    ],
    include_package_data=True,
)