[project]
name = "trap"
version = "0.1.0"
description = "Art installation with trajectory prediction"
authors = [{ name = "Ruben van de Ven", email = "git@rubenvandeven.com" }]
requires-python = "~=3.10.4"
readme = "README.md"
dependencies = [
    "trajectron-plus-plus",
    "torch==1.12.1",
    "torchvision==0.13.1",
    "deep-sort-realtime>=1.3.2,<2",
    "ultralytics~=8.3",
    "ffmpeg-python>=0.2.0,<0.3",
    "torchreid>=0.2.5,<0.3",
    "gdown>=4.7.1,<5",
    "pandas-helper-calc",
    "tsmoothie>=1.0.5,<2",
    "pyglet>=2.0.15,<3",
    "pyglet-cornerpin>=0.3.0,<0.4",
    "opencv-python",
    "setproctitle>=1.3.3,<2",
    "bytetracker",
    "jsonlines>=4.0.0,<5",
    "tensorboardx>=2.6.2.2,<3",
    "shapely>=1,<2",
    "baumer-neoapi",
    "qrcode~=8.0",
    "pyusb>=1.3.1,<2",
    "ipywidgets>=8.1.5,<9",
    "foucault",
    "python-statemachine>=2.5.0",
    "facenet-pytorch>=2.5.3",
    "simplification>=0.7.12",
]

[project.scripts]
start = "trap.conductofconduct:run"
trapserv = "trap.plumber:start"
tracker = "trap.tools:tracker_preprocess"
compare = "trap.tools:tracker_compare"
process_data = "trap.process_data:main"
blacklist = "trap.tools:blacklist_tracks"
rewrite_tracks = "trap.tools:rewrite_raw_track_files"
live_video_source = "trap.frame_emitter:run"
live_tracker = "trap.tracker:run"

[tool.uv]

[tool.uv.sources]
trajectron-plus-plus = { path = "../Trajectron-plus-plus/", editable = true }
torch = [{ url = "https://download.pytorch.org/whl/cu113/torch-1.12.1%2Bcu113-cp310-cp310-linux_x86_64.whl", marker = "python_version ~= '3.10' and sys_platform == 'linux'" }]
torchvision = [{ url = "https://download.pytorch.org/whl/cu113/torchvision-0.13.1%2Bcu113-cp310-cp310-linux_x86_64.whl", marker = "python_version ~= '3.10' and sys_platform == 'linux'" }]
pandas-helper-calc = { git = "https://github.com/scls19fr/pandas-helper-calc" }
bytetracker = { git = "https://github.com/rubenvandeven/bytetrack-pip" }
baumer-neoapi = { path = "../../Downloads/Baumer_neoAPI_1.4.1_lin_x86_64_python/wheel/baumer_neoapi-1.4.1-cp34.cp35.cp36.cp37.cp38.cp39.cp310.cp311.cp312-none-linux_x86_64.whl" }
foucault = { git = "https://git.rubenvandeven.com/r/conductofconduct" }
opencv-python = {path="./opencv_python-4.10.0.84-cp310-cp310-linux_x86_64.whl"}

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