Skip to content

archaeo_super_prompt.visualization.display_fields

[docs] module archaeo_super_prompt.visualization.display_fields 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
from dash import Dash, html, callback, Output, Input, dcc
from dash.dash_table.DataTable import DataTable
import plotly.express as px
from pandera.typing.pandas import DataFrame

from ..types.results import ResultSchema
from .types import DashComponent
from .prettify_field_names import prettify_field_names


def display_results(score_results: DataFrame[ResultSchema]) -> DashComponent:
    """Call this method once so the callbacks are initiated once"""
    score_results = prettify_field_names(score_results)

    field_grouping_keys = ["field_name", "evaluation_method"]

    resultsPerField = {
        fieldName: {
            "method": evalMethod,
            "table": resultForField.drop(columns=field_grouping_keys),
        }
        for (fieldName, evalMethod), resultForField in score_results.groupby(
            field_grouping_keys
        )
    }
    fieldNames = list(resultsPerField.keys())

    DEFAULT_SELECTED_FIELD = "Comune"

    layout = [
        html.H1(children="Results", style={"textAlign": "center"}),
        html.H2(children="Global results"),
        dcc.Graph(
            figure=px.histogram(
                score_results, y="field_name", x="metric_value", histfunc="avg"
            )
        ),
        html.H2(children="Per field results"),
        dcc.Dropdown(
            fieldNames, DEFAULT_SELECTED_FIELD, id="dropdown-selection"
        ),
        html.H3(children="Evaluation method used"),
        html.Blockquote(id="eval-method-description"),
        DataTable(id="table-content", page_size=10),
    ]

    def init_callbacks(app: Dash):
        # unused
        app = app

        @callback(
            Output("eval-method-description", "children"),
            Input("dropdown-selection", "value"),
        )
        def updateEvalMethod(fieldName: str):
            return f"Evaluation method used: {resultsPerField[fieldName]['method']}"

        @callback(
            Output("table-content", "data"),
            Input("dropdown-selection", "value"),
        )
        def updatePerFieldResultTable(fieldName: str):
            return resultsPerField[fieldName]["table"].to_dict("records")

        # these functions are globally used thanks to their callback decorator
        updateEvalMethod = updateEvalMethod
        updatePerFieldResultTable = updatePerFieldResultTable

    return layout, init_callbacks