# Simplest SkyPilot test - just validates provisioning and basic setup
name: test-simple

resources:
  use_spot: true
  cpus: 2+  # CPU only to avoid GPU quota issues
  memory: 8+
  disk_size: 30

setup: |
  echo "=================================================="
  echo "🧪 SIMPLE SKYPILOT TEST"
  echo "=================================================="
  echo ""

  # Machine info
  echo "📊 Machine Info:"
  echo "  Hostname: $(hostname)"
  echo "  CPU cores: $(nproc)"
  echo "  Memory: $(free -h | grep Mem | awk '{print $2}')"
  echo ""

  # Install Python deps
  echo "📦 Installing dependencies..."
  pip install -q numpy soundfile librosa
  echo "  ✓ Installed: numpy, soundfile, librosa"
  echo ""

  # Clone repo
  echo "📥 Cloning repository..."
  git clone -q https://huggingface.co/marcosremar2/ensemble-tts-annotation
  echo "  ✓ Repository cloned"
  echo ""

  echo "✅ Setup complete!"

run: |
  cd ensemble-tts-annotation

  echo ""
  echo "=================================================="
  echo "🧪 RUNNING TESTS"
  echo "=================================================="
  echo ""

  # Check files
  echo "📁 Checking key files..."
  for file in README.md QUICK_START_SKYPILOT.md ensemble_tts/__init__.py; do
    if [ -f "$file" ]; then
      echo "  ✓ $file"
    else
      echo "  ❌ $file MISSING"
    fi
  done
  echo ""

  # Test Python imports
  echo "🐍 Testing Python imports..."
  python3 -c "import numpy; import soundfile; import librosa; print('  ✓ All imports work')"
  echo ""

  # Generate 1 test audio
  echo "🎵 Generating test audio..."
  python3 scripts/data/create_synthetic_test_data.py --output test_tmp --samples 1
  echo ""

  # Count files
  AUDIO_COUNT=$(find test_tmp -name "*.wav" | wc -l)
  echo "📊 Created $AUDIO_COUNT audio files"

  if [ "$AUDIO_COUNT" -eq 7 ]; then
    echo "  ✓ Expected 7 files - CORRECT"
  else
    echo "  ❌ Expected 7 but got $AUDIO_COUNT"
  fi
  echo ""

  # Cleanup
  rm -rf test_tmp

  echo "=================================================="
  echo "✅ ALL TESTS PASSED!"
  echo "=================================================="
  echo ""
  echo "🎉 SkyPilot successfully:"
  echo "  ✓ Provisioned machine"
  echo "  ✓ Installed dependencies"
  echo "  ✓ Cloned repository"
  echo "  ✓ Generated synthetic audio"
  echo ""
  echo "💰 Cost: ~\$0.05 for this test"
  echo ""