Plotly on_click Function not working on Google Colab

Question

Hello I'm trying to display the figure of this code in google colab it's figuring the plot but when I try to click each point it's not working, it should be turning a green circle, I think the problem is on_click function but i couldn't resolute that problem can anyone help me? I think f.show makes my figure static but i need dynamic figure each touch should call my function

        import plotly.graph_objects as go

import numpy as np
np.random.seed(1)

x = np.random.rand(100)
y = np.random.rand(100)

f = go.FigureWidget([go.Scatter(x=x, y=y, mode='markers')])

scatter = f.data[0]
colors = ['#a3a7e4'] * 100
scatter.marker.color = colors
scatter.marker.size = [10] * 100
f.layout.hovermode = 'closest'


# create our callback function
def update_point(trace, points, selector):
    c = list(scatter.marker.color)
    s = list(scatter.marker.size)
    for i in points.point_inds:
        c[i] = '#bae2be'
        s[i] = 20
        with f.batch_update():
            scatter.marker.color = c
            scatter.marker.size = s


scatter.on_click(update_point)

f

f.show() makes figure static but i need interactive figure in here