From e007dbf32f0b5e67e85b5436021b44fd8be18829 Mon Sep 17 00:00:00 2001 From: Mary Hipp Date: Tue, 5 Dec 2023 12:18:27 -0500 Subject: [PATCH] hide workflow library buttons if feature is disabled --- invokeai/frontend/web/src/app/types/invokeai.ts | 3 ++- .../flow/panels/TopCenterPanel/TopCenterPanel.tsx | 12 ++++++++++-- .../flow/panels/TopRightPanel/TopRightPanel.tsx | 6 +++++- 3 files changed, 17 insertions(+), 4 deletions(-) diff --git a/invokeai/frontend/web/src/app/types/invokeai.ts b/invokeai/frontend/web/src/app/types/invokeai.ts index 0fe7a36052..7e4cfb39aa 100644 --- a/invokeai/frontend/web/src/app/types/invokeai.ts +++ b/invokeai/frontend/web/src/app/types/invokeai.ts @@ -23,7 +23,8 @@ export type AppFeature = | 'resumeQueue' | 'prependQueue' | 'invocationCache' - | 'bulkDownload'; + | 'bulkDownload' + | 'workflowLibrary'; /** * A disable-able Stable Diffusion feature diff --git a/invokeai/frontend/web/src/features/nodes/components/flow/panels/TopCenterPanel/TopCenterPanel.tsx b/invokeai/frontend/web/src/features/nodes/components/flow/panels/TopCenterPanel/TopCenterPanel.tsx index 4ab6c0e9c3..4b515bcd0a 100644 --- a/invokeai/frontend/web/src/features/nodes/components/flow/panels/TopCenterPanel/TopCenterPanel.tsx +++ b/invokeai/frontend/web/src/features/nodes/components/flow/panels/TopCenterPanel/TopCenterPanel.tsx @@ -5,8 +5,12 @@ import UploadWorkflowButton from 'features/workflowLibrary/components/LoadWorkfl import ResetWorkflowEditorButton from 'features/workflowLibrary/components/ResetWorkflowButton'; import SaveWorkflowButton from 'features/workflowLibrary/components/SaveWorkflowButton'; import SaveWorkflowAsButton from 'features/workflowLibrary/components/SaveWorkflowAsButton'; +import { useFeatureStatus } from 'features/system/hooks/useFeatureStatus'; const TopCenterPanel = () => { + const isWorkflowLibraryEnabled = + useFeatureStatus('workflowLibrary').isFeatureEnabled; + return ( { > - - + {isWorkflowLibraryEnabled && ( + <> + + + + )} ); diff --git a/invokeai/frontend/web/src/features/nodes/components/flow/panels/TopRightPanel/TopRightPanel.tsx b/invokeai/frontend/web/src/features/nodes/components/flow/panels/TopRightPanel/TopRightPanel.tsx index dd039f31c1..a06b4a7656 100644 --- a/invokeai/frontend/web/src/features/nodes/components/flow/panels/TopRightPanel/TopRightPanel.tsx +++ b/invokeai/frontend/web/src/features/nodes/components/flow/panels/TopRightPanel/TopRightPanel.tsx @@ -2,11 +2,15 @@ import { Flex } from '@chakra-ui/react'; import WorkflowLibraryButton from 'features/workflowLibrary/components/WorkflowLibraryButton'; import { memo } from 'react'; import WorkflowEditorSettings from './WorkflowEditorSettings'; +import { useFeatureStatus } from 'features/system/hooks/useFeatureStatus'; const TopRightPanel = () => { + const isWorkflowLibraryEnabled = + useFeatureStatus('workflowLibrary').isFeatureEnabled; + return ( - + {isWorkflowLibraryEnabled && } );