pyproject.toml

[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "augmenator"
version = "0.1.0"
description = "Text-driven image augmentation via embedding-matched keywords"
readme = "README.md"
license = "MIT"
license-files = ["LICENSE"]
requires-python = ">=3.10"
authors = [{ name = "Dag Bjornberg" }]
keywords = ["image", "augmentation", "style-transfer", "computer-vision"]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Developers",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Image Processing",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
    "torch>=2.1.0",
    "torchvision>=0.16.0",
    "transformers>=4.40.0",
    "sentence-transformers>=2.7.0",
    "Pillow>=10.0.0",
    "numpy>=1.24.0",
    "rapidocr-onnxruntime>=1.3.0",
    "opencv-python-headless>=4.8.0",
    "rembg[cpu]>=2.0.50",
    "onnxruntime>=1.16.0",
    "requests>=2.28.0",
    "huggingface_hub>=0.20.0",
]

[project.optional-dependencies]
ui = ["gradio>=5.23.1"]
dev = ["build", "twine"]

[project.urls]
Homepage = "https://github.com/DageBjorne/augmenator"
Repository = "https://github.com/DageBjorne/augmenator"
"Hugging Face Space" = "https://huggingface.co/spaces/dagbjorn/text-driven-image-augmentation"

[project.scripts]
augmenator-batch = "augmenator.cli:main"

[tool.setuptools.packages.find]
include = ["augmenator*"]
exclude = ["scripts*", "backgrounds*", "generated_augmentations*"]