src/ai_dashboard/plot_plugin/hexbin_plot.py

"""Hexbin Plot plugin"""

# pylint: disable=R0801
from typing import Any, List
import plotly.express as px
from dash import html, dcc
from dash.dependencies import Input, Output

from ..base import BasePlotPlugin


class HexbinPlotPlugin(BasePlotPlugin):
    """Hexbin Plot Plugin Class."""

    name = "Hexbin Plot"

    def dropdown(self, id_suffix: str, label: str, options: List[str]) -> Any:
        """Create a dropdown control."""
        return html.Div(
            [
                html.Label(label),
                dcc.Dropdown(
                    id={"type": "control", "plot": self.name, "axis": id_suffix},
                    options=[{"label": c, "value": c} for c in options],  # type: ignore
                    value=options[0],
                    clearable=False,
                    persistence=True,
                    persistence_type="memory",
                    style={"color": "#000"},
                ),
            ],
            style={"width": "130px"},
        )

    def controls(self) -> Any:
        """Render the control panel for the hexbin plot."""
        nums = self.numeric_columns()
        return html.Div(
            [
                self.dropdown("x", "X-Axis", nums),
                self.dropdown("y", "Y-Axis", nums),
            ]
        )

    def render(self, **kwargs: Any) -> Any:  # pylint: disable=W0201
        """Render the hexbin plot based on selected axes."""
        fig = px.density_heatmap(
            self.dataframe,
            x=kwargs["x_axis"],
            y=kwargs["y_axis"],
            nbinsx=20,
            nbinsy=20,
            color_continuous_scale="Viridis",
        )
        return dcc.Graph(figure=fig)

    def register_callbacks(self, app: Any) -> None:
        """Register callbacks for the hexbin plot."""

        @app.callback(  # type: ignore
            Output({"type": "plot-output", "plot": self.name}, "children"),
            Input({"type": "control", "plot": self.name, "axis": "x"}, "value"),
            Input({"type": "control", "plot": self.name, "axis": "y"}, "value"),
        )
        def update(x_axis: str, y_axis: str) -> Any:
            """Update the hexbin plot based on selected axes."""
            return self.render(x_axis=x_axis, y_axis=y_axis)