dotfiles/modules/common/shell/bash/scripts/ocr.sh

60 lines
1.7 KiB
Bash
Raw Normal View History

2022-10-26 00:22:17 +00:00
#!/usr/bin/env bash
2021-05-16 23:09:38 +00:00
# Yoinked from https://github.com/JJGO/dotfiles
# Adapted from https://github.com/sdushantha/bin
set -x
TEXT_FILE="/tmp/ocr.txt"
IMAGE_FILE="/tmp/ocr.png"
function notify-send() {
2022-10-26 00:22:17 +00:00
/usr/bin/osascript -e "display notification \"$2\" with title \"OCR\""
2021-05-16 23:09:38 +00:00
}
PATH="/usr/local/bin/:$PATH"
# Take screenshot by selecting the area
2022-10-26 00:22:17 +00:00
/usr/sbin/screencapture -i "$IMAGE_FILE"
2021-05-16 23:09:38 +00:00
# Get the exit code of the previous command.
# So in this case, it is the screenshot command. If it did not exit with an
# exit code 0, then it means the user canceled the process of taking a
# screenshot by doing something like pressing the escape key
STATUS=$?
# If the user pressed the escape key or did something to terminate the proccess
# taking a screenshot, then just exit
[ $STATUS -ne 0 ] && exit 1
# Do the magic (∩^o^)⊃━☆゚.*・。゚
# Notice how I have removing the extension .txt from the file path. This is
# because tesseract adds .txt to the given file path anyways. So if we were to
# specify /tmp/ocr.txt as the file path, tesseract would out the text to
# /tmp/ocr.txt.txt
2024-03-26 15:17:54 +00:00
cd /tmp || {
echo "Failed to jump to directory."
exit 1
}
2021-05-16 23:09:38 +00:00
tesseract "$IMAGE_FILE" "${TEXT_FILE//\.txt/}"
# Check if the text was detected by checking number
# of lines in the file
2022-07-26 03:15:26 +00:00
LINES=$(wc -l <$TEXT_FILE)
2021-05-16 23:09:38 +00:00
if [ "$LINES" -eq 0 ]; then
notify-send "ocr" "no text was detected"
exit 1
fi
# Copy text to clipboard
# xclip -selection clip < "$TEXT_FILE"
2022-10-26 00:22:17 +00:00
/usr/bin/pbcopy <"$TEXT_FILE"
2021-05-16 23:09:38 +00:00
# Send a notification with the text that was grabbed using OCR
2022-07-26 03:15:26 +00:00
notify-send "ocr" "$(cat $TEXT_FILE)"
2021-05-16 23:09:38 +00:00
# Clean up
# "Always leave the area better than you found it"
# - My first grade teacher
rm "$TEXT_FILE"
rm "$IMAGE_FILE"