Update ui widget options based on selection in another widget

[depley]

Is there a way to update the options of one multiselect when I made my selection in another multiselect?

I illustrated the behavior in following code.
If I select/deselect something in select_a it's easy to filter and display the possible options for b in filtered_options_b.
But is there a way to update the options of select_b with these?

import marimo

__generated_with = "0.19.6"
app = marimo.App(width="medium")


@app.cell
def _():
    import marimo as mo
    import pandas as pd
    return mo, pd


@app.cell
def _(pd):
    data = {
        "a" : [1,2,3,1,2,3],
        "b" : [4,4,5,5,6,6],
    }
    df = pd.DataFrame(data=data)
    df
    return (df,)


@app.cell
def _(df, mo):
    options_a = df["a"].drop_duplicates()
    options_b = df["b"].drop_duplicates()

    select_a = mo.ui.multiselect(options=options_a, value=options_a, label="a")
    select_b = mo.ui.multiselect(options=options_b, value=options_b, label="b")
    return select_a, select_b


@app.cell
def _(df, select_a, select_b):
    filtered_options_a = df[df["b"].isin(select_b.value)]["a"].drop_duplicates()
    filtered_options_b = df[df["a"].isin(select_a.value)]["b"].drop_duplicates()

    list(filtered_options_a), list(filtered_options_b)
    return


@app.cell
def _(select_a, select_b):
    select_a, select_b
    return


if __name__ == "__main__":
    app.run()

[dmadisetti]

Does this work for you? https://marimo.app/l/b1c5pn

import marimo

__generated_with = "0.19.6"
app = marimo.App(width="medium")


@app.cell
def _():
    import marimo as mo
    import pandas as pd
    return mo, pd


@app.cell
def _(pd):
    data = {
        "a" : [1,2,3,1,2,3],
        "b" : [4,4,5,5,6,6],
    }
    df = pd.DataFrame(data=data)
    df
    return (df,)


@app.cell
def _(df, mo):
    options_a = df["a"].drop_duplicates()
    options_b = df["b"].drop_duplicates()

    select_a = mo.ui.multiselect(options=options_a, value=options_a, label="a")
    return select_a, select_b


@app.cell
def _(df, select_a, select_b):
    filtered_options_b = df[df["a"].isin(select_a.value)]["b"].drop_duplicates()
    select_b = mo.ui.multiselect(options=filtered_options_b, label="b")
    return


@app.cell
def _(select_a):
    select_a
    return


@app.cell
def _(select_b):
    select_b # ~You want this in a different cell to stop things thrashing~ See comment below
    return


if __name__ == "__main__":
    app.run()

[dmadisetti]

Actually, it works nicely in the same cell. I think it was a bug previously, but seems great now.

[depley]

Ok, nice. That works.
That way one can create a chain of selection like
"first choose a, that effects what you can select for b etc."

Is it possible to close the interactivity loop from here by adding for example on_change= to select_b that affects the options in select_a?

I tried to set up select_a with stateful options and update them when select_b changes, which has no affect currently

https://marimo.app/l/sfqa2x

import marimo

__generated_with = "unknown"
app = marimo.App()


@app.cell
def _():
    import marimo as mo
    import pandas as pd
    return mo, pd


@app.cell
def _(pd):
    data = {
        "a" : [1,2,3,1,2,3],
        "b" : [4,4,5,5,6,6],
    }
    df = pd.DataFrame(data=data)
    df
    return (df,)


@app.cell
def _(df):
    options_a = df["a"].drop_duplicates()
    options_b = df["b"].drop_duplicates()
    return (options_a,)


@app.cell
def _(mo, options_a):
    get_options_a, set_options_a = mo.state(options_a)

    select_a = mo.ui.multiselect(
        options=get_options_a(), 
        value=options_a, 
        label="a"
    )
    return select_a, set_options_a


@app.cell
def _(df, mo, select_a, set_options_a):
    filtered_options_b = df[df["a"].isin(select_a.value)]["b"].drop_duplicates()
    select_b = mo.ui.multiselect(
        options=filtered_options_b, 
        label="b",
        on_change= lambda v: set_options_a(v)   # <- this has no effect
    )
    return (select_b,)


@app.cell
def _(select_a):
    select_a  # How to effect these options by selection in `select_b`
    return


@app.cell
def _(select_b):
    select_b  # options change when for example in `select_a` only value '3' is selected
    return


if __name__ == "__main__":
    app.run()