birdnet-analyzer 2.0.1__py3-none-any.whl → 2.1.1__py3-none-any.whl

birdnet_analyzer/utils.py

@@ -1,426 +1,425 @@
-"""Module containing common function."""
-
-import itertools
-import os
-import sys
-import traceback
-from pathlib import Path
-
-import birdnet_analyzer.config as cfg
-
-SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
-FROZEN = getattr(sys, "frozen", False) and hasattr(sys, "_MEIPASS")
-
-
-def runtime_error_handler(f: callable):
-    """Decorator to catch runtime errors and write them to the error log.
-
-    Args:
-        f: The function to be decorated.
-
-    Returns:
-        The decorated function.
-    """
-
-    def wrapper(*args, **kwargs):
-        try:
-            return f(*args, **kwargs)
-        except Exception as ex:
-            write_error_log(ex)
-            raise
-
-    return wrapper
-
-
-def batched(iterable, n, *, strict=False):
-    # TODO: Remove this function when Python 3.12 is the minimum version
-    # batched('ABCDEFG', 3) → ABC DEF G
-    if n < 1:
-        raise ValueError("n must be at least one")
-    iterator = iter(iterable)
-    while batch := tuple(itertools.islice(iterator, n)):
-        if strict and len(batch) != n:
-            raise ValueError("batched(): incomplete batch")
-        yield batch
-
-
-def spectrogram_from_file(path, fig_num=None, fig_size=None, offset=0, duration=None, fmin=None, fmax=None, speed=1.0):
-    """
-    Generate a spectrogram from an audio file.
-
-    Parameters:
-    path (str): The path to the audio file.
-
-    Returns:
-    matplotlib.figure.Figure: The generated spectrogram figure.
-    """
-    from birdnet_analyzer import audio
-
-    # s, sr = librosa.load(path, offset=offset, duration=duration)
-    s, sr = audio.open_audio_file(path, offset=offset, duration=duration, fmin=fmin, fmax=fmax, speed=speed)
-
-    return spectrogram_from_audio(s, sr, fig_num, fig_size)
-
-
-def spectrogram_from_audio(s, sr, fig_num=None, fig_size=None):
-    """
-    Generate a spectrogram from an audio signal.
-
-    Parameters:
-    s: The signal
-    sr: The sample rate
-
-    Returns:
-    matplotlib.figure.Figure: The generated spectrogram figure.
-    """
-    import librosa
-    import librosa.display
-    import matplotlib
-    import matplotlib.pyplot as plt
-    import numpy as np
-
-    matplotlib.use("agg")
-
-    if isinstance(fig_size, tuple):
-        f = plt.figure(fig_num, figsize=fig_size)
-    elif fig_size == "auto":
-        duration = librosa.get_duration(y=s, sr=sr)
-        width = min(12, max(3, duration / 10))
-        f = plt.figure(fig_num, figsize=(width, 3))
-    else:
-        f = plt.figure(fig_num)
-
-    f.clf()
-
-    ax = f.add_subplot(111)
-
-    ax.set_axis_off()
-    f.tight_layout(pad=0)
-
-    D = librosa.stft(s, n_fft=1024, hop_length=512)  # STFT of y
-    S_db = librosa.amplitude_to_db(np.abs(D), ref=np.max)
-
-    return librosa.display.specshow(S_db, ax=ax, n_fft=1024, hop_length=512).figure
-
-
-def collect_audio_files(path: str, max_files: int | None = None):
-    """Collects all audio files in the given directory.
-
-    Args:
-        path: The directory to be searched.
-
-    Returns:
-        A sorted list of all audio files in the directory.
-    """
-    # Get all files in directory with os.walk
-    files = []
-
-    for root, _, flist in os.walk(path):
-        for f in flist:
-            if not f.startswith(".") and f.rsplit(".", 1)[-1].lower() in cfg.ALLOWED_FILETYPES:
-                files.append(os.path.join(root, f))
-
-                if max_files and len(files) >= max_files:
-                    return sorted(files)
-
-    return sorted(files)
-
-
-def collect_all_files(path: str, filetypes: list[str], pattern: str = ""):
-    """Collects all files of the given filetypes in the given directory.
-
-    Args:
-        path: The directory to be searched.
-        filetypes: A list of filetypes to be collected.
-
-    Returns:
-        A sorted list of all files in the directory.
-    """
-
-    files = []
-
-    for root, _, flist in os.walk(path):
-        files.extend(
-            os.path.join(root, f)
-            for f in flist
-            if not f.startswith(".") and f.rsplit(".", 1)[-1].lower() in filetypes and (pattern in f or not pattern)
-        )
-
-    return sorted(files)
-
-
-def read_lines(path: str):
-    """Reads the lines into a list.
-
-    Opens the file and reads its contents into a list.
-    It is expected to have one line for each species or label.
-
-    Args:
-        path: Absolute path to the species file.
-
-    Returns:
-        A list of all species inside the file.
-    """
-    return Path(path).read_text(encoding="utf-8").splitlines() if path else []
-
-
-def list_subdirectories(path: str):
-    """Lists all directories inside a path.
-
-    Retrieves all the subdirectories in a given path without recursion.
-
-    Args:
-        path: Directory to be searched.
-
-    Returns:
-        A filter sequence containing the absolute paths to all directories.
-    """
-    return filter(lambda el: os.path.isdir(os.path.join(path, el)), os.listdir(path))
-
-
-def save_to_cache(path, x_train, y_train, x_test, y_test, labels):
-    """Saves training data to cache.
-
-    Args:
-        path: Path to the cache file.
-        x_train: Training samples.
-        y_train: Training labels.
-        x_test: Test samples.
-        y_test: Test labels.
-        labels: Labels.
-    """
-    import numpy as np
-
-    # Make directory if needed
-    directory = os.path.dirname(path)
-    if directory and not os.path.exists(directory):
-        os.makedirs(directory)
-
-    # Save cache file with training data, test data, labels and configuration
-    np.savez(
-        path,
-        x_train=x_train,
-        y_train=y_train,
-        x_test=x_test,
-        y_test=y_test,
-        labels=np.array(labels, dtype=object),
-        binary_classification=cfg.BINARY_CLASSIFICATION,
-        multi_label=cfg.MULTI_LABEL,
-        fmin=cfg.BANDPASS_FMIN,
-        fmax=cfg.BANDPASS_FMAX,
-        audio_speed=cfg.AUDIO_SPEED,
-        crop_mode=cfg.SAMPLE_CROP_MODE,
-        overlap=cfg.SIG_OVERLAP,
-    )
-
-
-def load_from_cache(path):
-    """Loads training data from cache.
-
-    Args:
-        path: Path to the cache file.
-
-    Returns:
-        A tuple of (x_train, y_train, labels, binary_classification, multi_label).
-    """
-    import numpy as np
-
-    # Load cache file
-    data = np.load(path, allow_pickle=True)
-
-    # Check if cache contains needed preprocessing parameters
-    if (
-        "fmin" in data
-        and "fmax" in data
-        and "audio_speed" in data
-        and "crop_mode" in data
-        and "overlap" in data
-        and (  # Check if preprocessing parameters match current settings
-            data["fmin"] != cfg.BANDPASS_FMIN
-            or data["fmax"] != cfg.BANDPASS_FMAX
-            or data["audio_speed"] != cfg.AUDIO_SPEED
-            or data["crop_mode"] != cfg.SAMPLE_CROP_MODE
-            or data["overlap"] != cfg.SIG_OVERLAP
-        )
-    ):
-        print("\t...WARNING: Cache preprocessing parameters don't match current settings!", flush=True)
-        print(f"\t   Cache: fmin={data['fmin']}, fmax={data['fmax']}, speed={data['audio_speed']}", flush=True)
-        print(f"\t   Cache: crop_mode={data['crop_mode']}, overlap={data['overlap']}", flush=True)
-        print(f"\t   Current: fmin={cfg.BANDPASS_FMIN}, fmax={cfg.BANDPASS_FMAX}, speed={cfg.AUDIO_SPEED}", flush=True)
-        print(f"\t   Current: crop_mode={cfg.SAMPLE_CROP_MODE}, overlap={cfg.SIG_OVERLAP}", flush=True)
-
-    # Extract and return data
-    x_train = data["x_train"]
-    y_train = data["y_train"]
-
-    # Handle test data which might not be in older cache files
-    x_test = data.get("x_test", np.array([]))
-    y_test = data.get("y_test", np.array([]))
-
-    labels = data["labels"]
-    binary_classification = bool(data.get("binary_classification", False))
-    multi_label = bool(data.get("multi_label", False))
-
-    return x_train, y_train, x_test, y_test, labels, binary_classification, multi_label
-
-
-def clear_error_log():
-    """Clears the error log file.
-
-    For debugging purposes.
-    """
-    if os.path.isfile(cfg.ERROR_LOG_FILE):
-        os.remove(cfg.ERROR_LOG_FILE)
-
-
-def write_error_log(ex: Exception):
-    """Writes an exception to the error log.
-
-    Formats the stacktrace and writes it in the error log file configured in the config.
-
-    Args:
-        ex: An exception that occurred.
-    """
-    import datetime
-
-    with open(cfg.ERROR_LOG_FILE, "a") as elog:
-        elog.write(
-            datetime.datetime.now().strftime("[%Y-%m-%d %H:%M:%S]")
-            + "\n"
-            + "".join(traceback.TracebackException.from_exception(ex).format())
-            + "\n"
-        )
-
-
-def img2base64(path):
-    import base64
-
-    with open(path, "rb") as img_file:
-        return base64.b64encode(img_file.read()).decode("utf-8")
-
-
-def save_params(file_path, headers, values):
-    """Saves the params used to train the custom classifier.
-
-    The hyperparams will be saved to disk in a file named 'model_params.csv'.
-
-    Args:
-        file_path: The path to the file.
-        headers: The headers of the csv file.
-        values: The values of the csv file.
-    """
-    import csv
-
-    with open(file_path, "w", newline="") as paramsfile:
-        paramswriter = csv.writer(paramsfile)
-        paramswriter.writerow(headers)
-        paramswriter.writerow(values)
-
-
-def save_result_file(result_path: str, out_string: str):
-    """Saves the result to a file.
-
-    Args:
-        result_path: The path to the result file.
-        out_string: The string to be written to the file.
-    """
-
-    # Make directory if it doesn't exist
-    os.makedirs(os.path.dirname(result_path), exist_ok=True)
-
-    # Write the result to the file
-    with open(result_path, "w", encoding="utf-8") as rfile:
-        rfile.write(out_string)
-
-
-def check_model_files():
-    checkpoint_dir = os.path.join(SCRIPT_DIR, "checkpoints", "V2.4")
-    required_files = [
-        "BirdNET_GLOBAL_6K_V2.4_Model/variables/variables.data-00000-of-00001",
-        "BirdNET_GLOBAL_6K_V2.4_Model/variables/variables.index",
-        "BirdNET_GLOBAL_6K_V2.4_Model/saved_model.pb",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard1of8.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard2of8.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard3of8.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard4of8.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard5of8.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard6of8.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard7of8.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard8of8.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/model.json",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard1of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard2of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard3of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard4of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard5of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard6of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard7of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard8of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard9of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard10of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard11of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard12of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard13of13.bin",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/model.json",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/labels.json",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/main.js",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/sample.wav",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/templates/index.html",
-        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/app.py",
-        "BirdNET_GLOBAL_6K_V2.4_Labels.txt",
-        "BirdNET_GLOBAL_6K_V2.4_MData_Model_V2_FP16.tflite",
-        "BirdNET_GLOBAL_6K_V2.4_Model_FP16.tflite",
-        "BirdNET_GLOBAL_6K_V2.4_Model_FP32.tflite",
-        "BirdNET_GLOBAL_6K_V2.4_Model_INT8.tflite",
-    ]
-
-    for file in required_files:
-        if not os.path.exists(os.path.join(checkpoint_dir, file)):
-            print(f"Missing {file}")
-
-            return False
-
-    print("Model found!")
-
-    return True
-
-
-def ensure_model_exists():
-    import zipfile
-
-    import requests
-    from tqdm import tqdm
-
-    if FROZEN or check_model_files():
-        return
-
-    checkpoint_dir = os.path.join(SCRIPT_DIR, "checkpoints")
-
-    os.makedirs(checkpoint_dir, exist_ok=True)
-
-    url = "https://tuc.cloud/index.php/s/3BsizWy5M7CtQ5w/download/V2.4.zip"
-    download_path = os.path.join(checkpoint_dir, "V2.4.zip")
-
-    response = requests.get(url, stream=True, timeout=30)
-    total_size = int(response.headers.get("content-length", 0))
-    block_size = 1024
-
-    with (
-        tqdm(total=total_size, unit="iB", unit_scale=True, desc="Downloading model") as tqdm_bar,
-        open(download_path, "wb") as file,
-    ):
-        for data in response.iter_content(block_size):
-            tqdm_bar.update(len(data))
-            file.write(data)
-
-    if response.status_code != 200 or (total_size not in (0, tqdm_bar.n)):
-        raise ValueError(f"Failed to download the file. Status code: {response.status_code}")
-
-    with zipfile.ZipFile(download_path, "r") as zip_ref:
-        zip_ref.extractall(os.path.dirname(download_path))
-
-    os.remove(download_path)
-
-
-if __name__ == "__main__":
-    ensure_model_exists()
+"""Module containing common function."""
+
+import itertools
+import os
+import sys
+import traceback
+from pathlib import Path
+
+import birdnet_analyzer.config as cfg
+
+SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
+FROZEN = getattr(sys, "frozen", False) and hasattr(sys, "_MEIPASS")
+
+
+def runtime_error_handler(f):
+    """Decorator to catch runtime errors and write them to the error log.
+
+    Args:
+        f: The function to be decorated.
+
+    Returns:
+        The decorated function.
+    """
+
+    def wrapper(*args, **kwargs):
+        try:
+            return f(*args, **kwargs)
+        except Exception as ex:
+            write_error_log(ex)
+            raise
+
+    return wrapper
+
+
+def batched(iterable, n, *, strict=False):
+    # TODO: Remove this function when Python 3.12 is the minimum version
+    # batched('ABCDEFG', 3) → ABC DEF G
+    if n < 1:
+        raise ValueError("n must be at least one")
+    iterator = iter(iterable)
+    while batch := tuple(itertools.islice(iterator, n)):
+        if strict and len(batch) != n:
+            raise ValueError("batched(): incomplete batch")
+        yield batch
+
+
+def spectrogram_from_file(path, fig_num=None, fig_size=None, offset=0, duration=None, fmin=None, fmax=None, speed=1.0):
+    """
+    Generate a spectrogram from an audio file.
+
+    Parameters:
+    path (str): The path to the audio file.
+
+    Returns:
+    matplotlib.figure.Figure: The generated spectrogram figure.
+    """
+    from birdnet_analyzer import audio
+
+    s, sr = audio.open_audio_file(path, offset=offset, duration=duration, fmin=fmin, fmax=fmax, speed=speed)
+
+    return spectrogram_from_audio(s, sr, fig_num=fig_num, fig_size=fig_size)
+
+
+def spectrogram_from_audio(s, sr, fig_num=None, fig_size=None):
+    """
+    Generate a spectrogram from an audio signal.
+
+    Parameters:
+    s: The signal
+    sr: The sample rate
+
+    Returns:
+    matplotlib.figure.Figure: The generated spectrogram figure.
+    """
+    import librosa
+    import librosa.display
+    import matplotlib
+    import matplotlib.pyplot as plt
+    import numpy as np
+
+    matplotlib.use("agg")
+
+    if isinstance(fig_size, tuple):
+        f = plt.figure(fig_num, figsize=fig_size)
+    elif fig_size == "auto":
+        duration = librosa.get_duration(y=s, sr=sr)
+        width = min(12, max(3, duration / 10))
+        f = plt.figure(fig_num, figsize=(width, 3))
+    else:
+        f = plt.figure(fig_num)
+
+    f.clf()
+
+    ax = f.add_subplot(111)
+
+    ax.set_axis_off()
+    f.tight_layout(pad=0)
+
+    D = librosa.stft(s, n_fft=1024, hop_length=512)  # STFT of y
+    S_db = librosa.amplitude_to_db(np.abs(D), ref=np.max)
+
+    return librosa.display.specshow(S_db, ax=ax, n_fft=1024, hop_length=512).figure
+
+
+def collect_audio_files(path: str, max_files: int | None = None):
+    """Collects all audio files in the given directory.
+
+    Args:
+        path: The directory to be searched.
+
+    Returns:
+        A sorted list of all audio files in the directory.
+    """
+    # Get all files in directory with os.walk
+    files = []
+
+    for root, _, flist in os.walk(path):
+        for f in flist:
+            if not f.startswith(".") and f.rsplit(".", 1)[-1].lower() in cfg.ALLOWED_FILETYPES:
+                files.append(os.path.join(root, f))
+
+                if max_files and len(files) >= max_files:
+                    return sorted(files)
+
+    return sorted(files)
+
+
+def collect_all_files(path: str, filetypes: list[str], pattern: str = ""):
+    """Collects all files of the given filetypes in the given directory.
+
+    Args:
+        path: The directory to be searched.
+        filetypes: A list of filetypes to be collected.
+
+    Returns:
+        A sorted list of all files in the directory.
+    """
+
+    files = []
+
+    for root, _, flist in os.walk(path):
+        files.extend(
+            os.path.join(root, f)
+            for f in flist
+            if not f.startswith(".") and f.rsplit(".", 1)[-1].lower() in filetypes and (pattern in f or not pattern)
+        )
+
+    return sorted(files)
+
+
+def read_lines(path: str | Path):
+    """Reads the lines into a list.
+
+    Opens the file and reads its contents into a list.
+    It is expected to have one line for each species or label.
+
+    Args:
+        path: Absolute path to the species file.
+
+    Returns:
+        A list of all species inside the file.
+    """
+    return Path(path).read_text(encoding="utf-8").splitlines() if path else []
+
+
+def list_subdirectories(path: str):
+    """Lists all directories inside a path.
+
+    Retrieves all the subdirectories in a given path without recursion.
+
+    Args:
+        path: Directory to be searched.
+
+    Returns:
+        A filter sequence containing the absolute paths to all directories.
+    """
+    return filter(lambda el: os.path.isdir(os.path.join(path, el)), os.listdir(path))
+
+
+def save_to_cache(path, x_train, y_train, x_test, y_test, labels):
+    """Saves training data to cache.
+
+    Args:
+        path: Path to the cache file.
+        x_train: Training samples.
+        y_train: Training labels.
+        x_test: Test samples.
+        y_test: Test labels.
+        labels: Labels.
+    """
+    import numpy as np
+
+    # Make directory if needed
+    directory = os.path.dirname(path)
+    if directory and not os.path.exists(directory):
+        os.makedirs(directory)
+
+    # Save cache file with training data, test data, labels and configuration
+    np.savez(
+        path,
+        x_train=x_train,
+        y_train=y_train,
+        x_test=x_test,
+        y_test=y_test,
+        labels=np.array(labels, dtype=object),
+        binary_classification=cfg.BINARY_CLASSIFICATION,
+        multi_label=cfg.MULTI_LABEL,
+        fmin=cfg.BANDPASS_FMIN,
+        fmax=cfg.BANDPASS_FMAX,
+        audio_speed=cfg.AUDIO_SPEED,
+        crop_mode=cfg.SAMPLE_CROP_MODE,
+        overlap=cfg.SIG_OVERLAP,
+    )
+
+
+def load_from_cache(path):
+    """Loads training data from cache.
+
+    Args:
+        path: Path to the cache file.
+
+    Returns:
+        A tuple of (x_train, y_train, labels, binary_classification, multi_label).
+    """
+    import numpy as np
+
+    # Load cache file
+    data = np.load(path, allow_pickle=True)
+
+    # Check if cache contains needed preprocessing parameters
+    if (
+        "fmin" in data
+        and "fmax" in data
+        and "audio_speed" in data
+        and "crop_mode" in data
+        and "overlap" in data
+        and (  # Check if preprocessing parameters match current settings
+            data["fmin"] != cfg.BANDPASS_FMIN
+            or data["fmax"] != cfg.BANDPASS_FMAX
+            or data["audio_speed"] != cfg.AUDIO_SPEED
+            or data["crop_mode"] != cfg.SAMPLE_CROP_MODE
+            or data["overlap"] != cfg.SIG_OVERLAP
+        )
+    ):
+        print("\t...WARNING: Cache preprocessing parameters don't match current settings!", flush=True)
+        print(f"\t   Cache: fmin={data['fmin']}, fmax={data['fmax']}, speed={data['audio_speed']}", flush=True)
+        print(f"\t   Cache: crop_mode={data['crop_mode']}, overlap={data['overlap']}", flush=True)
+        print(f"\t   Current: fmin={cfg.BANDPASS_FMIN}, fmax={cfg.BANDPASS_FMAX}, speed={cfg.AUDIO_SPEED}", flush=True)
+        print(f"\t   Current: crop_mode={cfg.SAMPLE_CROP_MODE}, overlap={cfg.SIG_OVERLAP}", flush=True)
+
+    # Extract and return data
+    x_train = data["x_train"]
+    y_train = data["y_train"]
+
+    # Handle test data which might not be in older cache files
+    x_test = data.get("x_test", np.array([]))
+    y_test = data.get("y_test", np.array([]))
+
+    labels = data["labels"]
+    binary_classification = bool(data.get("binary_classification", False))
+    multi_label = bool(data.get("multi_label", False))
+
+    return x_train, y_train, x_test, y_test, labels, binary_classification, multi_label
+
+
+def clear_error_log():
+    """Clears the error log file.
+
+    For debugging purposes.
+    """
+    if os.path.isfile(cfg.ERROR_LOG_FILE):
+        os.remove(cfg.ERROR_LOG_FILE)
+
+
+def write_error_log(ex: Exception):
+    """Writes an exception to the error log.
+
+    Formats the stacktrace and writes it in the error log file configured in the config.
+
+    Args:
+        ex: An exception that occurred.
+    """
+    import datetime
+
+    with open(cfg.ERROR_LOG_FILE, "a") as elog:
+        elog.write(
+            datetime.datetime.now().strftime("[%Y-%m-%d %H:%M:%S]")
+            + "\n"
+            + "".join(traceback.TracebackException.from_exception(ex).format())
+            + "\n"
+        )
+
+
+def img2base64(path):
+    import base64
+
+    with open(path, "rb") as img_file:
+        return base64.b64encode(img_file.read()).decode("utf-8")
+
+
+def save_params(file_path, headers, values):
+    """Saves the params used to train the custom classifier.
+
+    The hyperparams will be saved to disk in a file named 'model_params.csv'.
+
+    Args:
+        file_path: The path to the file.
+        headers: The headers of the csv file.
+        values: The values of the csv file.
+    """
+    import csv
+
+    with open(file_path, "w", newline="") as paramsfile:
+        paramswriter = csv.writer(paramsfile)
+        paramswriter.writerow(headers)
+        paramswriter.writerow(values)
+
+
+def save_result_file(result_path: str, out_string: str):
+    """Saves the result to a file.
+
+    Args:
+        result_path: The path to the result file.
+        out_string: The string to be written to the file.
+    """
+
+    # Make directory if it doesn't exist
+    os.makedirs(os.path.dirname(result_path), exist_ok=True)
+
+    # Write the result to the file
+    with open(result_path, "w", encoding="utf-8") as rfile:
+        rfile.write(out_string)
+
+
+def check_model_files():
+    checkpoint_dir = os.path.join(SCRIPT_DIR, "checkpoints", "V2.4")
+    required_files = [
+        "BirdNET_GLOBAL_6K_V2.4_Model/variables/variables.data-00000-of-00001",
+        "BirdNET_GLOBAL_6K_V2.4_Model/variables/variables.index",
+        "BirdNET_GLOBAL_6K_V2.4_Model/saved_model.pb",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard1of8.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard2of8.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard3of8.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard4of8.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard5of8.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard6of8.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard7of8.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/group1-shard8of8.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/mdata/model.json",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard1of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard2of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard3of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard4of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard5of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard6of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard7of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard8of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard9of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard10of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard11of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard12of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/group1-shard13of13.bin",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/model.json",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/model/labels.json",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/main.js",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/static/sample.wav",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/templates/index.html",
+        "BirdNET_GLOBAL_6K_V2.4_Model_TFJS/app.py",
+        "BirdNET_GLOBAL_6K_V2.4_Labels.txt",
+        "BirdNET_GLOBAL_6K_V2.4_MData_Model_V2_FP16.tflite",
+        "BirdNET_GLOBAL_6K_V2.4_Model_FP16.tflite",
+        "BirdNET_GLOBAL_6K_V2.4_Model_FP32.tflite",
+        "BirdNET_GLOBAL_6K_V2.4_Model_INT8.tflite",
+    ]
+
+    for file in required_files:
+        if not os.path.exists(os.path.join(checkpoint_dir, file)):
+            print(f"Missing {file}")
+
+            return False
+
+    print("Model found!")
+
+    return True
+
+
+def ensure_model_exists():
+    import zipfile
+
+    import requests
+    from tqdm import tqdm
+
+    if FROZEN or check_model_files():
+        return
+
+    checkpoint_dir = os.path.join(SCRIPT_DIR, "checkpoints")
+
+    os.makedirs(checkpoint_dir, exist_ok=True)
+
+    url = "https://tuc.cloud/index.php/s/3BsizWy5M7CtQ5w/download/V2.4.zip"
+    download_path = os.path.join(checkpoint_dir, "V2.4.zip")
+
+    response = requests.get(url, stream=True, timeout=30)
+    total_size = int(response.headers.get("content-length", 0))
+    block_size = 1024
+
+    with (
+        tqdm(total=total_size, unit="iB", unit_scale=True, desc="Downloading model") as tqdm_bar,
+        open(download_path, "wb") as file,
+    ):
+        for data in response.iter_content(block_size):
+            tqdm_bar.update(len(data))
+            file.write(data)
+
+    if response.status_code != 200 or (total_size not in (0, tqdm_bar.n)):
+        raise ValueError(f"Failed to download the file. Status code: {response.status_code}")
+
+    with zipfile.ZipFile(download_path, "r") as zip_ref:
+        zip_ref.extractall(os.path.dirname(download_path))
+
+    os.remove(download_path)
+
+
+if __name__ == "__main__":
+    ensure_model_exists()