diff options
Diffstat (limited to 'dot_local')
| -rw-r--r-- | dot_local/bin/executable_dictate | 87 | ||||
| -rw-r--r-- | dot_local/bin/executable_ocr | 39 |
2 files changed, 126 insertions, 0 deletions
diff --git a/dot_local/bin/executable_dictate b/dot_local/bin/executable_dictate new file mode 100644 index 0000000..2e8a361 --- /dev/null +++ b/dot_local/bin/executable_dictate @@ -0,0 +1,87 @@ +#!/usr/bin/env sh +# Push-to-talk dictation. Toggle: 1st invocation starts recording; +# 2nd stops, transcribes via whisper.cpp, types the result into the focused +# window with wtype, and copies it to the clipboard. +# +# Pacman: pipewire-pulse (parecord), wtype, wl-clipboard, libnotify +# AUR: whisper.cpp, whisper.cpp-model-large-v3-turbo-q5_0 +# +# Override via env: +# WHISPER_MODEL path to a ggml-*.bin model +# WHISPER_LANG language code or 'auto' (default: auto) + +set -eu + +state_dir="${XDG_RUNTIME_DIR:-/tmp}/dictate" +pid_file="$state_dir/pid" +wav_file="$state_dir/audio.wav" +log_file="$state_dir/whisper.log" +default_model='/usr/share/whisper.cpp-model-large-v3-turbo-q5_0/ggml-large-v3-turbo-q5_0.bin' +model="${WHISPER_MODEL:-$default_model}" +lang="${WHISPER_LANG:-auto}" + +mkdir -p "$state_dir" + +is_recording() { + [ -r "$pid_file" ] && kill -0 "$(cat "$pid_file")" 2>/dev/null +} + +start_recording() { + if ! command -v whisper-cli >/dev/null; then + notify-send -u critical "๐๏ธ dictate" \ + "whisper-cli not found. Install whisper.cpp (AUR)." + exit 1 + fi + if [ ! -r "$model" ]; then + notify-send -u critical "๐๏ธ dictate" \ + "Model missing: $model. Install whisper.cpp-model-large-v3-turbo-q5_0 (AUR)." + exit 1 + fi + rm -f "$wav_file" + parecord --format=s16le --rate=16000 --channels=1 "$wav_file" \ + >"$log_file" 2>&1 & + echo "$!" >"$pid_file" + notify-send -t 1500 "๐๏ธ Recordingโฆ" "Press the bind again to stop." +} + +stop_and_transcribe() { + pid="$(cat "$pid_file")" + rm -f "$pid_file" + kill -TERM "$pid" 2>/dev/null || true + # Give parecord up to 2s to flush the WAV header. + i=0 + while kill -0 "$pid" 2>/dev/null && [ "$i" -lt 20 ]; do + sleep 0.1 + i=$((i + 1)) + done + kill -KILL "$pid" 2>/dev/null || true + + if [ ! -s "$wav_file" ]; then + notify-send -u low "๐๏ธ dictate" "No audio captured." + exit 1 + fi + + notify-send -t 1500 "๐๏ธ Transcribingโฆ" + + text="$( + whisper-cli -m "$model" -f "$wav_file" \ + -l "$lang" -nt -np -t "$(nproc)" 2>"$log_file" \ + | tr -s '[:space:]' ' ' \ + | sed -e 's/^ //; s/ $//' + )" + + if [ -z "$text" ]; then + notify-send -u low "๐๏ธ dictate" "Empty transcription. See $log_file." + exit 1 + fi + + printf '%s' "$text" | wl-copy + wtype -- "$text" + notify-send -t 2500 "๐๏ธ Dictated" "$text" +} + +if is_recording; then + stop_and_transcribe +else + start_recording +fi diff --git a/dot_local/bin/executable_ocr b/dot_local/bin/executable_ocr new file mode 100644 index 0000000..6f6191f --- /dev/null +++ b/dot_local/bin/executable_ocr @@ -0,0 +1,39 @@ +#!/usr/bin/env sh +# OCR a screen region (default) or an image file โ clipboard. +# +# Usage: +# ocr # interactive: select a region with slurp, OCR it +# ocr <file> # OCR an image file +# +# Requires: tesseract (+tesseract-data-eng, tesseract-data-por), +# grim, slurp, wl-clipboard, libnotify. +# +# Override languages via TESSERACT_LANG (e.g. TESSERACT_LANG=eng). + +set -eu + +lang="${TESSERACT_LANG:-eng+por}" + +if [ "${1:-}" ]; then + [ -r "$1" ] || { notify-send -u critical "๐ OCR" "Cannot read: $1"; exit 1; } + text="$(tesseract "$1" - -l "$lang" 2>/dev/null || true)" +else + region="$(slurp 2>/dev/null)" || exit 0 + text="$(grim -g "$region" - | tesseract - - -l "$lang" 2>/dev/null || true)" +fi + +# Trim trailing whitespace per line; collapse runs of blank lines; drop +# leading blanks. +text="$(printf '%s\n' "$text" | awk ' + { sub(/[[:space:]]+$/, "") } + NF { print; blank = 0; next } + !blank { print; blank = 1 } +' | sed -e '/./,$!d')" + +if [ -z "$text" ]; then + notify-send -u low "๐ OCR" "No text detected." + exit 1 +fi + +printf '%s' "$text" | wl-copy +notify-send -t 3000 "๐ OCR copied" "$(printf '%s' "$text" | head -c 200)" |
