From aed446f01307e7e51b1598825e3fd81ceb38feb8 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 12 Mar 2025 11:33:10 +1000 Subject: [PATCH] fix(ui): make the workflow load from file menu item work the same as the button in library Upload and save as instead of just upload as draft. --- .../WorkflowLibraryMenu/UploadWorkflowMenuItem.tsx | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/invokeai/frontend/web/src/features/workflowLibrary/components/WorkflowLibraryMenu/UploadWorkflowMenuItem.tsx b/invokeai/frontend/web/src/features/workflowLibrary/components/WorkflowLibraryMenu/UploadWorkflowMenuItem.tsx index a769f52909..cd2ceb50b6 100644 --- a/invokeai/frontend/web/src/features/workflowLibrary/components/WorkflowLibraryMenu/UploadWorkflowMenuItem.tsx +++ b/invokeai/frontend/web/src/features/workflowLibrary/components/WorkflowLibraryMenu/UploadWorkflowMenuItem.tsx @@ -1,4 +1,6 @@ import { MenuItem } from '@invoke-ai/ui-library'; +import { useWorkflowLibraryModal } from 'features/nodes/store/workflowLibraryModal'; +import { saveWorkflowAs } from 'features/workflowLibrary/components/SaveWorkflowAsDialog'; import { useLoadWorkflowFromFile } from 'features/workflowLibrary/hooks/useLoadWorkflowFromFile'; import { memo, useCallback, useRef } from 'react'; import { useDropzone } from 'react-dropzone'; @@ -8,7 +10,14 @@ import { PiUploadSimpleBold } from 'react-icons/pi'; const UploadWorkflowMenuItem = () => { const { t } = useTranslation(); const resetRef = useRef<() => void>(null); - const loadWorkflowFromFile = useLoadWorkflowFromFile({ resetRef }); + const workflowLibraryModal = useWorkflowLibraryModal(); + const loadWorkflowFromFile = useLoadWorkflowFromFile({ + resetRef, + onSuccess: (workflow) => { + workflowLibraryModal.close(); + saveWorkflowAs(workflow); + }, + }); const onDropAccepted = useCallback( (files: File[]) => {