From f056485e6fecf19c788b226ca3f710881da2b9ef Mon Sep 17 00:00:00 2001
From: Alexandre <44178713+alexbelgium@users.noreply.github.com>
Date: Mon, 14 Oct 2024 20:14:38 +0200
Subject: [PATCH] Update 71-newfeatures.sh

---
 .../rootfs/etc/cont-init.d/71-newfeatures.sh      | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/birdnet-pi/rootfs/etc/cont-init.d/71-newfeatures.sh b/birdnet-pi/rootfs/etc/cont-init.d/71-newfeatures.sh
index 02fa386b7..90f0cc416 100755
--- a/birdnet-pi/rootfs/etc/cont-init.d/71-newfeatures.sh
+++ b/birdnet-pi/rootfs/etc/cont-init.d/71-newfeatures.sh
@@ -8,6 +8,21 @@ set -e
 
 bashio::log.info "Adding optional features"
 
+if bashio::config.true "BAT_MODEL_ENABLED"; then
+    bashio::log.warning "... adding bats model"
+    sed -i '/thread_queue_size/a if [[ "$MODEL" == *[Bb][Aa][Tt]* ]]; then python3 "$HOME/BirdNET-Pi/scripts/bat_wav_translate.py" "${RECS_DIR}/StreamData/$(date "+%F")-birdnet-RTSP_${RTSP_STREAMS_STARTED_COUNT}-$(date "+%H:%M:%S").wav"; fi' "$HOME"/BirdNET-Pi/scripts/birdnet_recording.sh
+    echo "... installing wav translator"
+    cp /helpers/Bat_Model_BE_v1.py "$HOME"/BirdNET-Pi/scripts/bat_wav/translate.py
+    chmod +x "$HOME"/BirdNET-Pi/scripts/bat_wav_translate.py
+    echo "... moving model"
+    cp /helpers/Bat_Model_BE_v1.tflite "$HOME"/BirdNET-Pi/model/
+    chown pi:pi "$HOME"/BirdNET-Pi/model/*
+    echo "... installing labels"
+    mv /helpers/Bat_Model_BE_v1.txt "$HOME"/BirdNET-Pi/scripts/labels.txt
+    echo "... changing model to bats"
+    sed -i "/MODEL/c\MODEL=\"Bat_Model_BE_v1\"" /config/birdnet.conf
+fi
+
 # Denoiser
 #if bashio::config.true "DENOISER_ANALYSIS_ENABLED"; then
 #    sed -i "s|ar 48000|ar 48000 -af \"arnndn=m=sample.rnnn\"|g" "$HOME"/BirdNET-Pi/scripts/birdnet_recording.sh