test-space-1 / tutorial-reactive-viewer.py
krandiash's picture
Upload with huggingface_hub
908bd70
"""A reactive image viewer that allows you to select a class and see 16 random
images from that class.
This is a tutorial on how to use `reactive` functions in Meerkat, to
build complex reactive workflows.
"""
import meerkat as mk
df = mk.get("imagenette", version="160px")
IMAGE_COL = "img"
LABEL_COL = "label"
@mk.reactive()
def random_images(df: mk.DataFrame):
images = df.sample(16)[IMAGE_COL]
formatter = images.formatters["base"]
# formatter = images.formatters['tiny']
return [formatter.encode(img) for img in images]
labels = list(df[LABEL_COL].unique())
class_selector = mk.gui.Select(
values=list(labels),
value=labels[0],
)
# Note that neither of these will work:
# filtered_df = df[df[LABEL_COL] == class_selector.value]
# (doesn't react to changes in class_selector.value)
# filtered_df = mk.reactive(lambda df: df[df[LABEL_COL] == class_selector.value])(df)
# (doesn't react to changes in class_selector.value)
filtered_df = mk.reactive(lambda df, label: df[df[LABEL_COL] == label])(
df, class_selector.value
)
images = random_images(filtered_df)
# This won't work with a simple reactive fn like a random_images
# that only has df.sample
# as the encoding needs to be done in the reactive fn
# grid = mk.gui.html.gridcols2([
# mk.gui.Image(data=images.formatters["base"].encode(img)) for img in images
# ])
grid = mk.gui.html.div(
[
# Make the image square
mk.gui.html.div(mk.gui.Image(data=img))
for img in images
],
classes="h-fit grid grid-cols-4 gap-1",
)
layout = mk.gui.html.div(
[
mk.gui.html.div(
[mk.gui.Caption("Choose a class:"), class_selector],
classes="flex justify-center items-center mb-2 gap-2",
),
grid,
],
classes="h-full flex flex-col m-2",
)
page = mk.gui.Page(layout, id="reactive-viewer")
page.launch()