Add step count display to JupyterViz

Author: rlskoeser

mesa/experimental/jupyter_viz.py

@@ -181,9 +181,14 @@ def make_model():
     # 3. Buttons
     playing = solara.use_reactive(False)
 
+    # 4. Status indicators
+    #: current round step count
+    step = solara.use_reactive(False)
+
     def on_value_play(change):
         if viz.model.running:
             viz.do_step()
+            step.value = viz.model.schedule.steps
         else:
             playing.value = False
 
@@ -207,6 +212,12 @@ def on_value_play(change):
             playing=playing.value,
             on_playing=playing.set,
         )
+        widgets.IntText(
+            value=step.value,
+            description="Step:",
+            disabled=True,
+            interval=play_interval,
+        )
         # threaded_do_play is not used for now because it
         # doesn't work in Google colab. We use
         # ipywidgets.Play until it is fixed. The threading