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

birdnet_analyzer/gui/localization.py

@@ -1,68 +1,75 @@
-import json
-import os
-
-import birdnet_analyzer.gui.settings as settings
-
-SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
-LANGUAGE_DIR = os.path.join(os.path.dirname(SCRIPT_DIR), "lang")
-LANGUAGE_LOOKUP = {}
-TARGET_LANGUAGE = settings.FALLBACK_LANGUAGE
-
-
-def load_local_state():
-    """
-    Loads the local language settings and populates the LANGUAGE_LOOKUP dictionary with the appropriate translations.
-    This function performs the following steps:
-    """
-    global LANGUAGE_LOOKUP
-    global TARGET_LANGUAGE
-
-    settings.ensure_settings_file()
-
-    try:
-        TARGET_LANGUAGE = json.load(open(settings.GUI_SETTINGS_PATH, encoding="utf-8"))["language-id"]
-    except FileNotFoundError:
-        print(f"gui-settings.json not found. Using fallback language {settings.FALLBACK_LANGUAGE}.")
-
-    try:
-        with open(f"{LANGUAGE_DIR}/{TARGET_LANGUAGE}.json", "r", encoding="utf-8") as f:
-            LANGUAGE_LOOKUP = json.load(f)
-    except FileNotFoundError:
-        print(
-            f"Language file for {TARGET_LANGUAGE} not found in {LANGUAGE_DIR}. Using fallback language {settings.FALLBACK_LANGUAGE}."
-        )
-
-    if TARGET_LANGUAGE != settings.FALLBACK_LANGUAGE:
-        with open(f"{LANGUAGE_DIR}/{settings.FALLBACK_LANGUAGE}.json", "r") as f:
-            fallback: dict = json.load(f)
-
-        for key, value in fallback.items():
-            if key not in LANGUAGE_LOOKUP:
-                LANGUAGE_LOOKUP[key] = value
-
-
-def localize(key: str) -> str:
-    """
-    Translates a given key into its corresponding localized string.
-
-    Args:
-        key (str): The key to be localized.
-
-    Returns:
-        str: The localized string corresponding to the given key. If the key is not found in the localization lookup, the original key is returned.
-    """
-    return LANGUAGE_LOOKUP.get(key, key)
-
-
-def set_language(language: str):
-    """
-    Sets the language for the application by updating the GUI settings file.
-    This function ensures that the settings file exists, reads the current settings,
-    updates the "language-id" field with the provided language, and writes the updated
-    settings back to the file.
-
-    Args:
-        language (str): The language identifier to set in the settings file.
-    """
-    if language:
-        settings.set_setting("language-id", language)
+# ruff: noqa: PLW0603
+import json
+import os
+
+from birdnet_analyzer.gui import settings
+
+SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
+LANGUAGE_DIR = os.path.join(os.path.dirname(SCRIPT_DIR), "lang")
+LANGUAGE_LOOKUP = {}
+TARGET_LANGUAGE = settings.FALLBACK_LANGUAGE
+
+
+def load_local_state():
+    """
+    Loads the local language settings and populates the LANGUAGE_LOOKUP dictionary with the appropriate translations.
+    This function performs the following steps:
+    """
+    global LANGUAGE_LOOKUP
+    global TARGET_LANGUAGE
+
+    settings.ensure_settings_file()
+
+    try:
+        with open(settings.GUI_SETTINGS_PATH, encoding="utf-8") as f:
+            settings_data = json.load(f)
+
+            if "language-id" in settings_data:
+                TARGET_LANGUAGE = settings_data["language-id"]
+    except FileNotFoundError:
+        print(f"gui-settings.json not found. Using fallback language {settings.FALLBACK_LANGUAGE}.")
+
+    try:
+        with open(f"{LANGUAGE_DIR}/{TARGET_LANGUAGE}.json", encoding="utf-8") as f:
+            LANGUAGE_LOOKUP = json.load(f)
+    except FileNotFoundError:
+        print(
+            f"Language file for {TARGET_LANGUAGE} not found in {LANGUAGE_DIR}."
+            + "Using fallback language {settings.FALLBACK_LANGUAGE}."
+        )
+
+    if TARGET_LANGUAGE != settings.FALLBACK_LANGUAGE:
+        with open(f"{LANGUAGE_DIR}/{settings.FALLBACK_LANGUAGE}.json") as f:
+            fallback: dict = json.load(f)
+
+        for key, value in fallback.items():
+            if key not in LANGUAGE_LOOKUP:
+                LANGUAGE_LOOKUP[key] = value
+
+
+def localize(key: str) -> str:
+    """
+    Translates a given key into its corresponding localized string.
+
+    Args:
+        key (str): The key to be localized.
+
+    Returns:
+        str: The localized string corresponding to the given key.
+             If the key is not found in the localization lookup, the original key is returned.
+    """
+    return LANGUAGE_LOOKUP.get(key, key)
+
+
+def set_language(language: str):
+    """
+    Sets the language for the application by updating the GUI settings file.
+    This function ensures that the settings file exists, reads the current settings,
+    updates the "language-id" field with the provided language, and writes the updated
+    settings back to the file.
+
+    Args:
+        language (str): The language identifier to set in the settings file.
+    """
+    if language:
+        settings.set_setting("language-id", language)

