diff --git a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToolbar.tsx b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToolbar.tsx index e610ca0077..dfc131c87c 100644 --- a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToolbar.tsx +++ b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/ViewerToolbar.tsx @@ -9,7 +9,13 @@ import CurrentImageButtons from './CurrentImageButtons'; import { ViewerToggleMenu } from './ViewerToggleMenu'; export const ViewerToolbar = memo(() => { - const tab = useAppSelector(activeTabNameSelector); + const showToggle = useAppSelector((s) => { + const tab = activeTabNameSelector(s); + if (tab === 'upscaling' || tab === 'workflows') { + return false; + } + return true; + }); return ( @@ -23,7 +29,7 @@ export const ViewerToolbar = memo(() => { - {tab !== 'workflows' && } + {showToggle && } diff --git a/invokeai/frontend/web/src/features/ui/components/tabs/UpscalingTab.tsx b/invokeai/frontend/web/src/features/ui/components/tabs/UpscalingTab.tsx index 8d4c916c2a..e2da68ceb7 100644 --- a/invokeai/frontend/web/src/features/ui/components/tabs/UpscalingTab.tsx +++ b/invokeai/frontend/web/src/features/ui/components/tabs/UpscalingTab.tsx @@ -1,13 +1,11 @@ import { Box } from '@invoke-ai/ui-library'; import { ImageViewer } from 'features/gallery/components/ImageViewer/ImageViewer'; -import { useImageViewer } from 'features/gallery/components/ImageViewer/useImageViewer'; import { memo } from 'react'; const UpscalingTab = () => { - const imageViewer = useImageViewer(); return ( - {imageViewer.isOpen && } + ); };