#!/usr/bin/zsh
WHISPER_ARGS=(${(z)WHISPER_FLAGS})
MAX_LINE_WIDTH="${MAX_LINE_WIDTH:-50}"
MODEL="${MODEL:-medium}"
LANGUAGE="${LANGUAGE:-en}"
INPUT="$1"
BASE=$(basename "$INPUT" .mkv)
DIR=${2:-/home/sacha/sync/recordings/processed}
ffmpeg -y -i "$1" -map 0:a:0 -ac 1 "$DIR/$BASE-combined.wav"
ffmpeg -y -i "$1" -map 0:a:1 -ac 1 "$DIR/$BASE-sacha.wav"
ffmpeg -y -i "$1" -map 0:a:2 -ac 1 "$DIR/$BASE-other.wav"
normalize $DIR/$BASE-sacha.wav
normalize $DIR/$BASE-other.wav
cd $DIR
for FILE in "$BASE-sacha.wav" "$BASE-other.wav"; do
    echo "$FILE"
   	~/vendor/whisperx/.venv/bin/whisperx --model "$MODEL" --language "$LANGUAGE" --align_model WAV2VEC2_ASR_LARGE_LV60K_960H --compute_type int8 --print_progress True --max_line_width $MAX_LINE_WIDTH --segment_resolution chunk --max_line_count 1  "$FILE" $WHISPER_FLAGS
done