birdnet_analyzer/gui/multi_file.py

@@ -1,246 +1,245 @@
-import gradio as gr
-
-import birdnet_analyzer.config as cfg
-import birdnet_analyzer.gui.utils as gu
-import birdnet_analyzer.gui.localization as loc
-
-OUTPUT_TYPE_MAP = {
-    "Raven selection table": "table",
-    "Audacity": "audacity",
-    "CSV": "csv",
-    "Kaleidoscope": "kaleidoscope",
-}
-
-
-@gu.gui_runtime_error_handler
-def run_batch_analysis(
-    output_path,
-    use_top_n,
-    top_n,
-    confidence,
-    sensitivity,
-    overlap,
-    merge_consecutive,
-    audio_speed,
-    fmin,
-    fmax,
-    species_list_choice,
-    species_list_file,
-    lat,
-    lon,
-    week,
-    use_yearlong,
-    sf_thresh,
-    custom_classifier_file,
-    output_type,
-    combine_tables,
-    locale,
-    batch_size,
-    threads,
-    input_dir,
-    skip_existing,
-    progress=gr.Progress(),
-):
-    from birdnet_analyzer.gui.analysis import run_analysis
-
-    gu.validate(input_dir, loc.localize("validation-no-directory-selected"))
-    batch_size = int(batch_size)
-    threads = int(threads)
-
-    if species_list_choice == gu._CUSTOM_SPECIES:
-        gu.validate(species_list_file, loc.localize("validation-no-species-list-selected"))
-
-    if fmin is None or fmax is None or fmin < cfg.SIG_FMIN or fmax > cfg.SIG_FMAX or fmin > fmax:
-        raise gr.Error(f"{loc.localize('validation-no-valid-frequency')} [{cfg.SIG_FMIN}, {cfg.SIG_FMAX}]")
-
-    return run_analysis(
-        None,
-        output_path,
-        use_top_n,
-        top_n,
-        confidence,
-        sensitivity,
-        overlap,
-        merge_consecutive,
-        audio_speed,
-        fmin,
-        fmax,
-        species_list_choice,
-        species_list_file,
-        lat,
-        lon,
-        week,
-        use_yearlong,
-        sf_thresh,
-        custom_classifier_file,
-        output_type,
-        combine_tables,
-        "en" if not locale else locale,
-        batch_size if batch_size and batch_size > 0 else 1,
-        threads if threads and threads > 0 else 4,
-        input_dir,
-        skip_existing,
-        True,
-        progress,
-    )
-
-
-def build_multi_analysis_tab():
-    with gr.Tab(loc.localize("multi-tab-title")):
-        input_directory_state = gr.State()
-        output_directory_predict_state = gr.State()
-
-        with gr.Row():
-            with gr.Column():
-                select_directory_btn = gr.Button(loc.localize("multi-tab-input-selection-button-label"))
-                directory_input = gr.Matrix(
-                    interactive=False,
-                    headers=[
-                        loc.localize("multi-tab-samples-dataframe-column-subpath-header"),
-                        loc.localize("multi-tab-samples-dataframe-column-duration-header"),
-                    ],
-                )
-
-                def select_directory_on_empty():  # Nishant - Function modified for For Folder selection
-                    folder = gu.select_folder(state_key="batch-analysis-data-dir")
-
-                    if folder:
-                        files_and_durations = gu.get_audio_files_and_durations(folder)
-                        if len(files_and_durations) > 100:
-                            return [folder, files_and_durations[:100] + [["..."]]]  # hopefully fixes issue#272
-                        return [folder, files_and_durations]
-
-                    return ["", [[loc.localize("multi-tab-samples-dataframe-no-files-found")]]]
-
-                select_directory_btn.click(
-                    select_directory_on_empty, outputs=[input_directory_state, directory_input], show_progress=True
-                )
-
-            with gr.Column():
-                select_out_directory_btn = gr.Button(loc.localize("multi-tab-output-selection-button-label"))
-                selected_out_textbox = gr.Textbox(
-                    label=loc.localize("multi-tab-output-textbox-label"),
-                    interactive=False,
-                    placeholder=loc.localize("multi-tab-output-textbox-placeholder"),
-                )
-
-                def select_directory_wrapper():  # Nishant - Function modified for For Folder selection
-                    folder = gu.select_folder(state_key="batch-analysis-output-dir")
-                    return (folder, folder) if folder else ("", "")
-
-                select_out_directory_btn.click(
-                    select_directory_wrapper,
-                    outputs=[output_directory_predict_state, selected_out_textbox],
-                    show_progress=False,
-                )
-
-        (
-            use_top_n,
-            top_n_input,
-            confidence_slider,
-            sensitivity_slider,
-            overlap_slider,
-            merge_consecutive_slider,
-            audio_speed_slider,
-            fmin_number,
-            fmax_number,
-        ) = gu.sample_sliders()
-
-        (
-            species_list_radio,
-            species_file_input,
-            lat_number,
-            lon_number,
-            week_number,
-            sf_thresh_number,
-            yearlong_checkbox,
-            selected_classifier_state,
-            map_plot,
-        ) = gu.species_lists()
-
-        with gr.Accordion(loc.localize("multi-tab-output-accordion-label"), open=True):
-            with gr.Group():
-                output_type_radio = gr.CheckboxGroup(
-                    list(OUTPUT_TYPE_MAP.items()),
-                    value="table",
-                    label=loc.localize("multi-tab-output-radio-label"),
-                    info=loc.localize("multi-tab-output-radio-info"),
-                )
-
-                with gr.Row():
-                    with gr.Column():
-                        combine_tables_checkbox = gr.Checkbox(
-                            False,
-                            label=loc.localize("multi-tab-output-combine-tables-checkbox-label"),
-                            info=loc.localize("multi-tab-output-combine-tables-checkbox-info"),
-                        )
-
-                with gr.Row():
-                    skip_existing_checkbox = gr.Checkbox(
-                        False,
-                        label=loc.localize("multi-tab-skip-existing-checkbox-label"),
-                        info=loc.localize("multi-tab-skip-existing-checkbox-info"),
-                    )
-
-        with gr.Row():
-            batch_size_number = gr.Number(
-                precision=1,
-                label=loc.localize("multi-tab-batchsize-number-label"),
-                value=1,
-                info=loc.localize("multi-tab-batchsize-number-info"),
-                minimum=1,
-            )
-            threads_number = gr.Number(
-                precision=1,
-                label=loc.localize("multi-tab-threads-number-label"),
-                value=4,
-                info=loc.localize("multi-tab-threads-number-info"),
-                minimum=1,
-            )
-
-        locale_radio = gu.locale()
-
-        start_batch_analysis_btn = gr.Button(loc.localize("analyze-start-button-label"), variant="huggingface")
-
-        result_grid = gr.Matrix(
-            headers=[
-                loc.localize("multi-tab-result-dataframe-column-file-header"),
-                loc.localize("multi-tab-result-dataframe-column-execution-header"),
-            ],
-        )
-
-        inputs = [
-            output_directory_predict_state,
-            use_top_n,
-            top_n_input,
-            confidence_slider,
-            sensitivity_slider,
-            overlap_slider,
-            merge_consecutive_slider,
-            audio_speed_slider,
-            fmin_number,
-            fmax_number,
-            species_list_radio,
-            species_file_input,
-            lat_number,
-            lon_number,
-            week_number,
-            yearlong_checkbox,
-            sf_thresh_number,
-            selected_classifier_state,
-            output_type_radio,
-            combine_tables_checkbox,
-            locale_radio,
-            batch_size_number,
-            threads_number,
-            input_directory_state,
-            skip_existing_checkbox,
-        ]
-
-        start_batch_analysis_btn.click(run_batch_analysis, inputs=inputs, outputs=result_grid)
-
-    return lat_number, lon_number, map_plot
-
-
-if __name__ == "__main__":
-    gu.open_window(build_multi_analysis_tab)
+ # ruff: noqa: I001
+import gradio as gr
+
+import birdnet_analyzer.config as cfg
+import birdnet_analyzer.gui.utils as gu
+import birdnet_analyzer.gui.localization as loc
+
+OUTPUT_TYPE_MAP = {
+    "Raven selection table": "table",
+    "Audacity": "audacity",
+    "CSV": "csv",
+    "Kaleidoscope": "kaleidoscope",
+}
+
+
+@gu.gui_runtime_error_handler
+def run_batch_analysis(
+    output_path,
+    use_top_n,
+    top_n,
+    confidence,
+    sensitivity,
+    overlap,
+    merge_consecutive,
+    audio_speed,
+    fmin,
+    fmax,
+    species_list_choice,
+    species_list_file,
+    lat,
+    lon,
+    week,
+    use_yearlong,
+    sf_thresh,
+    custom_classifier_file,
+    output_type,
+    combine_tables,
+    locale,
+    batch_size,
+    threads,
+    input_dir,
+    skip_existing,
+    progress=gr.Progress(),
+):
+    from birdnet_analyzer.gui.analysis import run_analysis
+
+    gu.validate(input_dir, loc.localize("validation-no-directory-selected"))
+    batch_size = int(batch_size)
+    threads = int(threads)
+
+    if species_list_choice == gu._CUSTOM_SPECIES:
+        gu.validate(species_list_file, loc.localize("validation-no-species-list-selected"))
+
+    if fmin is None or fmax is None or fmin < cfg.SIG_FMIN or fmax > cfg.SIG_FMAX or fmin > fmax:
+        raise gr.Error(f"{loc.localize('validation-no-valid-frequency')} [{cfg.SIG_FMIN}, {cfg.SIG_FMAX}]")
+
+    return run_analysis(
+        None,
+        output_path,
+        use_top_n,
+        top_n,
+        confidence,
+        sensitivity,
+        overlap,
+        merge_consecutive,
+        audio_speed,
+        fmin,
+        fmax,
+        species_list_choice,
+        species_list_file,
+        lat,
+        lon,
+        week,
+        use_yearlong,
+        sf_thresh,
+        custom_classifier_file,
+        output_type,
+        combine_tables,
+        locale if locale else "en",
+        batch_size if batch_size and batch_size > 0 else 1,
+        threads if threads and threads > 0 else 4,
+        input_dir,
+        skip_existing,
+        True,
+        progress,
+    )
+
+
+def build_multi_analysis_tab():
+    with gr.Tab(loc.localize("multi-tab-title")):
+        input_directory_state = gr.State()
+        output_directory_predict_state = gr.State()
+
+        with gr.Row():
+            with gr.Column():
+                select_directory_btn = gr.Button(loc.localize("multi-tab-input-selection-button-label"))
+                directory_input = gr.Matrix(
+                    interactive=False,
+                    headers=[
+                        loc.localize("multi-tab-samples-dataframe-column-subpath-header"),
+                        loc.localize("multi-tab-samples-dataframe-column-duration-header"),
+                    ],
+                )
+
+                def select_directory_on_empty():  # Nishant - Function modified for For Folder selection
+                    folder = gu.select_folder(state_key="batch-analysis-data-dir")
+
+                    if folder:
+                        files_and_durations = gu.get_audio_files_and_durations(folder)
+                        if len(files_and_durations) > 100:
+                            return [folder, files_and_durations[:100] + [["..."]]]  # hopefully fixes issue#272
+                        return [folder, files_and_durations]
+
+                    return ["", [[loc.localize("multi-tab-samples-dataframe-no-files-found")]]]
+
+                select_directory_btn.click(
+                    select_directory_on_empty, outputs=[input_directory_state, directory_input], show_progress=True
+                )
+
+            with gr.Column():
+                select_out_directory_btn = gr.Button(loc.localize("multi-tab-output-selection-button-label"))
+                selected_out_textbox = gr.Textbox(
+                    label=loc.localize("multi-tab-output-textbox-label"),
+                    interactive=False,
+                    placeholder=loc.localize("multi-tab-output-textbox-placeholder"),
+                )
+
+                def select_directory_wrapper():  # Nishant - Function modified for For Folder selection
+                    folder = gu.select_folder(state_key="batch-analysis-output-dir")
+                    return (folder, folder) if folder else ("", "")
+
+                select_out_directory_btn.click(
+                    select_directory_wrapper,
+                    outputs=[output_directory_predict_state, selected_out_textbox],
+                    show_progress=False,
+                )
+
+        (
+            use_top_n,
+            top_n_input,
+            confidence_slider,
+            sensitivity_slider,
+            overlap_slider,
+            merge_consecutive_slider,
+            audio_speed_slider,
+            fmin_number,
+            fmax_number,
+        ) = gu.sample_sliders()
+
+        (
+            species_list_radio,
+            species_file_input,
+            lat_number,
+            lon_number,
+            week_number,
+            sf_thresh_number,
+            yearlong_checkbox,
+            selected_classifier_state,
+            map_plot,
+        ) = gu.species_lists()
+
+        with gr.Accordion(loc.localize("multi-tab-output-accordion-label"), open=True), gr.Group():
+            output_type_radio = gr.CheckboxGroup(
+                list(OUTPUT_TYPE_MAP.items()),
+                value="table",
+                label=loc.localize("multi-tab-output-radio-label"),
+                info=loc.localize("multi-tab-output-radio-info"),
+            )
+
+            with gr.Row():
+                combine_tables_checkbox = gr.Checkbox(
+                    False,
+                    label=loc.localize("multi-tab-output-combine-tables-checkbox-label"),
+                    info=loc.localize("multi-tab-output-combine-tables-checkbox-info"),
+                )
+
+            with gr.Row():
+                skip_existing_checkbox = gr.Checkbox(
+                    False,
+                    label=loc.localize("multi-tab-skip-existing-checkbox-label"),
+                    info=loc.localize("multi-tab-skip-existing-checkbox-info"),
+                )
+
+        with gr.Row():
+            batch_size_number = gr.Number(
+                precision=1,
+                label=loc.localize("multi-tab-batchsize-number-label"),
+                value=1,
+                info=loc.localize("multi-tab-batchsize-number-info"),
+                minimum=1,
+            )
+            threads_number = gr.Number(
+                precision=1,
+                label=loc.localize("multi-tab-threads-number-label"),
+                value=4,
+                info=loc.localize("multi-tab-threads-number-info"),
+                minimum=1,
+            )
+
+        locale_radio = gu.locale()
+
+        start_batch_analysis_btn = gr.Button(loc.localize("analyze-start-button-label"), variant="huggingface")
+
+        result_grid = gr.Matrix(
+            headers=[
+                loc.localize("multi-tab-result-dataframe-column-file-header"),
+                loc.localize("multi-tab-result-dataframe-column-execution-header"),
+            ],
+        )
+
+        inputs = [
+            output_directory_predict_state,
+            use_top_n,
+            top_n_input,
+            confidence_slider,
+            sensitivity_slider,
+            overlap_slider,
+            merge_consecutive_slider,
+            audio_speed_slider,
+            fmin_number,
+            fmax_number,
+            species_list_radio,
+            species_file_input,
+            lat_number,
+            lon_number,
+            week_number,
+            yearlong_checkbox,
+            sf_thresh_number,
+            selected_classifier_state,
+            output_type_radio,
+            combine_tables_checkbox,
+            locale_radio,
+            batch_size_number,
+            threads_number,
+            input_directory_state,
+            skip_existing_checkbox,
+        ]
+
+        start_batch_analysis_btn.click(run_batch_analysis, inputs=inputs, outputs=result_grid)
+
+    return lat_number, lon_number, map_plot
+
+
+if __name__ == "__main__":
+    gu.open_window(build_multi_analysis_tab)