diff --git a/tasks/momotalk/ui.py b/tasks/momotalk/ui.py index 3c4ba14..0a97c14 100644 --- a/tasks/momotalk/ui.py +++ b/tasks/momotalk/ui.py @@ -102,6 +102,7 @@ class MomoTalkUI(UI): if self.match_color(NOTIFICATION_BADGE, threshold=120): self.ui_ensure(page_momo_talk) while SWITCH_SIDEBAR.get(self) != "message": + self.device.screenshot() SWITCH_SIDEBAR.set("message", self) return True logger.warning("No students available for interaction")