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

fix metadata references, add support for negative_conditioning syntax

Browse Source
This commit is contained in:
Mary Hipp 2023-05-25 14:47:16 -04:00 committed by psychedelicious
parent 3829ffbe66
commit 1e94d7739a
6 changed files with 30 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
invokeai/frontend/web/src/features/gallery/components/CurrentImageButtons.tsx
View File

@ -109,8 +109,9 @@ const currentImageButtonsSelector = createSelector(
isLightboxOpen, isLightboxOpen,
shouldHidePreview, shouldHidePreview,
image: selectedImage, image: selectedImage,
seed: selectedImage?.metadata?.invokeai?.node?.seed, seed: selectedImage?.metadata?.seed,
prompt: selectedImage?.metadata?.invokeai?.node?.prompt, prompt: selectedImage?.metadata?.positive_conditioning,
negativePrompt: selectedImage?.metadata?.negative_conditioning,
}; };
}, },
{ {
@ -245,13 +246,16 @@ const CurrentImageButtons = (props: CurrentImageButtonsProps) => {
); );
const handleUseSeed = useCallback(() => { const handleUseSeed = useCallback(() => {
recallSeed(image?.metadata?.invokeai?.node?.seed); recallSeed(image?.metadata?.seed);
}, [image, recallSeed]); }, [image, recallSeed]);
useHotkeys('s', handleUseSeed, [image]); useHotkeys('s', handleUseSeed, [image]);
const handleUsePrompt = useCallback(() => { const handleUsePrompt = useCallback(() => {
recallPrompt(image?.metadata?.invokeai?.node?.prompt); recallPrompt(
image?.metadata?.positive_conditioning,
image?.metadata?.negative_conditioning
);
}, [image, recallPrompt]); }, [image, recallPrompt]);
useHotkeys('p', handleUsePrompt, [image]); useHotkeys('p', handleUsePrompt, [image]);
@ -454,7 +458,7 @@ const CurrentImageButtons = (props: CurrentImageButtonsProps) => {
{t('parameters.copyImageToLink')} {t('parameters.copyImageToLink')}
</IAIButton> </IAIButton>
<Link download={true} href={getUrl(image?.url ?? '')}> <Link download={true} href={getUrl(image?.image_url ?? '')}>
<IAIButton leftIcon={<FaDownload />} size="sm" w="100%"> <IAIButton leftIcon={<FaDownload />} size="sm" w="100%">
{t('parameters.downloadImage')} {t('parameters.downloadImage')}
</IAIButton> </IAIButton>
@ -500,7 +504,7 @@ const CurrentImageButtons = (props: CurrentImageButtonsProps) => {
icon={<FaQuoteRight />} icon={<FaQuoteRight />}
tooltip={`${t('parameters.usePrompt')} (P)`} tooltip={`${t('parameters.usePrompt')} (P)`}
aria-label={`${t('parameters.usePrompt')} (P)`} aria-label={`${t('parameters.usePrompt')} (P)`}
isDisabled={!image?.metadata?.invokeai?.node?.prompt} isDisabled={!image?.metadata?.positive_conditioning}
onClick={handleUsePrompt} onClick={handleUsePrompt}
/> />
@ -508,7 +512,7 @@ const CurrentImageButtons = (props: CurrentImageButtonsProps) => {
icon={<FaSeedling />} icon={<FaSeedling />}
tooltip={`${t('parameters.useSeed')} (S)`} tooltip={`${t('parameters.useSeed')} (S)`}
aria-label={`${t('parameters.useSeed')} (S)`} aria-label={`${t('parameters.useSeed')} (S)`}
isDisabled={!image?.metadata?.invokeai?.node?.seed} isDisabled={!image?.metadata?.seed}
onClick={handleUseSeed} onClick={handleUseSeed}
/> />
@ -517,9 +521,8 @@ const CurrentImageButtons = (props: CurrentImageButtonsProps) => {
tooltip={`${t('parameters.useAll')} (A)`} tooltip={`${t('parameters.useAll')} (A)`}
aria-label={`${t('parameters.useAll')} (A)`} aria-label={`${t('parameters.useAll')} (A)`}
isDisabled={ isDisabled={
!['txt2img', 'img2img', 'inpaint'].includes( // not sure what this list should be
String(image?.metadata?.invokeai?.node?.type) !['t2l', 'l2l', 'inpaint'].includes(String(image?.metadata?.type))
)
} }
onClick={handleClickUseAllParameters} onClick={handleClickUseAllParameters}
/> />

5
invokeai/frontend/web/src/features/gallery/components/HoverableImage.tsx
View File

@ -155,7 +155,10 @@ const HoverableImage = memo((props: HoverableImageProps) => {
// Recall parameters handlers // Recall parameters handlers
const handleRecallPrompt = useCallback(() => { const handleRecallPrompt = useCallback(() => {
recallPrompt(image.metadata?.positive_conditioning); recallPrompt(
image.metadata?.positive_conditioning,
image.metadata?.negative_conditioning
);
}, [image, recallPrompt]); }, [image, recallPrompt]);
const handleRecallSeed = useCallback(() => { const handleRecallSeed = useCallback(() => {

12
invokeai/frontend/web/src/features/gallery/components/ImageMetaDataViewer/ImageMetadataViewer.tsx
View File

@ -8,29 +8,20 @@ import {
Text, Text,
Tooltip, Tooltip,
} from '@chakra-ui/react'; } from '@chakra-ui/react';
import * as InvokeAI from 'app/types/invokeai';
import { useAppDispatch } from 'app/store/storeHooks'; import { useAppDispatch } from 'app/store/storeHooks';
import { useGetUrl } from 'common/util/getUrl'; import { useGetUrl } from 'common/util/getUrl';
import promptToString from 'common/util/promptToString'; import promptToString from 'common/util/promptToString';
import { seedWeightsToString } from 'common/util/seedWeightPairs';
import useSetBothPrompts from 'features/parameters/hooks/usePrompt';
import { import {
setCfgScale, setCfgScale,
setHeight, setHeight,
setImg2imgStrength, setImg2imgStrength,
setNegativePrompt, setNegativePrompt,
setPerlin,
setPositivePrompt, setPositivePrompt,
setScheduler, setScheduler,
setSeamless,
setSeed, setSeed,
setSeedWeights,
setShouldFitToWidthHeight,
setSteps, setSteps,
setThreshold,
setWidth, setWidth,
} from 'features/parameters/store/generationSlice'; } from 'features/parameters/store/generationSlice';
import { setHiresFix } from 'features/parameters/store/postprocessingSlice';
import { setShouldShowImageDetails } from 'features/ui/store/uiSlice'; import { setShouldShowImageDetails } from 'features/ui/store/uiSlice';
import { memo } from 'react'; import { memo } from 'react';
import { useHotkeys } from 'react-hotkeys-hook'; import { useHotkeys } from 'react-hotkeys-hook';
@ -39,7 +30,6 @@ import { FaCopy } from 'react-icons/fa';
import { IoArrowUndoCircleOutline } from 'react-icons/io5'; import { IoArrowUndoCircleOutline } from 'react-icons/io5';
import { OverlayScrollbarsComponent } from 'overlayscrollbars-react'; import { OverlayScrollbarsComponent } from 'overlayscrollbars-react';
import { ImageDTO } from 'services/api'; import { ImageDTO } from 'services/api';
import { filter } from 'lodash-es';
import { Scheduler } from 'app/constants'; import { Scheduler } from 'app/constants';
type MetadataItemProps = { type MetadataItemProps = {
@ -126,8 +116,6 @@ const memoEqualityCheck = (
const ImageMetadataViewer = memo(({ image }: ImageMetadataViewerProps) => { const ImageMetadataViewer = memo(({ image }: ImageMetadataViewerProps) => {
const dispatch = useAppDispatch(); const dispatch = useAppDispatch();
const setBothPrompts = useSetBothPrompts();
useHotkeys('esc', () => { useHotkeys('esc', () => {
dispatch(setShouldShowImageDetails(false)); dispatch(setShouldShowImageDetails(false));
}); });

2
invokeai/frontend/web/src/features/nodes/util/graphBuilders/buildTextToImageGraph.ts
View File

@ -11,7 +11,7 @@ import { addNoiseNodes } from '../nodeBuilders/addNoiseNodes';
const POSITIVE_CONDITIONING = 'positive_conditioning'; const POSITIVE_CONDITIONING = 'positive_conditioning';
const NEGATIVE_CONDITIONING = 'negative_conditioning'; const NEGATIVE_CONDITIONING = 'negative_conditioning';
const TEXT_TO_LATENTS = 'text_to_latents'; const TEXT_TO_LATENTS = 'text_to_latents';
const LATENTS_TO_IMAGE = 'latnets_to_image'; const LATENTS_TO_IMAGE = 'latents_to_image';
/** /**
* Builds the Text to Image tab graph. * Builds the Text to Image tab graph.

13
invokeai/frontend/web/src/features/parameters/hooks/useParameters.ts
View File

@ -21,8 +21,8 @@ export const useParameters = () => {
* Sets prompt with toast * Sets prompt with toast
*/ */
const recallPrompt = useCallback( const recallPrompt = useCallback(
(prompt: unknown) => { (prompt: unknown, negativePrompt?: unknown) => {
if (!isString(prompt)) { if (!isString(prompt) || !isString(negativePrompt)) {
toaster({ toaster({
title: t('toast.promptNotSet'), title: t('toast.promptNotSet'),
description: t('toast.promptNotSetDesc'), description: t('toast.promptNotSetDesc'),
@ -33,7 +33,7 @@ export const useParameters = () => {
return; return;
} }
setBothPrompts(prompt); setBothPrompts(prompt, negativePrompt);
toaster({ toaster({
title: t('toast.promptSet'), title: t('toast.promptSet'),
status: 'info', status: 'info',
@ -112,12 +112,13 @@ export const useParameters = () => {
const recallAllParameters = useCallback( const recallAllParameters = useCallback(
(image: ImageDTO | undefined) => { (image: ImageDTO | undefined) => {
const type = image?.metadata?.type; const type = image?.metadata?.type;
if (['txt2img', 'img2img', 'inpaint'].includes(String(type))) { // not sure what this list should be
if (['t2l', 'l2l', 'inpaint'].includes(String(type))) {
dispatch(allParametersSet(image)); dispatch(allParametersSet(image));
if (image?.metadata?.type === 'img2img') { if (image?.metadata?.type === 'l2l') {
dispatch(setActiveTab('img2img')); dispatch(setActiveTab('img2img'));
} else if (image?.metadata?.type === 'txt2img') { } else if (image?.metadata?.type === 't2l') {
dispatch(setActiveTab('txt2img')); dispatch(setActiveTab('txt2img'));
} }

15
invokeai/frontend/web/src/features/parameters/hooks/usePrompt.ts
View File

@ -12,16 +12,11 @@ const useSetBothPrompts = () => {
const dispatch = useAppDispatch(); const dispatch = useAppDispatch();
return useCallback( return useCallback(
(inputPrompt: InvokeAI.Prompt) => { (inputPrompt: InvokeAI.Prompt, negativePrompt?: InvokeAI.Prompt) => {
const promptString = dispatch(setPositivePrompt(inputPrompt));
typeof inputPrompt === 'string' if (negativePrompt) {
? inputPrompt dispatch(setNegativePrompt(negativePrompt));
: promptToString(inputPrompt); }
const [prompt, negativePrompt] = getPromptAndNegative(promptString);
dispatch(setPositivePrompt(prompt));
dispatch(setNegativePrompt(negativePrompt));
}, },
[dispatch] [dispatch]
); );