import numpy as np
from bokeh.models import ColumnDataSource, HoverTool
from bokeh.palettes import Cividis256 as Pallete
from bokeh.plotting import Figure, figure
from bokeh.transform import factor_cmap


def draw_interactive_scatter_plot(
    texts: np.ndarray,
    xs: np.ndarray,
    ys: np.ndarray,
    values: np.ndarray,
    labels: np.ndarray,
    text_column: str,
    label_column: str,
) -> Figure:
    # Smooth down values for coloring, by taking the entropy = log10(perplexity) and multiply it by 10000
    values = ((np.log10(values)) * 10000).round().astype(int)
    # Normalize values to range between 0-255, to assign a color for each value
    max_value = values.max()
    min_value = values.min()
    if max_value - min_value == 0:
        values_color = np.ones(len(values))
    else:
        values_color = (
            ((values - min_value) / (max_value - min_value) * 255).round().astype(int)
        )
    values_color_sorted = sorted(values_color)

    values_list = values.astype(str).tolist()
    values_sorted = sorted(values_list)
    labels_list = labels.astype(str).tolist()

    source = ColumnDataSource(
        data=dict(x=xs, y=ys, text=texts, label=values_list, original_label=labels_list)
    )
    hover = HoverTool(
        tooltips=[(text_column, "@text{safe}"), (label_column, "@original_label")]
    )
    p = figure(plot_width=800, plot_height=800, tools=[hover])
    p.circle(
        "x",
        "y",
        size=10,
        source=source,
        fill_color=factor_cmap(
            "label",
            palette=[Pallete[id_] for id_ in values_color_sorted],
            factors=values_sorted,
        ),
    )

    p.axis.visible = False
    p.xgrid.grid_line_color = None
    p.ygrid.grid_line_color = None
    p.toolbar.logo = None
    return p