app/backend/multi_header_util.py

"""
handle multi_header options
"""
from st_aggrid import JsCode
from streamlit_theme import st_theme


HEADER_STYLE = {'fontSize': '18px'}
CELL_STYLE = {'fontSize': '18px'}
LINK = ' https://huggingface.co/datasets/embedding-benchmark/'


def mutil_header_options(column_list: list, avg_column: str, is_section: bool):
    """
    get mutil_header_options - 优化版本，减少组件实例化
    :param column_list:
    :param avg_column:
    :param is_section:
    :return:
    """
    if is_section:
        column_def_list = [
            {'headerName': column if "Average" not in column else column.replace("Average", "").strip().capitalize(),
             'field': column,
             'headerClass': 'custom-header-style',
             'cellClass': 'custom-cell-style',
             'headerTooltip': column if "Average" not in column else column.replace("Average",
                                                                                    "").strip().capitalize()
             # 'suppressSizeToFit': True
             } for column in column_list if
            column not in (avg_column, "Closed average", "Open average")]

        return column_def_list

    mutil_column_list = [column for column in column_list if
                         column not in (avg_column, "Closed average", "Open average")]
    close_group_list = list(filter(lambda x: x.startswith('_'), mutil_column_list))
    open_group_list = list(filter(lambda x: not x.startswith('_'), mutil_column_list))

    theme = st_theme(key="st_theme_1")
    if theme:
        current_theme = theme.get("base", "light")
    else:
        current_theme = "light"



    close_group_def = {
        'headerName': 'CLOSED DATASETS',
        'children': [
            {'headerName': column.replace('_', ''),
             'field': column,

             "headerComponentParams": {
                 "innerHeaderComponent": "linkHeaderComponent",
                 "url": LINK + column.replace('_', ''),
                 "headerName": column
             },

             'headerClass': 'custom-header-style',
             'cellClass': 'custom-cell-style',
             'sortable': True,
             'width': 150,
             # "suppressColumnVirtualisation": True,

             } for column in close_group_list
        ],

    }
    open_group_def = {
        'headerName': 'OPEN DATASETS',
        'children': [
            {'headerName': column,
             'field': column,
             "headerComponentParams": {
                 "innerHeaderComponent": "linkHeaderComponent",
                 "url": LINK + column.replace('_', ''),
                 "headerName": column
             },
             'headerClass': 'custom-header-style',
             'cellClass': 'custom-cell-style',
             'sortable': True,
             'width': 150,
             "suppressColumnVirtualisation": True,

             } for column in open_group_list
        ],

    }
    return [close_group_def, open_group_def]


def get_header_options(column_list: list, avg_column: str,  is_section: bool):
    grid_options = {
        'columnDefs': [
            {
                'headerName': 'Model Name',
                'field': 'model_name',
                'pinned': 'left',
                'sortable': False,
                'headerClass': 'custom-header-style',
                'cellClass': 'custom-cell-style',
                'autoHeaderHeight': True,
                "tooltipValueGetter": JsCode(
                    """function(p) {return p.value}"""
                ),
                "width": 250,
                'cellRenderer': JsCode("""class CustomHTML {
                            init(params) {
                                const link = params.data.reference;
                                this.eGui = document.createElement('div');
                                this.eGui.innerHTML = link ? 
                                    `<a href="${link}" class="a-cell" target="_blank">${params.value} </a>` : 
                                    params.value;
                            }

                            getGui() {
                                return this.eGui;
                            }
                        }"""),
                'suppressSizeToFit': True

            },
            {'headerName': "Vendor",
             'field': 'vendor',
             'headerClass': 'custom-header-style',
             'cellClass': 'custom-cell-style',
             # 'suppressSizeToFit': True
             },
            {'headerName': "Overall Score",
             'field': avg_column,
             'headerClass': 'custom-header-style',
             'cellClass': 'custom-cell-style',
             # 'suppressSizeToFit': True
             },

            # Add Open average column definition
            {'headerName': 'Open Average',
             'field': 'Open average',
             'headerClass': 'custom-header-style',
             'cellClass': 'custom-cell-style',
             # 'suppressSizeToFit': True
             },

            {'headerName': 'Closed Average',
             'field': 'Closed average',
             'headerClass': 'custom-header-style',
             'cellClass': 'custom-cell-style',
             # 'suppressSizeToFit': True
             },

            {
                'headerName': 'Embd Dtype',
                'field': 'embd_dtype',
                'headerClass': 'custom-header-style',
                'cellClass': 'custom-cell-style',
                # 'suppressSizeToFit': True,
            },
            {
                'headerName': 'Embd Dim',
                'field': 'embd_dim',
                'headerClass': 'custom-header-style',
                'cellClass': 'custom-cell-style',
                # 'suppressSizeToFit': True,
            },
            {
                'headerName': 'Number of Parameters',
                'field': 'num_params',
                'cellDataType': 'number',
                "colId": "num_params",
                'headerClass': 'custom-header-style',
                'cellClass': 'custom-cell-style',
                'valueFormatter': JsCode(
                    """function(params) {
                    const num = params.value;
                    if (num >= 1e9) return (num / 1e9).toFixed(2) + "B";
                    if (num >= 1e6) return (num / 1e6).toFixed(2) + "M";
                    if (num >= 1e3) return (num / 1e3).toFixed(2) + "K";
                    return num;
                    }"""
                ),
                "width": 120,
                # 'suppressSizeToFit': True,
            },
            {
                'headerName': 'Context Length',
                'field': 'max_tokens',
                'headerClass': 'custom-header-style',
                'cellClass': 'custom-cell-style',
                # 'suppressSizeToFit': True,
            },

            *mutil_header_options(column_list, avg_column, is_section)
        ],
        'defaultColDef': {
            'filter': True,
            'sortable': True,
            'resizable': True,
            'headerClass': "multi-line-header",
            'autoHeaderHeight': True,
            'width': 105
        },

        "autoSizeStrategy": {
            "type": 'fitCellContents',
            "colIds": [column for column in column_list if column not in (avg_column, "Closed average", "Open average")]
        },
        "tooltipShowDelay": 500,

    }

    return grid_options