diff --git a/invokeai/frontend/web/src/features/workflowLibrary/hooks/useLoadWorkflowFromLibrary.ts b/invokeai/frontend/web/src/features/workflowLibrary/hooks/useLoadWorkflowFromLibrary.ts index c096cc8e60..257e2fc833 100644 --- a/invokeai/frontend/web/src/features/workflowLibrary/hooks/useLoadWorkflowFromLibrary.ts +++ b/invokeai/frontend/web/src/features/workflowLibrary/hooks/useLoadWorkflowFromLibrary.ts @@ -1,4 +1,6 @@ import { useToast } from '@invoke-ai/ui-library'; +import { useAppDispatch } from 'app/store/storeHooks'; +import { workflowIsPublishedChanged } from 'features/nodes/store/workflowSlice'; import type { WorkflowV3 } from 'features/nodes/types/workflow'; import { useValidateAndLoadWorkflow } from 'features/workflowLibrary/hooks/useValidateAndLoadWorkflow'; import { useCallback } from 'react'; @@ -17,6 +19,7 @@ export const useLoadWorkflowFromLibrary = () => { const validateAndLoadWorkflow = useValidateAndLoadWorkflow(); const [getWorkflow] = useLazyGetWorkflowQuery(); const [updateOpenedAt] = useUpdateOpenedAtMutation(); + const dispatch = useAppDispatch(); const loadWorkflowFromLibrary = useCallback( async ( workflowId: string, @@ -36,6 +39,7 @@ export const useLoadWorkflowFromLibrary = () => { onError?.(); return; } + dispatch(workflowIsPublishedChanged(res.is_published ?? false)); updateOpenedAt({ workflow_id: workflowId }); onSuccess?.(validatedWorkflow); } catch { @@ -50,7 +54,7 @@ export const useLoadWorkflowFromLibrary = () => { onCompleted?.(); } }, - [getWorkflow, validateAndLoadWorkflow, updateOpenedAt, toast, t] + [getWorkflow, validateAndLoadWorkflow, updateOpenedAt, toast, t, dispatch] ); return loadWorkflowFromLibrary;