Initial commit, first public version.
This commit is contained in:
ben
BIN
tools/aichat
Executable file
BIN
tools/aichat
Executable file
Binary file not shown.
121
tools/stt.sh
Executable file
121
tools/stt.sh
Executable file
@@ -0,0 +1,121 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Function to print usage information
|
||||
usage() {
|
||||
echo "Usage: $0 [record|transcription] <options>"
|
||||
echo ""
|
||||
echo "Actions:"
|
||||
echo " record Record audio from a selected source"
|
||||
echo " transcription Transcribe audio from a .wav file"
|
||||
echo ""
|
||||
echo "Options for 'record':"
|
||||
echo " -s, --source Specify the audio source (required)"
|
||||
echo ""
|
||||
echo "Options for 'transcription':"
|
||||
echo " -f, --file Specify the audio file to transcribe (required)"
|
||||
echo " -l, --lang Specify the audio file language (default: en)"
|
||||
exit 1
|
||||
}
|
||||
|
||||
if [[ $# -eq 0 ]]; then
|
||||
usage
|
||||
fi
|
||||
|
||||
# Check for required environment variable
|
||||
if [[ -z "${LLM_API_KEY}" ]]; then
|
||||
echo "The environment variable LLM_API_KEY is not set."
|
||||
echo 'You can use the following command: export $(xargs < ../.env))'
|
||||
exit 1
|
||||
fi
|
||||
|
||||
ACTION=$1
|
||||
shift
|
||||
|
||||
host=${STT_API_HOST:-"http://localhost:8001"}
|
||||
LANG="en" # Default language
|
||||
|
||||
if [ "$ACTION" == "record" ]; then
|
||||
if [ "$#" -eq 0 ]; then
|
||||
echo "Error: Source is required for record action."
|
||||
echo "Available sources:"
|
||||
pactl list short sources | awk '{print $2}'
|
||||
exit 1
|
||||
fi
|
||||
|
||||
SOURCE=""
|
||||
while [[ "$#" -gt 0 ]]; do
|
||||
case $1 in
|
||||
-s | --source)
|
||||
SOURCE="$2"
|
||||
shift
|
||||
;;
|
||||
*)
|
||||
echo "Unknown parameter passed: $1"
|
||||
usage
|
||||
;;
|
||||
esac
|
||||
shift
|
||||
done
|
||||
|
||||
# Validate the provided source
|
||||
if ! pactl list short sources | awk '{print $2}' | grep -q "^$SOURCE$"; then
|
||||
echo "Error: Invalid audio source. Available sources:"
|
||||
pactl list short sources | awk '{print $2}'
|
||||
exit 1
|
||||
fi
|
||||
|
||||
timestamp=$(date +"%Y%m%d_%H%M%S")
|
||||
filename="record_${timestamp}.wav"
|
||||
echo "Start recording to ${filename} ; use CTRL+C to terminate."
|
||||
parec -d "${SOURCE}" --file-format=wav "${filename}"
|
||||
elif [ "$ACTION" == "transcription" ]; then
|
||||
if [ "$#" -eq 0 ]; then
|
||||
echo "Error: File is required for transcription action."
|
||||
usage
|
||||
fi
|
||||
|
||||
FILE=""
|
||||
while [[ "$#" -gt 0 ]]; do
|
||||
case $1 in
|
||||
-f | --file)
|
||||
FILE="$2"
|
||||
shift
|
||||
;;
|
||||
-l | --lang)
|
||||
LANG="$2"
|
||||
shift
|
||||
;;
|
||||
*)
|
||||
echo "Unknown parameter passed: $1"
|
||||
usage
|
||||
;;
|
||||
esac
|
||||
shift
|
||||
done
|
||||
|
||||
if [ -z "$FILE" ]; then
|
||||
echo "Error: File is required for transcription action."
|
||||
usage
|
||||
fi
|
||||
|
||||
# Check if the file exists
|
||||
if [ ! -f "$FILE" ]; then
|
||||
echo "Error: File '$FILE' does not exist."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Ensure that curl is available
|
||||
if ! command -v curl &>/dev/null; then
|
||||
echo "curl is required for transcription but could not be found on your system. Please install it."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Transcribe the specified file
|
||||
echo "Transcribing file $FILE, be patient"
|
||||
curl "${host}/v1/audio/transcriptions" -H "Authorization: Bearer ${LLM_API_KEY}" \
|
||||
-F "file=@${FILE}" \
|
||||
-F "stream=true" \
|
||||
-F "language=${LANG}"
|
||||
else
|
||||
usage
|
||||
fi
|
||||
115
tools/tts.sh
Executable file
115
tools/tts.sh
Executable file
@@ -0,0 +1,115 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Function to display usage information
|
||||
usage() {
|
||||
echo "Usage: $0 -l <lang> -v <voice> -s <speed> [--play] \"<text>\""
|
||||
echo " -l|--lang : Specify the language (french|english)"
|
||||
echo " -v|--voice : Specify the voice"
|
||||
echo " -s|--speed : Specify the speed (0.0 > 3.0, default is 1.0)"
|
||||
echo " --play : Play the generated audio file using ffplay"
|
||||
echo " <text> : The text to synthesize"
|
||||
exit 1
|
||||
}
|
||||
|
||||
# Function to check if a value is a valid float between 0 and 3.0
|
||||
is_valid_float() {
|
||||
local value=$1
|
||||
# Check if the value is a valid number
|
||||
if [[ $value =~ ^-?[0-9]+(\.[0-9]+)?$ ]]; then
|
||||
# Check if the value is between 0 and 3.0
|
||||
if (($(echo "$value >= 0" | bc -l))) && (($(echo "$value <= 3.0" | bc -l))); then
|
||||
return 0
|
||||
fi
|
||||
fi
|
||||
return 1
|
||||
}
|
||||
|
||||
# Check for required environment variable
|
||||
if [[ -z "${LLM_API_KEY}" ]]; then
|
||||
echo "The environment variable LLM_API_KEY is not set."
|
||||
echo 'You can use the following command: export $(xargs < ../.env))'
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Default values
|
||||
speed=1.0
|
||||
host=${TTS_API_HOST:-"http://localhost:8000"}
|
||||
play_audio=false
|
||||
|
||||
# Parse command line arguments
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
-l | --lang)
|
||||
lang="$2"
|
||||
shift 2
|
||||
;;
|
||||
-v | --voice)
|
||||
voice="$2"
|
||||
shift 2
|
||||
;;
|
||||
-s | --speed)
|
||||
speed="$2"
|
||||
shift 2
|
||||
;;
|
||||
--play)
|
||||
play_audio=true
|
||||
shift 1
|
||||
;;
|
||||
-h | --help)
|
||||
usage
|
||||
;;
|
||||
-* | --*)
|
||||
echo "Unknown option $1"
|
||||
usage
|
||||
;;
|
||||
*)
|
||||
break
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
# Optionally grab the text after the options
|
||||
if [[ $# -gt 0 ]]; then
|
||||
text="$*"
|
||||
else
|
||||
echo "Error: Text to synthesize is required."
|
||||
usage
|
||||
fi
|
||||
|
||||
# Generate a timestamp
|
||||
timestamp=$(date +"%Y%m%d_%H%M%S")
|
||||
|
||||
# Construct the filename with the current date and time
|
||||
filename="speech_${timestamp}.wav"
|
||||
|
||||
# Validate language and voice options
|
||||
if [[ -z "$lang" || -z "$voice" ]]; then
|
||||
echo "Error: Language (-l) and voice (-v) options are required."
|
||||
usage
|
||||
fi
|
||||
|
||||
# Check if the speed is valid
|
||||
if ! is_valid_float "$speed"; then
|
||||
echo "Error: Speed must be a float between 0.0 and 3.0."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Fetch the audio file from the API
|
||||
http_status_code=$(curl -s "${host}/v1/audio/speech" -o "${filename}" -w "%{http_code}" -H "Authorization: Bearer ${LLM_API_KEY}" -H "Content-Type: application/json" -d "{\"model\": \"tts-1\",\"input\": \"${text}\",\"voice\": \"${voice}\",\"response_format\": \"wav\",\"speed\": ${speed}}")
|
||||
|
||||
# Check the response code for successful HTTP request
|
||||
if [[ "$http_status_code" -ne 200 ]]; then
|
||||
echo "Error: Failed to fetch audio file. Received HTTP status code: $http_status_code"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Optionally play the generated WAV file with ffplay
|
||||
if [ "$play_audio" = true ]; then
|
||||
if ! command -v ffplay &>/dev/null; then
|
||||
echo "Error: ffplay is not installed. Please install mpv to play audio files."
|
||||
exit 1
|
||||
fi
|
||||
ffplay ${filename} -nodisp -nostats -hide_banner -autoexit -v quiet
|
||||
fi
|
||||
|
||||
echo "Audio file '$filename' generated successfully."
|
||||
Reference in New Issue
Block a user