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

birdnet_analyzer/gui/train.py

@@ -9,6 +9,7 @@ import birdnet_analyzer.config as cfg
 import birdnet_analyzer.gui.localization as loc
 import birdnet_analyzer.gui.utils as gu
 from birdnet_analyzer import utils
+from birdnet_analyzer.gui.settings import APPDIR
 
 _GRID_MAX_HEIGHT = 240
 
@@ -198,16 +199,14 @@ def start_training(
 
     def trial_progression(trial):
         if progress is not None:
-            progress(
-                (trial, autotune_trials), total=autotune_trials, unit="trials", desc=loc.localize("progress-autotune")
-            )
+            progress((trial, autotune_trials), total=autotune_trials, unit="trials", desc=loc.localize("progress-autotune"))
 
     try:
         history_result = train_model(
             on_epoch_end=epoch_progression,
             on_trial_result=trial_progression,
             on_data_load_end=data_load_progression,
-            autotune_directory=gu.APPDIR if utils.FROZEN else "autotune",
+            autotune_directory=APPDIR if utils.FROZEN else "autotune",
         )
 
         # Unpack history and metrics
@@ -315,9 +314,7 @@ def build_train_tab():
                 info=loc.localize("training-tab-cache-mode-radio-info"),
             )
             with gr.Column(visible=False) as new_cache_file_row:
