samples/opentrack/setup.sh

#!/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 PyTorch (CPU version for compatibility)
echo ""
echo "🔥 Installing PyTorch..."
pip install --no-cache-dir \
    torch==2.5.1 \
    torchvision==0.20.1 \
    torchaudio==2.5.1 \
    --index-url https://download.pytorch.org/whl/cpu
echo "✓ PyTorch installed"

# Install OpenTrack requirements
echo ""
echo "📋 Installing OpenTrack requirements..."
if [ -f "$OPENTRACK_DIR/requirements.txt" ]; then
    pip install --no-cache-dir -r "$OPENTRACK_DIR/requirements.txt"
    echo "✓ OpenTrack requirements installed"
else
    echo "⚠️  Warning: requirements.txt not found in OpenTrack repo"
fi

# Install additional packages for video handling
echo ""
echo "🎬 Installing video handling packages..."
pip install --no-cache-dir imageio imageio-ffmpeg
echo "✓ Video packages installed"

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