Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): make generation mode calculation more granular

Browse Source
This commit is contained in:
psychedelicious
2023-07-24 18:16:15 +10:00
parent 28031ead70
commit 61fa960a18
5 changed files with 103 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
invokeai/frontend/web/src/features/parameters/components/Parameters/Canvas/GenerationModeStatusText.tsx
View File

@ -1,9 +1,5 @@
import { Box } from '@chakra-ui/react';
import { useAppDispatch, useAppSelector } from 'app/store/storeHooks';
import { generationModeChanged } from 'features/canvas/store/canvasSlice';
import { getCanvasData } from 'features/canvas/util/getCanvasData';
import { getCanvasGenerationMode } from 'features/canvas/util/getCanvasGenerationMode';
import { useDebounce } from 'react-use';
import { useCanvasGenerationMode } from 'features/canvas/hooks/useCanvasGenerationMode';
const GENERATION_MODE_NAME_MAP = {
txt2img: 'Text to Image',
@ -12,38 +8,8 @@ const GENERATION_MODE_NAME_MAP = {
outpaint: 'Inpaint',
};
export const useGenerationMode = () => {
const dispatch = useAppDispatch();
const canvasState = useAppSelector((state) => state.canvas);
useDebounce(
async () => {
// Build canvas blobs
const canvasBlobsAndImageData = await getCanvasData(canvasState);
if (!canvasBlobsAndImageData) {
return;
}
const { baseImageData, maskImageData } = canvasBlobsAndImageData;
// Determine the generation mode
const generationMode = getCanvasGenerationMode(
baseImageData,
maskImageData
);
dispatch(generationModeChanged(generationMode));
},
1000,
[dispatch, canvasState, generationModeChanged]
);
};
const GenerationModeStatusText = () => {
const generationMode = useAppSelector((state) => state.canvas.generationMode);
useGenerationMode();
const generationMode = useCanvasGenerationMode();
return (
<Box>