captionbot-clients/captionbot.sh

62 lines
1.9 KiB
Bash
Raw Normal View History

2016-04-15 14:14:42 +02:00
#!/bin/bash
2016-04-15 15:14:47 +02:00
# Bash captionbot client
# Created by Daniil Gentili
# Licensed under GPLv3
2016-04-15 14:14:42 +02:00
2016-04-15 15:14:47 +02:00
help() {
echo "captionbot.ai api client.
This script will try to recognize the content of any image you give him using Microsoft's captionbot.ai website api.
2016-04-15 14:14:42 +02:00
2016-04-15 15:14:47 +02:00
Usage: $0 file
or
Usage: $0 url
"
exit 1
}
[ $* = "" ] && help
echo "Connecting to captionbot.ai..."
conversationId=$(curl -s https://www.captionbot.ai/api/init)
2016-04-15 15:54:18 +02:00
if [ "$1" = "bashbotmode" ]; then
echo "Send me the image you want me to recognize."
read input
input=$(echo "$input" | sed 's/\s*//g')
else
input="$1"
fi
if [ ! -f "$input" -o "$1" = "bashbotmode" ]; then
url="\"$input\""
else echo "Uploading image..."
url=$(curl -s https://www.captionbot.ai/api/upload -F "image1=@$input")
fi
2016-04-15 15:14:47 +02:00
echo "Analyzing image..."
result=$(curl -s https://www.captionbot.ai/api/message -H "Content-Type: application/json; charset=utf-8" -X POST -d '{"userMessage":'$url', "conversationId":'$conversationId'}' | sed 's/\\"/"/g;s/^"//g;s/"$//g' | ./JSON.sh -s)
watermark=$(echo "$result" | sed '/\["WaterMark"\]/!d;s/\["WaterMark"\]\t//g')
message=$(echo "$result" | sed '/\[".*Message"\]/!d;s/\[".*Message"\]\t//g;s/^"//g;s/"$//g')
echo $message
echo
2016-04-15 15:54:18 +02:00
2016-04-15 15:14:47 +02:00
until [ $s -gt 0 -a $s -le 5 ] 2>/dev/null ;do
2016-04-15 15:54:18 +02:00
if [ "$1" = "bashbotmode" ]; then
echo 'How did I do? mykeyboardstartshere "1 star" "2 stars" "3 stars" "4 stars" "5 stars"'
read s
s=$(echo ${s//[^0-9]/})
else
read -p "How did I do (rate 1 to 5)? " s
fi
[ $s -gt 0 -a $s -le 5 ] 2>/dev/null || echo "You didn't input a valid number. Please try again!"
2016-04-15 15:14:47 +02:00
done
result=$(curl -s https://www.captionbot.ai/api/message -H "Content-Type: application/json; charset=utf-8" -X POST -d '{"userMessage":"'$s'", "conversationId":'$conversationId', "waterMark":'$watermark'}' | sed 's/\\"/"/g;s/^"//g;s/"$//g' | ./JSON.sh -s)
message=$(echo "$result" | sed '/\[".*Message"\]/!d;s/\[".*Message"\]\t//g;s/^"//g;s/"$//g')
echo $message
2016-04-15 15:54:18 +02:00
sleep 1
2016-04-15 15:14:47 +02:00
echo "Thanks for having used captionbot.ai!"