Add support for marimo notebooks

[williambdean]

Struggle with this a bit. Not sure if there is an issue with callback being called from rust

Here is the code from their end:

https://github.com/marimo-team/marimo/blob/cdcb42ce8016d5f149ed06f5f36b6728eef81e38/marimo/_runtime/output/_output.py?plain=1#L28-L49

Closes #223

[williambdean]

This is failing to get the context

https://github.com/marimo-team/marimo/blob/77e95562fb485a81c83bca15aa18378456869070/marimo/_runtime/context/types.py#L226-L234

Ive posted a discussion on their side marimo-team/marimo#5405

[williambdean]

Got it to work!

[williambdean]

Screen.Recording.2025-06-21.at.11.33.15.PM.mov

[aseyboldt]

Thanks, that's really cool!

Just one minor nitpick: maybe prefix the new marimo-related functions with _mo or so, to mark them as private and make the context a bit clearer?

[williambdean]

Yes. Will do

[williambdean]

Feedback addressed. Thanks for the review!