setup.sh

#!/bin/bash

# Create necessary directories
mkdir -p downloads predictions vis hanyu_obj cookies

# Download ImageMagick if not exists
if [ ! -f "magick" ]; then
    wget https://imagemagick.org/archive/binaries/magick
    chmod +x magick
fi

# Install Python packages
pip install -r requirements.txt

# Install local packages
cd SkeletonDiffusion && pip install -e . && cd ..
cd src_joints2smpl_demo && pip install -e . && cd ..

# Set up headless rendering
bash ./SkeletonDiffusion_demo/setup_headless.bash

# Set environment variables
export PYOPENGL_PLATFORM=egl
export IMAGEMAGICK_BINARY=./magick