pyproject.toml

[tool.poetry]
name = "koregraph"
version = "0.0.0"
description = "An AI for music driven choregraphy generation."
authors = [
    "Anatole-DC <adechauveron@gmail.com>",
    "Kelly-an <kdanne206@gmail.com>",
    "Robin Le Floch <robin.le.floch@gmail.com>",
    "maudbenichou <maudb.benichou@gmail.com>"
]
readme = "README.md"
homepage = "https://anatole-dc.github.io/koregraph/#/"


[tool.poetry.dependencies]
python = "^3.10"
python-dotenv = "^1.0.0"
urllib3 = "^2.2.1"
requests = "^2.32.2"
librosa = "^0.10.2.post1"
matplotlib = "^3.9.0"
pydub = "^0.25.1"

[tool.poetry.group.dev]
optional = true

[tool.poetry.group.dev.dependencies]
pytest = "^7.4.2"
flake8 = "^6.1.0"
black = "^24.4.2"


[tool.poetry.group.notebook]
optional = true

[tool.poetry.group.notebook.dependencies]
jupyterlab = "^4.2.1"
black = {extras = ["jupyter"], version = "^24.4.2"}


[tool.poetry.group.viewer]
optional = true

[tool.poetry.group.viewer.dependencies]
mutagen = "^1.47.0"
pillow = "^10.3.0"
moviepy = "^1.0.3"


[tool.poetry.group.chunk]
optional = true

[tool.poetry.group.chunk.dependencies]
soundfile = "^0.12.1"


[tool.poetry.group.training]
optional = true

[tool.poetry.group.training.dependencies]
mlflow = "^2.13.1"
tensorflow = "^2.16.1"



[tool.poetry.group.cloud.dependencies]
google-cloud-storage = "^2.16.0"


[tool.poetry.scripts]
preprocess = "koregraph.cli.preprocess:main"
viewer = "koregraph.cli.viewer:main"
chunk = "koregraph.cli.chunk_command:main"
train = "koregraph.cli.model:main"
predict = "koregraph.cli.predict:main"
evaluate = "koregraph.cli.evaluate:main"
data = "koregraph.cli.dataset:main"


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