[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.metadata]
allow-direct-references = true

[tool.yapf]
based_on_style = "pep8"
indent_width = 4
column_limit = 100

[project]
name = "matanyone2"
version = "1.0.0"
authors = [{ name = "Peiqing Yang", email = "peiqingyang99@outlook.com" }]
description = "Robust and Real-Time Video Matting with Temporal Guidance"
readme = "README.md"
requires-python = ">=3.10"
classifiers = [
  "Programming Language :: Python :: 3",
  "Operating System :: OS Independent",
]
dependencies = [
  'numpy >= 1.21',
  'Pillow >= 9.5',
  'opencv-python >= 4.8',
  'scipy >= 1.7',
  'tqdm >= 4.66.1',
  'einops >= 0.6',
  'hydra-core >= 1.3.2',
  'requests',
  'imageio >= 2.25.0',
  'imageio[ffmpeg]',
  'huggingface_hub >= 0.25.0',
  'safetensors',
  'kornia',
  'easydict',
  'torch',
  'torchvision',
  'typer >= 0.9.0',
  "av>=16.1.0",
]

[project.optional-dependencies]
dev = [
  'cython',
  'gitpython >= 3.1',
  'thinplate@git+https://github.com/cheind/py-thin-plate-spline',
  'hickle >= 5.0',
  'tensorboard >= 2.11',
  'pycocotools >= 2.0.7',
  'gdown >= 4.7.1',
  'xlsxwriter',
]
gui = [
  'gradio >= 6.9.0',
  'PySide6 >= 6.2.0',
  'pyqtdarktheme',
]
all = [
  'cython',
  'gitpython >= 3.1',
  'thinplate@git+https://github.com/cheind/py-thin-plate-spline',
  'hickle >= 5.0',
  'tensorboard >= 2.11',
  'pycocotools >= 2.0.7',
  'gdown >= 4.7.1',
  'xlsxwriter',
  'gradio >= 6.9.0',
  'PySide6 >= 6.2.0',
  'pyqtdarktheme',
]

[project.scripts]
matanyone2 = "matanyone2.cli:app"

[tool.hatch.build.targets.wheel]
packages = ["matanyone2"]

[tool.hatch.build.targets.wheel.force-include]
"matanyone2/config" = "matanyone2/config"

[tool.hatch.build.targets.sdist]
include = [
  "matanyone2/",
  "inference_matanyone2.py",
  "README.md",
  "LICENSE.txt",
]

[tool.uv]
no-build-isolation-package = ["thinplate"]

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true

[tool.uv.sources]
torch = { index = "pytorch-cu128" }
torchvision = { index = "pytorch-cu128" }

[project.urls]
"Homepage" = "https://github.com/pq-yang/MatAnyone2"
"Bug Tracker" = "https://github.com/pq-yang/MatAnyone2/issues"