Calling python function from JS in Colab

js_link.py

from IPython.display import HTML, JSON
from google.colab import output

# first, register the python function to use
def array():
  return JSON([1,2,3])  # can send any JSON 
output.register_callback('array', array)

# then invoke it later in an %%html block
%%html
{{PRINT}}
<script> 
(async function() {
  invoke = google.colab.kernel.invokeFunction
  result = await invoke('array', [], {})  // *args, **kw

  print(result.data['application/json'])
})()
</script>