github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

add actions for workflow library (#5669)

Browse Source 
Co-authored-by: Mary Hipp <maryhipp@Marys-MacBook-Air.local>
This commit is contained in:
Mary Hipp Rogers
2024-02-07 09:14:54 -05:00
committed by GitHub
parent 79ae9c4e64
commit 800c481515
7 changed files with 43 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
invokeai/frontend/web/src/features/workflowLibrary/hooks/useDownloadWorkflow.ts
View File

@@ -1,19 +1,25 @@
import { useAppDispatch } from 'app/store/storeHooks';
import { $builtWorkflow } from 'features/nodes/hooks/useWorkflowWatcher';
const downloadWorkflow = () => {
const workflow = $builtWorkflow.get();
if (!workflow) {
return;
}
const blob = new Blob([JSON.stringify(workflow, null, 2)]);
const a = document.createElement('a');
a.href = URL.createObjectURL(blob);
a.download = `${workflow.name || 'My Workflow'}.json`;
document.body.appendChild(a);
a.click();
a.remove();
};
import { workflowDownloaded } from 'features/workflowLibrary/store/actions';
import { useCallback } from 'react';
export const useDownloadWorkflow = () => {
const dispatch = useAppDispatch();
const downloadWorkflow = useCallback(() => {
const workflow = $builtWorkflow.get();
if (!workflow) {
return;
}
const blob = new Blob([JSON.stringify(workflow, null, 2)]);
const a = document.createElement('a');
a.href = URL.createObjectURL(blob);
a.download = `${workflow.name || 'My Workflow'}.json`;
document.body.appendChild(a);
a.click();
a.remove();
dispatch(workflowDownloaded());
}, [dispatch]);
return downloadWorkflow;
};

2
invokeai/frontend/web/src/features/workflowLibrary/hooks/useLoadWorkflowFromFile.tsx
View File

@@ -3,6 +3,7 @@ import { useAppDispatch } from 'app/store/storeHooks';
import { workflowLoadRequested } from 'features/nodes/store/actions';
import { addToast } from 'features/system/store/systemSlice';
import { makeToast } from 'features/system/util/makeToast';
import { workflowLoadedFromFile } from 'features/workflowLibrary/store/actions';
import type { RefObject } from 'react';
import { useCallback } from 'react';
import { useTranslation } from 'react-i18next';
@@ -29,6 +30,7 @@ export const useLoadWorkflowFromFile: UseLoadWorkflowFromFile = ({ resetRef }) =
try {
const parsedJSON = JSON.parse(String(rawJSON));
dispatch(workflowLoadRequested({ workflow: parsedJSON, asCopy: true }));
dispatch(workflowLoadedFromFile());
} catch (e) {
// There was a problem reading the file
logger.error(t('nodes.unableToLoadWorkflow'));

2
invokeai/frontend/web/src/features/workflowLibrary/hooks/useSaveWorkflow.ts
View File

@@ -4,6 +4,7 @@ import { useAppDispatch } from 'app/store/storeHooks';
import { $builtWorkflow } from 'features/nodes/hooks/useWorkflowWatcher';
import { workflowIDChanged, workflowSaved } from 'features/nodes/store/workflowSlice';
import type { WorkflowV2 } from 'features/nodes/types/workflow';
import { workflowUpdated } from 'features/workflowLibrary/store/actions';
import { useCallback, useRef } from 'react';
import { useTranslation } from 'react-i18next';
import { useCreateWorkflowMutation, useUpdateWorkflowMutation, workflowsApi } from 'services/api/endpoints/workflows';
@@ -41,6 +42,7 @@ export const useSaveLibraryWorkflow: UseSaveLibraryWorkflow = () => {
try {
if (isWorkflowWithID(workflow)) {
await updateWorkflow(workflow).unwrap();
dispatch(workflowUpdated());
} else {
const data = await createWorkflow(workflow).unwrap();
dispatch(workflowIDChanged(data.workflow.id));

2
invokeai/frontend/web/src/features/workflowLibrary/hooks/useSaveWorkflowAs.ts
View File

@@ -9,6 +9,7 @@ import {
workflowSaved,
} from 'features/nodes/store/workflowSlice';
import type { WorkflowCategory } from 'features/nodes/types/workflow';
import { newWorkflowSaved } from 'features/workflowLibrary/store/actions';
import { useCallback, useRef } from 'react';
import { useTranslation } from 'react-i18next';
import { useCreateWorkflowMutation, workflowsApi } from 'services/api/endpoints/workflows';
@@ -56,6 +57,7 @@ export const useSaveWorkflowAs: UseSaveWorkflowAs = () => {
dispatch(workflowNameChanged(data.workflow.name));
dispatch(workflowCategoryChanged(data.workflow.meta.category));
dispatch(workflowSaved());
dispatch(newWorkflowSaved({ category }));
onSuccess && onSuccess();
toast.update(toastRef.current, {

10
invokeai/frontend/web/src/features/workflowLibrary/store/actions.ts Normal file
View File

@@ -0,0 +1,10 @@
import { createAction } from '@reduxjs/toolkit';
import type { WorkflowCategory } from 'features/nodes/types/workflow';
export const workflowDownloaded = createAction('workflowLibrary/workflowDownloaded');
export const workflowLoadedFromFile = createAction('workflowLibrary/workflowLoadedFromFile');
export const newWorkflowSaved = createAction<{ category: WorkflowCategory }>('workflowLibrary/newWorkflowSaved');
export const workflowUpdated = createAction('workflowLibrary/workflowUpdated');