Spaces:
Sleeping
Sleeping
File size: 3,524 Bytes
222e0c8 3eaf500 b8ae749 222e0c8 b8ae749 222e0c8 b8ae749 222e0c8 3eaf500 222e0c8 3eaf500 222e0c8 b8ae749 222e0c8 b8ae749 222e0c8 3eaf500 222e0c8 | 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 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 | """This is the scatter plot module for AI Dashboard."""
# pylint: disable=R0801
from typing import Any, List
import logging
import plotly.express as px
from dash import html, dcc
from dash.dependencies import Input, Output
from ..base import BasePlotPlugin
logger = logging.getLogger(__name__)
class ScatterPlotPlugin(BasePlotPlugin):
"""Scatter plot plugin for AI Dashboard."""
name = "Scatter Plot"
def dropdown(self, id_suffix: str, label: str, option: List[str]) -> Any:
"""Generate a dropdown for plot controls."""
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 option], # type: ignore
value=option[0],
clearable=False,
persistence=True,
persistence_type="memory",
style={"color": "#000"},
),
],
style={"width": "130px"},
)
def controls(self) -> Any:
"""Connect the scatter plot plugin to the Dash app."""
nums = self.numeric_columns()
cats = self.categorical_columns()
all_cols = nums + cats
return html.Div(
[
self.dropdown("x", "X-Axis", nums),
self.dropdown("y", "Y-Axis", nums),
self.dropdown("size", "Size", nums),
self.dropdown("color", "Color", all_cols),
self.dropdown("hover", "Hover", all_cols),
],
style={"display": "flex", "flexWrap": "wrap"},
)
def render(self, **kwargs: Any) -> Any:
"""Render the scatter plot."""
x_axis = kwargs.get("x_axis")
y_axis = kwargs.get("y_axis")
size_axis = kwargs.get("size_axis")
color_axis = kwargs.get("color_axis")
hover_axis = kwargs.get("hover_axis")
fig = px.scatter(
self.dataframe,
x=x_axis,
y=y_axis,
size=size_axis if size_axis in self.dataframe.columns else None,
color=color_axis if color_axis in self.dataframe.columns else None,
hover_name=hover_axis if hover_axis in self.dataframe.columns else None,
log_x=True,
size_max=60,
)
return dcc.Graph(figure=fig)
def register_callbacks(self, app: Any) -> None:
"""Register the callbacks for the scatter plot plugin."""
@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"),
Input({"type": "control", "plot": self.name, "axis": "size"}, "value"),
Input({"type": "control", "plot": self.name, "axis": "color"}, "value"),
Input({"type": "control", "plot": self.name, "axis": "hover"}, "value"),
)
def update_scatter(
x_axis: str, y_axis: str, size_axis: str, color_axis: str, hover_axis: str
) -> Any:
"""Update the scatter plot based on user input."""
return self.render(
x_axis=x_axis,
y_axis=y_axis,
size_axis=size_axis,
color_axis=color_axis,
hover_axis=hover_axis,
)
|