diff --git a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToggleMenu.tsx b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToggleMenu.tsx
index 7dc13afb48..78a57bf192 100644
--- a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToggleMenu.tsx
+++ b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToggleMenu.tsx
@@ -1,18 +1,8 @@
-import {
- Button,
- Flex,
- Icon,
- Popover,
- PopoverArrow,
- PopoverBody,
- PopoverContent,
- PopoverTrigger,
- Text,
-} from '@invoke-ai/ui-library';
+import { ButtonGroup, Flex, IconButton, Text, Tooltip } from '@invoke-ai/ui-library';
import { useImageViewer } from 'features/gallery/components/ImageViewer/useImageViewer';
import { useHotkeys } from 'react-hotkeys-hook';
import { useTranslation } from 'react-i18next';
-import { PiCaretDownBold, PiCheckBold, PiEyeBold, PiPencilBold } from 'react-icons/pi';
+import { PiEyeBold, PiPencilBold } from 'react-icons/pi';
export const ViewerToggleMenu = () => {
const { t } = useTranslation();
@@ -21,49 +11,45 @@ export const ViewerToggleMenu = () => {
useHotkeys('esc', imageViewer.onClose, [imageViewer]);
return (
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+
+
+
+ {t('common.viewing')}
+ {t('common.viewingDesc')}
+
+ }
+ >
+ }
+ onClick={imageViewer.onOpen}
+ variant={imageViewer.isOpen ? 'solid' : 'outline'}
+ colorScheme={imageViewer.isOpen ? 'invokeBlue' : 'base'}
+ aria-label={t('common.viewing')}
+ w={12}
+ />
+
+
+ {t('common.editing')}
+ {t('common.editingDesc')}
+
+ }
+ >
+ }
+ onClick={imageViewer.onClose}
+ variant={!imageViewer.isOpen ? 'solid' : 'outline'}
+ colorScheme={!imageViewer.isOpen ? 'invokeBlue' : 'base'}
+ aria-label={t('common.editing')}
+ w={12}
+ />
+
+
+
);
};