diff --git a/.local/bin/video2srt b/.local/bin/video2srt index e50682f..f5a1e84 100755 --- a/.local/bin/video2srt +++ b/.local/bin/video2srt @@ -1,137 +1,41 @@ -#!/usr/bin/env bash -set -euo pipefail +#!/usr/bin/env python3 +"""Generate an English .srt next to each video using whisper.cpp.""" -WHISPER_DIR="${WHISPER_DIR:-$HOME/repos/whisper.cpp}" -WHISPER_MODEL="${WHISPER_MODEL:-$WHISPER_DIR/models/ggml-large-v3.bin}" -WHISPER_BIN="${WHISPER_BIN:-$WHISPER_DIR/build/bin/whisper-cli}" -VAD_MODEL="${VAD_MODEL:-$WHISPER_DIR/models/ggml-silero-v6.2.0.bin}" -TRANSLATE="${TRANSLATE:-1}" -SRC_LANG="${SRC_LANG:-auto}" -FORCE="${FORCE:-0}" -USE_VAD="${USE_VAD:-1}" -MAX_LEN="${MAX_LEN:-84}" -LINE_LEN="${LINE_LEN:-42}" -OUTPUT="" +import argparse +import os +import re +import shutil +import subprocess +import sys +import tempfile +from pathlib import Path -usage() { - cat < [