app/event_handlers/languages.py

"""
File: languages.py
Author: Elena Ryumina and Dmitry Ryumin
Description: Selected language event handlers for Gradio app.
License: MIT License
"""

import gradio as gr
from pathlib import Path

# Importing necessary components for the Gradio app
from app.description import DESCRIPTIONS
from app.description_steps import STEP_1, STEP_2
from app.config import config_data
from app.components import (
    files_create_ui,
    video_create_ui,
    dropdown_create_ui,
    button,
    html_message,
    dataframe,
)
from app.utils import get_language_settings


def event_handler_languages(languages, files, video, pt_scores, csv_pt_scores):
    lang_id, choices = get_language_settings(languages)

    if not video:
        video = video_create_ui(label=config_data.OtherMessages_VIDEO_PLAYER[lang_id])
        noti_videos = html_message(
            config_data.InformationMessages_NOTI_VIDEOS[lang_id], False
        )
    else:
        video = video_create_ui(
            value=video,
            label=config_data.OtherMessages_VIDEO_PLAYER[lang_id],
            file_name=Path(video).name,
        )
        noti_videos = html_message(
            config_data.OtherMessages_NOTI_CALCULATE[lang_id],
            True,
            False if pt_scores.shape[1] >= 7 else True,
        )

    csv_pt_scores = files_create_ui(
        csv_pt_scores if pt_scores.shape[1] >= 7 else None,
        "single",
        [".csv"],
        config_data.OtherMessages_EXPORT_PT_SCORES[lang_id],
        True,
        False,
        True if pt_scores.shape[1] >= 7 else False,
        "csv-container",
    )
    step_2 = gr.HTML(
        value=STEP_2[lang_id], visible=True if pt_scores.shape[1] >= 7 else False
    )

    if pt_scores.shape[1] >= 7:
        pt_scores = dataframe(
            headers=(config_data.Dataframes_PT_SCORES[lang_id]),
            values=pt_scores.values.tolist(),
            visible=True,
        )
    else:
        pt_scores = dataframe(visible=False)

    return (
        gr.Markdown(value=DESCRIPTIONS[lang_id]),
        gr.HTML(value=STEP_1[lang_id]),
        gr.Image(
            value=config_data.StaticPaths_IMAGES + config_data.Images_LANGUAGES[lang_id]
        ),
        dropdown_create_ui(
            label=None,
            info=None,
            choices=choices,
            value=choices[lang_id],
            visible=True,
            show_label=False,
            elem_classes="dropdown-language-container",
        ),
        gr.Tab(config_data.Labels_APP_LABEL[lang_id]),
        gr.Tab(config_data.Labels_ABOUT_APP_LABEL[lang_id]),
        gr.Tab(config_data.Labels_ABOUT_AUTHORS_LABEL[lang_id]),
        gr.Tab(config_data.Labels_REQUIREMENTS_LABEL[lang_id]),
        files_create_ui(
            value=files,
            label="{} ({})".format(
                config_data.OtherMessages_VIDEO_FILES[lang_id],
                ", ".join(config_data.Settings_SUPPORTED_VIDEO_EXT),
            ),
            file_types=[f".{ext}" for ext in config_data.Settings_SUPPORTED_VIDEO_EXT],
        ),
        video,
        button(
            config_data.OtherMessages_EXAMPLES_APP[lang_id],
            True,
            1,
            "./images/examples.ico",
            True,
            "examples_oceanai",
        ),
        button(
            config_data.OtherMessages_CALCULATE_PT_SCORES[lang_id],
            True if files else False,
            3,
            "./images/calculate_pt_scores.ico",
            True,
            "calculate_oceanai",
        ),
        button(
            config_data.OtherMessages_CLEAR_APP[lang_id],
            True if files else False,
            1,
            "./images/clear.ico",
            True,
            "clear_oceanai",
        ),
        noti_videos,
        pt_scores,
        csv_pt_scores,
        step_2,
    )