# Search-UI RunPod worker environment
# LEGACY REFERENCE ONLY.
# Supported RunPod setup uses `bash deploy/runpod/setup-pod.sh`, which writes
# `/workspace/runpod.env` for the active deployment flow.

# Main data root used by setup-pod.sh
export SEARCH_UI_DATA_ROOT=/workspace/data
export SEARCH_UI_WORKER_VENV=/workspace/venv
export SEARCH_UI_DB_DIR=/tmp/search-ui-db

# Optional explicit overrides
# export SEARCH_UI_DB_DIR=/workspace/data/db
# export SEARCH_UI_JSON_DIR=/workspace/data/json
# export SEARCH_UI_SUBTITLES_DIR=/workspace/data/subtitles
# export SEARCH_UI_VIDEOS_DIR=/workspace/data/videos
# export SEARCH_UI_LOGS_DIR=/workspace/data/logs
# export SEARCH_UI_CHECKPOINTS_DIR=/workspace/data/checkpoints
# export SEARCH_UI_PUBLICATIONS_DIR=/workspace/data/publications

# Model/cache locations
export HF_HOME=/workspace/data/cache/huggingface
export TRANSFORMERS_CACHE=/workspace/data/cache/huggingface/transformers
export SENTENCE_TRANSFORMERS_HOME=/workspace/data/cache/sentence-transformers
export TORCH_HOME=/workspace/data/cache/torch
export XDG_CACHE_HOME=/workspace/data/cache

# Run DeepFace/TensorFlow on CPU by default. setup-pod.sh can now install
# TensorFlow's pip GPU runtime and apply TensorFlow's documented venv symlink
# fix when RUNPOD_ENABLE_DEEPFACE_GPU=1, but keep CPU as the default until the
# guarded probe proves the runtime is healthy on that pod.
export SEARCH_UI_DEEPFACE_FORCE_CPU=1
# Optional guarded DeepFace GPU mode. Only enable after validating the pod's
# TensorFlow runtime can actually see a usable GPU without CuDNN/runtime errors.
# export SEARCH_UI_DEEPFACE_ALLOW_GPU=1
# Optional setup-time probe to flip DeepFace to guarded GPU mode automatically.
# export RUNPOD_ENABLE_DEEPFACE_GPU=1
export SEARCH_UI_SKIP_EMBEDDED_SUBTITLES=1
export SEARCH_UI_PURGE_LARGE_THUMBNAILS=1
export SEARCH_UI_SIGLIP_BATCH_SIZE=8
export SEARCH_UI_STAGE_VIDEO_PROCESSING=1
export SEARCH_UI_PROCESSING_STAGING_ROOT=/tmp/search-ui-processing

# Worker defaults
export RUNPOD_WORKER_LANGUAGE=E
export RUNPOD_WORKER_LABEL=720p
export RUNPOD_DOWNLOAD_CONCURRENCY=6
export RUNPOD_PROCESS_SHARDS=2
export RUNPOD_PROCESS_LIMIT=0
# Optional multi-GPU shard pinning. Example: "0,1,2,3"
# export RUNPOD_GPU_IDS=0,1
# export RUNPOD_WORKER_GPU_IDS=0,1
# Optional deterministic backlog partitioning across multiple pods.
# Example for pod 1 of 2:
# export RUNPOD_PARTITION_TOTAL=2
# export RUNPOD_PARTITION_INDEX=0
export RUNPOD_WORKER_REPORT_DIR=/workspace/data/reports
export RUNPOD_DELETE_VIDEO_AFTER_PROCESSING=1