ui/widgets.py

# ui/widgets.py
"""
Factory functions for common Gradio widgets.
"""
import gradio as gr
from metrics import BERT_FRIENDLY_TO_MODEL
from config import METRIC_CHOICES, DEFAULT_METRICS, DEFAULT_BERTS


def MetricCheckboxGroup(label="Which metrics to compute", default=None, visible=True):
    return gr.CheckboxGroup(
        choices=METRIC_CHOICES,
        label=label,
        value=default or DEFAULT_METRICS,
        visible=visible,
    )


def BertCheckboxGroup(label="Which BERT models (for BERTScore)", default=None, visible=False):
    return gr.CheckboxGroup(
        choices=list(BERT_FRIENDLY_TO_MODEL.keys()),
        label=label,
        value=default or DEFAULT_BERTS,
        visible=visible,
    )