#!/bin/bash
# Fixed OpenPose Colab Loader
set -e

# Install dependencies
pip install huggingface_hub
apt-get install -y pigz

# Download from Hugging Face
REPO_ID="jrs-a/openpose-colab"  # Replace with your repo
FILENAME="openpose_colab_build.tar.gz"

# Download and get the actual path
DOWNLOAD_PATH=$(python -c "from huggingface_hub import hf_hub_download; print(hf_hub_download(repo_id='$REPO_ID', filename='$FILENAME'))")

# Extract
tar --use-compress-program=pigz -xvf "$DOWNLOAD_PATH"

# Install runtime dependencies
apt-get install -y libgoogle-glog-dev libboost-all-dev libatlas-base-dev libopencv-dev
pip install numpy opencv-python

echo "OpenPose restored successfully!"