using on_session_created

src/baraddur/server.py

@@ -203,7 +203,29 @@ class BaraddurBokehHandler(BokehHandler):
         self.model = config.model
         self.nmodels = 0
 
-        self.source = ColumnDataSource()
+    def on_session_created(self, doc):
+        doc.source = ColumnDataSource()
+        doc.model = self.model
+        doc.nmodels = 0
+
+        def update_source(self, func):
+            new_nmodels, new_data = func(
+                skip_nmodels=self.nmodels,
+                keys=self.source.data.keys())
+            self.source.stream(new_data)
+            self.nmodels += new_nmodels
+
+        @gen.coroutine
+        def update_models(self):
+            return self.update_source(self.model.get_models)
+
+        @gen.coroutine
+        def update_misfits(self):
+            return self.update_source(self.model.get_misfits)
+
+        doc.update_source = update_source
+        doc.update_models = update_models
+        doc.update_misfits = update_misfits
 
     def update_source(self, func):
         new_nmodels, new_data = func(