Fix live mode polling after document is closed #538

- also set live buttons (play/record) to their correct state when switching documents

ai_diffusion/model.py

@@ -599,7 +599,7 @@ def handle_job_finished(self, job: Job):
             eventloop.run(_report_errors(self._model, self._continue_generating()))
 
     async def _continue_generating(self):
-        while self.is_active:
+        while self.is_active and self._model.document.is_active:
             new_input = await self._model._generate_live(self._last_input)
             if new_input is not None:  # frame was scheduled
                 self._last_input = new_input

ai_diffusion/ui/live.py

@@ -181,6 +181,8 @@ def model(self, model: Model):
                 model.progress_changed.connect(self.update_progress),
                 model.live.result_available.connect(self.show_result),
             ]
+            self.update_is_active()
+            self.update_is_recording()
             self.control_list.model = model
             self.preview_area.clear()