# ComfyUI-Workflow-Studio - Tagger tab dependencies
#
# numpy and Pillow are already bundled with ComfyUI and do not need to be installed.
# torch is managed by ComfyUI and must NOT be pinned here.
#
# Install into ComfyUI's embedded Python:
#   python_embeded\python.exe -m pip install -r requirements.txt
#
# For GPU-accelerated ONNX inference, install onnxruntime-gpu instead of onnxruntime:
#   python_embeded\python.exe -m pip install onnxruntime-gpu

# -- Required for WD Tagger / SwinV2 ONNX inference --
onnxruntime>=1.16

# -- Required for JPEG EXIF tag writing --
piexif>=1.1.3

# -- Optional: DeepDanbooru (.h5 model) support --
# Omit if you do not use DeepDanbooru models.
# TensorFlow is large (~600MB) and conflicts with some CUDA setups.
# tensorflow>=2.10

# -- Required to load UNSLOTH_API_KEY from a local .env file (Unsloth backend, AI TOOL tab) --
python-dotenv>=1.0,<2.0

# -- Multi-layer PSD export (Gallery tab bulk action) --
# Pure-Python, ships prebuilt wheels (no C compiler required on Windows).
psd-tools>=1.10

# -- MP4 thumbnail generation (Gallery tab) --
# Uses PyAV (av) to decode the first frame. ComfyUI core already requires av>=17.0.0,
# so it is normally pre-installed in python_embeded and no action is needed here.
# If missing, mp4 thumbnail generation simply fails and falls back to a placeholder
# (Gallery itself keeps working). Not pinned here on purpose — see torch note above.
# av>=17.0.0

# -- Gallery Semantic Search: EmbeddingGemma 2 (optional) --
# Runs in Ollama (no extra Python packages). Install Ollama and download the model:
#   ollama pull embeddinggemma-2:740m      (or embeddinggemma-2:440m - same vectors, no audio encoder)
