#!/bin/bash
set -ex

export PATH="/home/ubuntu/sam_audio_env/bin:$PATH"

echo "=== Step 1: Install ImageBind ==="
pip install git+https://github.com/facebookresearch/ImageBind.git 2>&1 || {
    echo "Direct pip install failed, trying clone..."
    cd /home/ubuntu
    git clone https://github.com/facebookresearch/ImageBind.git 2>/dev/null || true
    cd ImageBind
    pip install . 2>&1
}

echo "=== Step 2: Verify ImageBind ==="
python -c "import imagebind; print('ImageBind imported OK')"

echo "=== Step 3: Run SAM-Audio ==="
cd /home/ubuntu/sam_audio_test
python run_sam_audio.py 2>&1

echo "=== Step 4: Output files ==="
ls -lh /home/ubuntu/sam_audio_test/*.wav 2>&1

echo "=== DONE ==="