diff --git a/src/aare/gui/widgets/busy_overlay.py b/src/aare/gui/widgets/busy_overlay.py index 6013a7b2..e41604f6 100644 --- a/src/aare/gui/widgets/busy_overlay.py +++ b/src/aare/gui/widgets/busy_overlay.py @@ -15,9 +15,6 @@ from aare.gui.styles import ( BUSY_PSI_RED, BUSY_PSI_RED_BORDER, BUSY_PSI_RED_DOT, - BUSY_PURPLE, - BUSY_PURPLE_BORDER, - BUSY_PURPLE_DOT, BUSY_RED_BADGE, BUSY_RED_BORDER, BUSY_RED_DOT, @@ -108,7 +105,11 @@ def build_busy_overlay_style( tell_state: TellStateModel | None, session_state: SessionsStateEnum | None = None, ) -> BusyOverlayStyle | None: - if session_state == SessionsStateEnum.Vacant: + if session_state in { + SessionsStateEnum.OwnedByElse, + SessionsStateEnum.PendingYouToElse, + SessionsStateEnum.Vacant, + }: return BusyOverlayStyle( text="In viewing mode", badge_bg=BUSY_YELLOW, @@ -117,20 +118,8 @@ def build_busy_overlay_style( overlay_border=qcolor(BUSY_YELLOW_BORDER, 235), overlay_text=qcolor(WHITE), accent_dot=BUSY_YELLOW_DOT, - subtext="Click here to grab baton if need to interact with GUI", + subtext="Click here to grab the baton if you need to interact with GUI", ) - - if session_state in {SessionsStateEnum.OwnedByElse, SessionsStateEnum.PendingYouToElse}: - return BusyOverlayStyle( - text="GUEST MODE", - badge_bg=BUSY_PURPLE, - badge_fg=WHITE, - overlay_fill=qcolor(BUSY_PURPLE, 190), - overlay_border=qcolor(BUSY_PURPLE_BORDER, 230), - overlay_text=qcolor(WHITE), - accent_dot=BUSY_PURPLE_DOT, - ) - if not is_busy: return None