#!/usr/bin/env bash

set -ex

script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$script_dir"

# see
# https://k2-fsa.github.io/sherpa/onnx/pretrained_models/online-paraformer/paraformer-models.html
repo_url=https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models
model=sherpa-onnx-streaming-paraformer-bilingual-zh-en
archive=$model.tar.bz2

model_dir="$script_dir/$model"
archive_path="$script_dir/$archive"

if [ ! -f "$model_dir/encoder.int8.onnx" ] ||
   [ ! -f "$model_dir/decoder.int8.onnx" ] ||
   [ ! -f "$model_dir/tokens.txt" ] ||
   [ ! -f "$model_dir/test_wavs/0.wav" ]; then
  if command -v curl >/dev/null 2>&1; then
    curl -L -o "$archive_path" "$repo_url/$archive"
  elif command -v wget >/dev/null 2>&1; then
    wget -O "$archive_path" "$repo_url/$archive"
  else
    echo "Error: Neither curl nor wget found. Please install one of them."
    exit 1
  fi

  tar xvf "$archive_path" -C "$script_dir"
  rm "$archive_path"
fi

cargo run --example streaming_paraformer -- \
  --encoder "$model_dir/encoder.int8.onnx" \
  --decoder "$model_dir/decoder.int8.onnx" \
  --tokens "$model_dir/tokens.txt" \
  --wav "$model_dir/test_wavs/0.wav" \
  --provider cpu