diff --git a/invokeai/frontend/web/src/app/hooks/useStudioInitAction.ts b/invokeai/frontend/web/src/app/hooks/useStudioInitAction.ts index 6cc83854d4..db486b5cee 100644 --- a/invokeai/frontend/web/src/app/hooks/useStudioInitAction.ts +++ b/invokeai/frontend/web/src/app/hooks/useStudioInitAction.ts @@ -15,7 +15,7 @@ import { $isWorkflowLibraryModalOpen } from 'features/nodes/store/workflowLibrar import { $isStylePresetsMenuOpen, activeStylePresetIdChanged } from 'features/stylePresets/store/stylePresetSlice'; import { toast } from 'features/toast/toast'; import { activeTabCanvasRightPanelChanged, setActiveTab } from 'features/ui/store/uiSlice'; -import { useLoadWorkflowFromLibrary } from 'features/workflowLibrary/hooks/useLoadWorkflowFromLibrary'; +import { useLoadWorkflowWithDialog } from 'features/workflowLibrary/components/LoadWorkflowConfirmationAlertDialog'; import { atom } from 'nanostores'; import { useCallback, useEffect } from 'react'; import { useTranslation } from 'react-i18next'; @@ -57,7 +57,7 @@ export const useStudioInitAction = (action?: StudioInitAction) => { const { t } = useTranslation(); const didParseOpenAPISchema = useStore($hasTemplates); const store = useAppStore(); - const loadWorkflowFromLibrary = useLoadWorkflowFromLibrary(); + const loadWorkflowWithDialog = useLoadWorkflowWithDialog(); const handleSendToCanvas = useCallback( async (imageName: string) => { @@ -113,13 +113,15 @@ export const useStudioInitAction = (action?: StudioInitAction) => { const handleLoadWorkflow = useCallback( async (workflowId: string) => { // This shows a toast - await loadWorkflowFromLibrary(workflowId, { + await loadWorkflowWithDialog({ + type: 'library', + data: workflowId, onSuccess: () => { store.dispatch(setActiveTab('workflows')); }, }); }, - [loadWorkflowFromLibrary, store] + [loadWorkflowWithDialog, store] ); const handleSelectStylePreset = useCallback(