#!/usr/bin/env bash
set -euo pipefail

BASE="${BASE:-/data/home/user-test/SHC/COVT}"
TRAIN_DIR="$BASE/upstream/CoVT/train"
LOG_DIR="$BASE/logs"
mkdir -p "$LOG_DIR"

cd "$TRAIN_DIR"

echo "Installing CoVT training dependencies into the user Python scope only."
echo "No system packages, drivers, CUDA, or hardware settings are changed."
echo "Log: $LOG_DIR/prepare_user_env.log"

{
  date
  echo "python=$(command -v python3)"
  python3 --version
  python3 -m pip install --user --upgrade pip
  python3 -m pip install --user -r requirements.txt
  python3 -m pip install --user git+https://github.com/facebookresearch/segment-anything.git
  python3 -m pip install --user -r src/anchors/DepthAnything/requirements.txt
  python3 -m pip install --user "deepspeed>=0.15"
  python3 -m pip install --user "xformers==0.0.29.post1"
  python3 - <<'PY'
mods = ["torch", "transformers", "peft", "deepspeed", "safetensors", "qwen_vl_utils"]
for name in mods:
    mod = __import__(name)
    print(f"{name}: {getattr(mod, '__version__', '')}")
PY
} 2>&1 | tee "$LOG_DIR/prepare_user_env.log"