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(ui): prompt preview not working preset starts with [prompt]

Browse Source
This commit is contained in:
psychedelicious 2024-08-14 12:21:38 +10:00
parent 29d6f48901
commit 41b2d5d013
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
invokeai/frontend/web/src/features/stylePresets/util/getViewModeChunks.tsx
View File

@ -1,13 +1,15 @@
import { PRESET_PLACEHOLDER } from 'features/stylePresets/hooks/usePresetModifiedPrompts'; import { PRESET_PLACEHOLDER } from 'features/stylePresets/hooks/usePresetModifiedPrompts';
export const getViewModeChunks = (currentPrompt: string, presetPrompt?: string): [string, string, string] => { export const getViewModeChunks = (currentPrompt: string, presetPrompt?: string): [string, string, string] => {
if (!presetPrompt || !presetPrompt.length) { if (!presetPrompt || !presetPrompt.length) {
return ['', currentPrompt, '']; return ['', currentPrompt, ''];
} }
const [firstPart, ...remainingParts] = presetPrompt.split(PRESET_PLACEHOLDER);
const chunks = [firstPart, remainingParts.join(PRESET_PLACEHOLDER)]; const [before, after] = presetPrompt.split(PRESET_PLACEHOLDER, 2);
if (chunks.filter((chunk) => chunk !== '').length === 1) {
return ['', currentPrompt, chunks[0] ?? '']; if (!before || !after) {
} else { return ['', `${currentPrompt} `, presetPrompt];
return [chunks[0] ?? '', currentPrompt, chunks[1] ?? ''];
} }
return [before ?? '', currentPrompt, after ?? ''];
}; };