summaryrefslogtreecommitdiffstats
path: root/roles/caption/templates
diff options
context:
space:
mode:
Diffstat (limited to '')
-rwxr-xr-xroles/caption/templates/caption.sh18
1 files changed, 18 insertions, 0 deletions
diff --git a/roles/caption/templates/caption.sh b/roles/caption/templates/caption.sh
new file mode 100755
index 0000000..9600a3c
--- /dev/null
+++ b/roles/caption/templates/caption.sh
@@ -0,0 +1,18 @@
+#!/bin/bash
+# {{ ansible_managed }}
+FILE="$1"
+MODEL="${2:small}"
+AUDIO=$(basename "$FILE" | sed s/\\.[a-z][a-z][a-z][a-z]?$//).ogg
+if [[ ! -f $AUDIO ]]; then
+ if [[ "$FILE" == *webm ]]; then
+ ffmpeg -y -i "$FILE" -acodec copy -vn $AUDIO
+ else
+ ffmpeg -y -i "$FILE" -acodec libvorbis -vn $AUDIO
+ fi
+fi
+date > $AUDIO-$MODEL.log
+time whisper $AUDIO --model $MODEL --threads 12 >> $AUDIO-$MODEL.log
+for EXT in vtt txt srt; do
+ mv $AUDIO.$EXT $(basename -s .webm.$EXT $AUDIO.$EXT)
+done
+date >> $AUDIO-$MODEL.log