What is the least-bad way to create *reusable* interactive UI elements?

[AwdeDarkar]

I am a huge fan of marimo and I love its reactive UIs but having to make everything global has become a pretty big pain point. A common pattern for me is to create data classes with rich UI visualizations, for example

class SomeStructure(BaseModel):
    name: str
    value: int
    table: Dict
    
    def _display_(self) -> mo.Html:
        return mo.vstack([
            mo.md(f"**SomeStructure** ({self.name}) [{self.value}]"),
            mo.json(self.table),
        ])

This is helpful, the json widget handles (most) dictionaries I throw at it gracefully with collapsible sub-menus and everything. But suppose the tables are getting too big and nested to navigate easily and I want to be able to filter the display based on a dropdown. Something like,

class SomeStructure(BaseModel):
    name: str
    value: int
    table: Dict

    def _display_(self) -> mo.Html:
        selection = mo.ui.dropdown(options=self.table.keys())
 
        return mo.vstack([
            mo.md(f"**SomeStructure** ({self.name}) [{self.value}]"),
            selection,
            mo.json(self.table[selection.value]),
        ])

Now it raises a RuntimeError. My understanding is this happens because marimo's reactivity model relies on tracking changes in global variables to update UI elements. All of the 'state' here is self-contained, no other values can depend on selection, but there doesn't seem to be a way for marimo to respond to this sort of change.

Frustratingly, even using mo.state and the dropdown's on_change argument doesn't work. It no longer raises an exception but the changes are never reflected in the UI. I don't know if this is because the getter call isn't correctly forcing a UI update when state changes, or if on_change isn't being called in the first place but either way it didn't function as I expended.

Is there any way to make this work? The only approach I've found is using AnyWidget but that throws out all the marimo UI widgets and writing javascript in python strings is painful.