birdnet-analyzer 2.0.0__py3-none-any.whl

birdnet_analyzer/gui/localization.py

@@ -0,0 +1,68 @@
+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)

birdnet_analyzer/gui/multi_file.py

@@ -0,0 +1,246 @@
+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)