#!/bin/bash

# Install torchcodec and run optimized processing

echo "🔧 Installing torchcodec and running optimized processing..."

# Activate virtual environment if it exists
if [ -d ".venv" ]; then
    echo "🔧 Activating virtual environment..."
    source .venv/bin/activate
fi

# Install torchcodec first
echo "📦 Installing torchcodec..."
uv pip install torchcodec>=0.1.0

# Also install other audio dependencies
echo "📦 Installing additional audio dependencies..."
uv pip install librosa>=0.10.0
uv pip install ffmpeg-python>=0.2.0

# Test the installation
echo "🧪 Testing torchcodec installation..."
python -c "import torchcodec; print('torchcodec installed successfully')"

if [ $? -ne 0 ]; then
    echo "❌ Failed to install torchcodec. Trying alternative installation..."
    uv pip install --force-reinstall torchcodec>=0.1.0
fi

# Run the optimized processing
echo "🚀 Running optimized processing..."
python batch_confirm_hf_optimized.py