-                select_cache_file_directory_btn = gr.Button(
-                    loc.localize("training-tab-cache-select-directory-button-label")
-                )
+                select_cache_file_directory_btn = gr.Button(loc.localize("training-tab-cache-select-directory-button-label"))
 
                 with gr.Column():
                     cache_file_name = gr.Textbox(
@@ -426,9 +423,7 @@ def build_train_tab():
 
             def on_crop_select(new_crop_mode):
                 # Make overlap slider visible for both "segments" and "smart" crop modes
-                return gr.Number(
-                    visible=new_crop_mode in ["segments", "smart"], interactive=new_crop_mode in ["segments", "smart"]
-                )
+                return gr.Number(visible=new_crop_mode in ["segments", "smart"], interactive=new_crop_mode in ["segments", "smart"])
 
             crop_mode.change(on_crop_select, inputs=crop_mode, outputs=crop_overlap)
 
@@ -576,9 +571,7 @@ def build_train_tab():
         def on_focal_loss_change(value):
             return gr.Row(visible=value)
 
-        use_focal_loss.change(
-            on_focal_loss_change, inputs=use_focal_loss, outputs=focal_loss_params, show_progress=False
-        )
+        use_focal_loss.change(on_focal_loss_change, inputs=use_focal_loss, outputs=focal_loss_params, show_progress=False)
 
         def on_autotune_change(value):
             return (
@@ -610,9 +603,7 @@ def build_train_tab():
             visible=False,
             label="Model Performance Metrics (Default Threshold 0.5)",
         )
-        start_training_button = gr.Button(
-            loc.localize("training-tab-start-training-button-label"), variant="huggingface"
-        )
+        start_training_button = gr.Button(loc.localize("training-tab-start-training-button-label"), variant="huggingface")
 
         def train_and_show_metrics(*args):
             history, metrics = start_training(*args)

birdnet_analyzer/gui/utils.py

@@ -1,4 +1,6 @@
 # ruff: noqa: PLW0603
+import base64
+import io
 import multiprocessing
 import os
 import sys
@@ -10,27 +12,8 @@ import gradio as gr
 import webview
 
 import birdnet_analyzer.config as cfg
-from birdnet_analyzer import utils
-
-if utils.FROZEN:
-    # divert stdout & stderr to logs.txt file since we have no console when deployed
-    userdir = Path.home()
-
-    if sys.platform == "win32":
-        userdir /= "AppData/Roaming"
-    elif sys.platform == "linux":
-        userdir /= ".local/share"
-    elif sys.platform == "darwin":
-        userdir /= "Library/Application Support"
-
-    APPDIR = userdir / "BirdNET-Analyzer-GUI"
-
-    APPDIR.mkdir(parents=True, exist_ok=True)
-
-    sys.stderr = sys.stdout = open(str(APPDIR / "logs.txt"), "a")  # noqa: SIM115
-    cfg.ERROR_LOG_FILE = str(APPDIR / os.path.basename(cfg.ERROR_LOG_FILE))
-
 import birdnet_analyzer.gui.localization as loc
+from birdnet_analyzer import utils
 from birdnet_analyzer.gui import settings
 
 loc.load_local_state()
@@ -41,11 +24,12 @@ _CUSTOM_SPECIES = loc.localize("species-list-radio-option-custom-list")
 _PREDICT_SPECIES = loc.localize("species-list-radio-option-predict-list")
 _CUSTOM_CLASSIFIER = loc.localize("species-list-radio-option-custom-classifier")
 _ALL_SPECIES = loc.localize("species-list-radio-option-all")
-_WINDOW: webview.Window = None
+_WINDOW: webview.Window | None = None
 _URL = ""
+_HEART_LOGO = "data:image/svg+xml;base64,PHN2ZyBoZWlnaHQ9IjE2IiB2aWV3Qm94PSIwIDAgMTYgMTYiIHZlcnNpb249IjEuMSIgd2lkdGg9IjE2IiBkYXRhLXZpZXctY29tcG9uZW50PSJ0cnVlIiB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciPg0KICAgIDxwYXRoIGQ9Im04IDE0LjI1LjM0NS42NjZhLjc1Ljc1IDAgMCAxLS42OSAwbC0uMDA4LS4wMDQtLjAxOC0uMDFhNy4xNTIgNy4xNTIgMCAwIDEtLjMxLS4xNyAyMi4wNTUgMjIuMDU1IDAgMCAxLTMuNDM0LTIuNDE0QzIuMDQ1IDEwLjczMSAwIDguMzUgMCA1LjUgMCAyLjgzNiAyLjA4NiAxIDQuMjUgMSA1Ljc5NyAxIDcuMTUzIDEuODAyIDggMy4wMiA4Ljg0NyAxLjgwMiAxMC4yMDMgMSAxMS43NSAxIDEzLjkxNCAxIDE2IDIuODM2IDE2IDUuNWMwIDIuODUtMi4wNDUgNS4yMzEtMy44ODUgNi44MThhMjIuMDY2IDIyLjA2NiAwIDAgMS0zLjc0NCAyLjU4NGwtLjAxOC4wMS0uMDA2LjAwM2gtLjAwMlpNNC4yNSAyLjVjLTEuMzM2IDAtMi43NSAxLjE2NC0yLjc1IDMgMCAyLjE1IDEuNTggNC4xNDQgMy4zNjUgNS42ODJBMjAuNTggMjAuNTggMCAwIDAgOCAxMy4zOTNhMjAuNTggMjAuNTggMCAwIDAgMy4xMzUtMi4yMTFDMTIuOTIgOS42NDQgMTQuNSA3LjY1IDE0LjUgNS41YzAtMS44MzYtMS40MTQtMy0yLjc1LTMtMS4zNzMgMC0yLjYwOS45ODYtMy4wMjkgMi40NTZhLjc0OS43NDkgMCAwIDEtMS40NDIgMEM2Ljg1OSAzLjQ4NiA1LjYyMyAyLjUgNC4yNSAyLjVaIj48L3BhdGg+DQo8L3N2Zz4="  # noqa: E501
 
 
-def gui_runtime_error_handler(f: callable):
+def gui_runtime_error_handler(f):
     """
     A decorator function to handle errors during the execution of a callable.
 
@@ -196,9 +180,7 @@ def select_directory(collect_files=True, max_files=None, state_key=None):
 
         files = utils.collect_audio_files(dir_name, max_files=max_files)
 
-        return dir_name, [
-            [os.path.relpath(file, dir_name), format_seconds(librosa.get_duration(filename=file))] for file in files
-        ]
+        return dir_name, [[os.path.relpath(file, dir_name), format_seconds(librosa.get_duration(filename=file))] for file in files]
 
     return dir_name if dir_name else None
 
@@ -228,9 +210,12 @@ def build_footer():
         <div>Model version: {cfg.MODEL_VERSION}</div>
     </div>
     <div>K. Lisa Yang Center for Conservation Bioacoustics<br>Chemnitz University of Technology</div>
-    <div>{loc.localize("footer-help")}:<br><a href='https://birdnet.cornell.edu/analyzer'
-            target='_blank'>birdnet.cornell.edu/analyzer</a></div>
-</div>"""
+    <div>{loc.localize("footer-help")}:&nbsp;<a href='https://birdnet.cornell.edu/analyzer'
+            target='_blank'>birdnet.cornell.edu/analyzer</a>
+            <br><img id='heart' src='{_HEART_LOGO}'>{loc.localize("footer-support")}: <a href='https://birdnet.cornell.edu/donate' target='_blank'>birdnet.cornell.edu/donate</a>
+    </div>
+
+</div>"""  # noqa: E501
         )
 
 
@@ -417,9 +402,7 @@ def locale():
         The dropdown element.
     """
     label_files = os.listdir(ORIGINAL_TRANSLATED_LABELS_PATH)
-    options = ["EN"] + [
-        label_file.split("BirdNET_GLOBAL_6K_V2.4_Labels_", 1)[1].split(".txt")[0].upper() for label_file in label_files
-    ]
+    options = ["EN"] + [label_file.split("BirdNET_GLOBAL_6K_V2.4_Labels_", 1)[1].split(".txt")[0].upper() for label_file in label_files]
 
     return gr.Dropdown(
         options,
@@ -460,18 +443,12 @@ def species_list_coordinates(show_map=False):
 
         map_plot = gr.Plot(plot_map_scatter_mapbox(0, 0), show_label=False, scale=2, visible=show_map)
 
-        lat_number.change(
-            plot_map_scatter_mapbox, inputs=[lat_number, lon_number], outputs=map_plot, show_progress=False
-        )
-        lon_number.change(
-            plot_map_scatter_mapbox, inputs=[lat_number, lon_number], outputs=map_plot, show_progress=False
-        )
+        lat_number.change(plot_map_scatter_mapbox, inputs=[lat_number, lon_number], outputs=map_plot, show_progress=False)
+        lon_number.change(plot_map_scatter_mapbox, inputs=[lat_number, lon_number], outputs=map_plot, show_progress=False)
 
     with gr.Group():
         with gr.Row():
-            yearlong_checkbox = gr.Checkbox(
-                True, label=loc.localize("species-list-coordinates-yearlong-checkbox-label")
-            )
+            yearlong_checkbox = gr.Checkbox(True, label=loc.localize("species-list-coordinates-yearlong-checkbox-label"))
             week_number = gr.Slider(
                 minimum=1,
                 maximum=48,
@@ -509,9 +486,7 @@ def save_file_dialog(filetypes=(), state_key=None, default_filename=""):
         The selected file or None of the dialog was canceled.
     """
     initial_selection = settings.get_state(state_key, "") if state_key else ""
-    file = _WINDOW.create_file_dialog(
-        webview.SAVE_DIALOG, file_types=filetypes, directory=initial_selection, save_filename=default_filename
-    )
+    file = _WINDOW.create_file_dialog(webview.SAVE_DIALOG, file_types=filetypes, directory=initial_selection, save_filename=default_filename)
 
     if file:
         if state_key:
@@ -608,19 +583,13 @@ def species_lists(opened=True):
         )
 
         with gr.Column(visible=False) as position_row:
-            lat_number, lon_number, week_number, sf_thresh_number, yearlong_checkbox, map_plot = (
-                species_list_coordinates()
-            )
+            lat_number, lon_number, week_number, sf_thresh_number, yearlong_checkbox, map_plot = species_list_coordinates()
 
-        species_file_input = gr.File(
-            file_types=[".txt"], visible=False, label=loc.localize("species-list-custom-list-file-label")
-        )
+        species_file_input = gr.File(file_types=[".txt"], visible=False, label=loc.localize("species-list-custom-list-file-label"))
         empty_col = gr.Column()
 
         with gr.Column(visible=False) as custom_classifier_selector:
-            classifier_selection_button = gr.Button(
-                loc.localize("species-list-custom-classifier-selection-button-label")
-            )
+            classifier_selection_button = gr.Button(loc.localize("species-list-custom-classifier-selection-button-label"))
             classifier_file_input = gr.Files(file_types=[".tflite"], visible=False, interactive=False)
             selected_classifier_state = gr.State()
 
@@ -663,6 +632,26 @@ def species_lists(opened=True):
         )
 
 
+def download_plot(plot, filename=""):
+    from PIL import Image
+
+    imgdata = base64.b64decode(plot.plot.split(",", 1)[1])
+    res = _WINDOW.create_file_dialog(
+        webview.SAVE_DIALOG,
+        file_types=("PNG (*.png)", "Webp (*.webp)", "JPG (*.jpg)"),
+        save_filename=filename,
+    )
+
+    if res:
+        if res.endswith(".webp"):
+            with open(res, "wb") as f:
+                f.write(imgdata)
+        else:
+            output_format = res.rsplit(".", 1)[-1].upper()
+            img = Image.open(io.BytesIO(imgdata))
+            img.save(res, output_format if output_format in ["PNG", "JPEG"] else "PNG")
+
+
 def _get_network_shortcuts():
     """
     Retrieves a list of network shortcut paths from the user's Network Shortcuts folder.
@@ -696,9 +685,7 @@ def _get_network_shortcuts():
                     try:
                         # https://learn.microsoft.com/de-de/windows/win32/shell/links
                         # CLSID_ShellLink: Class ID for Shell Link object
-                        shell_link = pythoncom.CoCreateInstance(
-                            shell.CLSID_ShellLink, None, pythoncom.CLSCTX_INPROC_SERVER, shell.IID_IShellLink
-                        )
+                        shell_link = pythoncom.CoCreateInstance(shell.CLSID_ShellLink, None, pythoncom.CLSCTX_INPROC_SERVER, shell.IID_IShellLink)
 
                         # https://learn.microsoft.com/de-de/windows/win32/api/objidl/nn-objidl-ipersistfile
                         # Query IPersistFile interface used to

birdnet_analyzer/labels/V2.4/BirdNET_GLOBAL_6K_V2.4_Labels_ca.txt

@@ -3211,7 +3211,7 @@ Lichenostomus cratitius_Menjamel de Boqueres
 Lichenostomus melanops_Menjamel Crestagroc
 Lichmera incana_Menjamel d'Orelles Grises
 Lichmera indistincta_Menjamel Bru
-Lichmera indistincta_Menjamel Bru
+Lichmera limbata_Indonesian Honeyeater
 Lichmera squamata_Menjamel Escatós
 Limnoctites rectirostris_Jonquer Becdret
 Limnoctites sulphuriferus_Cuaespinós Gorjagroc

birdnet_analyzer/labels/V2.4/BirdNET_GLOBAL_6K_V2.4_Labels_pl.txt

@@ -3211,7 +3211,7 @@ Lichenostomus cratitius_eukaliptusowczyk purpurowy
 Lichenostomus melanops_eukaliptusowczyk żółtoczelny
 Lichmera incana_miodojadek szarouchy
 Lichmera indistincta_miodojadek brązowy
-Lichmera indistincta_miodojadek brązowy
+Lichmera limbata_Indonesian Honeyeater
 Lichmera squamata_miodojadek łuskowany
 Limnoctites rectirostris_szydłodziobek
 Limnoctites sulphuriferus_moczarnik żółtogardy