diff --git a/invokeai/frontend/web/src/features/ui/layouts/TabWithLaunchpadIcon.tsx b/invokeai/frontend/web/src/features/ui/layouts/TabWithLaunchpadIcon.tsx index 50784966d4..5de641b853 100644 --- a/invokeai/frontend/web/src/features/ui/layouts/TabWithLaunchpadIcon.tsx +++ b/invokeai/frontend/web/src/features/ui/layouts/TabWithLaunchpadIcon.tsx @@ -1,10 +1,11 @@ -import { Flex, Text } from '@invoke-ai/ui-library'; +import { Flex, Icon, Text } from '@invoke-ai/ui-library'; import { useAppSelector } from 'app/store/storeHooks'; import { useCallbackOnDragEnter } from 'common/hooks/useCallbackOnDragEnter'; import type { IDockviewPanelHeaderProps } from 'dockview'; import { selectActiveTab } from 'features/ui/store/uiSelectors'; import type { TabName } from 'features/ui/store/uiTypes'; import { memo, useCallback, useRef } from 'react'; +import type { IconType } from 'react-icons'; import { PiBoundingBoxBold, PiCubeBold, @@ -14,13 +15,13 @@ import { PiTextAaBold, } from 'react-icons/pi'; -const TAB_ICONS: Record = { - generate: , - canvas: , - upscaling: , - workflows: , - models: , - queue: , +const TAB_ICONS: Record = { + generate: PiTextAaBold, + canvas: PiBoundingBoxBold, + upscaling: PiFrameCornersBold, + workflows: PiFlowArrowBold, + models: PiCubeBold, + queue: PiQueueBold, }; export const TabWithLaunchpadIcon = memo((props: IDockviewPanelHeaderProps) => { @@ -35,20 +36,10 @@ export const TabWithLaunchpadIcon = memo((props: IDockviewPanelHeaderProps) => { useCallbackOnDragEnter(setActive, ref, 300); - // Show icon only for Launchpad panel - const isLaunchpadPanel = props.api.id === 'launchpad'; - const currentTabIcon = TAB_ICONS[activeTab]; - return ( - - {isLaunchpadPanel && currentTabIcon && ( - - {currentTabIcon} - - )} - - {props.api.title ?? props.api.id} - + + + {props.api.title ?? props.api.id} ); });