如何在Bokeh中使用CustomJS更新散点图

from bokeh.plotting import figure, show
from bokeh.models import ColumnDataSource, CustomJS
from bokeh.layouts import column

source = ColumnDataSource(data=dict(x=[], y=[]))

plot = figure(plot_width=400, plot_height=400)
plot.circle('x', 'y', source=source)

callback = CustomJS(args=dict(source=source), code="""
    // 获取数据源对象
    var data = source.data;
    
    // 清空原有数据
    data['x'] = [];
    data['y'] = [];
    
    // 生成新的数据
    for (var i = 0; i < 10; i++) {
        data['x'].push(Math.random());
        data['y'].push(Math.random());
    }
    
    // 通知数据源对象更新数据
    source.change.emit();
""")

button = Button(label="更新散点图")
button.js_on_click(callback)

layout = column(plot, button)
show(layout)