From 52d0805121d536b9fd20643b46a83c853acbed51 Mon Sep 17 00:00:00 2001
From: Alexandre <44178713+alexbelgium@users.noreply.github.com>
Date: Mon, 27 May 2024 11:11:40 +0200
Subject: [PATCH] Allow argument, or all files

---
 ...{spectrum_analysis.py => spectral_analysis.py} | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)
 rename birdnet-pi/rootfs/helpers/{spectrum_analysis.py => spectral_analysis.py} (80%)

diff --git a/birdnet-pi/rootfs/helpers/spectrum_analysis.py b/birdnet-pi/rootfs/helpers/spectral_analysis.py
similarity index 80%
rename from birdnet-pi/rootfs/helpers/spectrum_analysis.py
rename to birdnet-pi/rootfs/helpers/spectral_analysis.py
index 13fd03c87..1ce6d3a00 100644
--- a/birdnet-pi/rootfs/helpers/spectrum_analysis.py
+++ b/birdnet-pi/rootfs/helpers/spectral_analysis.py
@@ -3,19 +3,28 @@ import scipy.io.wavfile as wavfile
 import matplotlib.pyplot as plt
 import os
 import glob
+import sys  # Import the sys module
+
+from utils.helpers import get_settings
 
 # Dependencies /usr/bin/pip install numpy scipy matplotlib
 
 # Define the directory containing the WAV files
-input_directory = '/tmp/StreamData'
+conf = get_settings()
+input_directory = os.path.join(conf['RECS_DIR'], 'StreamData')
 output_directory = '/config'
 
 # Ensure the output directory exists
 if not os.path.exists(output_directory):
     os.makedirs(output_directory)
 
-# Get a list of all WAV files in the input directory
-wav_files = glob.glob(os.path.join(input_directory, '*.wav'))
+# Check if a command-line argument is provided
+if len(sys.argv) > 1:
+    # If an argument is provided, use it as the file to analyze
+    wav_files = [sys.argv[1]]
+else:
+    # If no argument is provided, analyze all WAV files in the directory
+    wav_files = glob.glob(os.path.join(input_directory, '*.wav'))
 
 # Process each file
 for file_path in wav_files: