#!/bin/bash
# OpenTrack build-time setup script
# This script clones OpenTrack and installs its dependencies at Docker build time

set -e  # Exit on error

echo "🤖 Setting up OpenTrack at build time..."
echo "=========================================="

# Clone OpenTrack repository
OPENTRACK_DIR="$HOME/OpenTrack"
if [ ! -d "$OPENTRACK_DIR" ]; then
    echo "📦 Cloning OpenTrack repository..."
    git clone https://github.com/GalaxyGeneralRobotics/OpenTrack.git "$OPENTRACK_DIR"
    echo "✓ Repository cloned to $OPENTRACK_DIR"
else
    echo "✓ OpenTrack repository already exists"
fi

# Install OpenTrack-specific requirements (most are already in main requirements.txt)
# Main requirements.txt already has: jax[cuda12], mujoco, brax, playground, mediapy
echo ""
echo "📋 Installing additional OpenTrack dependencies..."
pip install --no-cache-dir \
    tyro \
    opencv-python \
    wandb \
    jaxopt \
    flax \
    absl-py \
    joblib \
    imageio[ffmpeg]
echo "✓ OpenTrack dependencies installed"

echo ""
echo "=========================================="
echo "✅ OpenTrack build-time setup complete!"
echo ""