feat: frontend: Allow user to specify custom Solara objects

Author: rht

examples/schelling_experimental/app.py

@@ -1,4 +1,4 @@
-from mesa_models.experimental import JupyterViz
+from mesa_models.experimental import JupyterViz, make_text
 from model import Schelling
 
 
@@ -46,7 +46,7 @@ def agent_portrayal(agent):
 page = JupyterViz(
     Schelling,
     model_params,
-    measures=["happy", get_happy_agents],
+    measures=["happy", make_text(get_happy_agents)],
     name="Schelling",
     agent_portrayal=agent_portrayal,
 )

experimental/__init__.py

@@ -1 +1 @@
-from .jupyter_viz import JupyterViz  # noqa
+from .jupyter_viz import JupyterViz, make_text  # noqa

experimental/jupyter_viz.py

@@ -108,6 +108,13 @@ def make_plot(viz, measure):
     solara.FigureMatplotlib(fig, dependencies=[viz.model, viz.df])
 
 
+def make_text(renderer):
+    def function(viz):
+        solara.Markdown(renderer(viz.model))
+
+    return function
+
+
 def make_user_input(user_input, k, v):
     if v["type"] == "SliderInt":
         solara.SliderInt(
@@ -160,7 +167,8 @@ def make_model():
         # 4. Plots
         for i, measure in enumerate(viz.measures):
             if callable(measure):
-                solara.Markdown(measure(viz.model))
+                # Is a custom object
+                measure(viz)
             else:
                 make_plot(viz, measure)