feat(ui): regional prompting layout, styling

This commit is contained in:
psychedelicious 2024-04-12 17:51:19 +10:00 committed by Kent Keirsey
parent e7523bd1d9
commit c1aae0815d
5 changed files with 20 additions and 37 deletions

View File

@ -1,19 +0,0 @@
import type { FlexProps } from '@invoke-ai/ui-library';
import { Flex, forwardRef } from '@invoke-ai/ui-library';
import { useMemo } from 'react';
import type { RgbaColor, RgbColor } from 'react-colorful';
type Props = FlexProps & {
previewColor: RgbColor | RgbaColor;
};
export const ColorPreview = forwardRef((props: Props, ref) => {
const { previewColor, ...rest } = props;
const colorString = useMemo(() => {
if ('a' in previewColor) {
return `rgba(${previewColor.r}, ${previewColor.g}, ${previewColor.b}, ${previewColor.a ?? 1})`;
}
return `rgba(${previewColor.r}, ${previewColor.g}, ${previewColor.b}, 1)`;
}, [previewColor]);
return <Flex ref={ref} w="full" h="full" borderRadius="base" backgroundColor={colorString} {...rest} />;
});

View File

@ -1,11 +1,11 @@
import { Popover, PopoverBody, PopoverContent, PopoverTrigger } from '@invoke-ai/ui-library';
import { IconButton, Popover, PopoverBody, PopoverContent, PopoverTrigger } from '@invoke-ai/ui-library';
import { useAppDispatch } from 'app/store/storeHooks';
import { ColorPreview } from 'common/components/ColorPreview';
import RgbColorPicker from 'common/components/RgbColorPicker';
import { useLayer } from 'features/regionalPrompts/hooks/layerStateHooks';
import { promptRegionLayerColorChanged } from 'features/regionalPrompts/store/regionalPromptsSlice';
import { useCallback } from 'react';
import type { RgbColor } from 'react-colorful';
import { PiEyedropperBold } from 'react-icons/pi';
type Props = {
id: string;
@ -23,7 +23,7 @@ export const LayerColorPicker = ({ id }: Props) => {
return (
<Popover isLazy>
<PopoverTrigger>
<ColorPreview previewColor={layer.color} />
<IconButton aria-label="color picker" size="sm" borderRadius="base" icon={<PiEyedropperBold />} />
</PopoverTrigger>
<PopoverContent>
<PopoverBody minH={64}>

View File

@ -1,11 +1,12 @@
import { Flex } from '@invoke-ai/ui-library';
import { Flex, Spacer } from '@invoke-ai/ui-library';
import { useAppDispatch, useAppSelector } from 'app/store/storeHooks';
import { rgbColorToString } from 'features/canvas/util/colorToString';
import { LayerColorPicker } from 'features/regionalPrompts/components/LayerColorPicker';
import { LayerMenu } from 'features/regionalPrompts/components/LayerMenu';
import { LayerVisibilityToggle } from 'features/regionalPrompts/components/LayerVisibilityToggle';
import { RegionalPromptsPrompt } from 'features/regionalPrompts/components/RegionalPromptsPrompt';
import { layerSelected } from 'features/regionalPrompts/store/regionalPromptsSlice';
import { useCallback, useMemo } from 'react';
import { useCallback } from 'react';
type Props = {
id: string;
@ -13,19 +14,24 @@ type Props = {
export const LayerListItem = ({ id }: Props) => {
const dispatch = useAppDispatch();
const selectedLayer = useAppSelector((s) => s.regionalPrompts.selectedLayer);
const bg = useMemo(() => (selectedLayer === id ? 'invokeBlue.500' : 'transparent'), [selectedLayer, id]);
const color = useAppSelector((s) => {
const color = s.regionalPrompts.layers.find((l) => l.id === id)?.color;
if (color) {
return rgbColorToString(color);
}
return 'base.700';
});
const onClickCapture = useCallback(() => {
// Must be capture so that the layer is selected before deleting/resetting/etc
dispatch(layerSelected(id));
}, [dispatch, id]);
return (
<Flex gap={2} onClickCapture={onClickCapture}>
<Flex w={2} borderRadius="base" bg={bg} flexShrink={0} py={4} />
<Flex flexDir="column" gap={2}>
<Flex gap={2} onClickCapture={onClickCapture} bg={color} borderRadius="base" p="1px" ps={2}>
<Flex flexDir="column" gap={2} w="full" bg="base.850" borderRadius="base" p={2}>
<Flex gap={2}>
<LayerColorPicker id={id} />
<LayerVisibilityToggle id={id} />
<Spacer />
<LayerMenu id={id} />
</Flex>
<RegionalPromptsPrompt layerId={id} />

View File

@ -1,13 +1,11 @@
import { Box, Textarea } from '@invoke-ai/ui-library';
import { useAppDispatch, useAppSelector } from 'app/store/storeHooks';
import { ShowDynamicPromptsPreviewButton } from 'features/dynamicPrompts/components/ShowDynamicPromptsPreviewButton';
import { useAppDispatch } from 'app/store/storeHooks';
import { PromptOverlayButtonWrapper } from 'features/parameters/components/Prompts/PromptOverlayButtonWrapper';
import { AddPromptTriggerButton } from 'features/prompt/AddPromptTriggerButton';
import { PromptPopover } from 'features/prompt/PromptPopover';
import { usePrompt } from 'features/prompt/usePrompt';
import { useLayerPrompt } from 'features/regionalPrompts/hooks/layerStateHooks';
import { promptChanged } from 'features/regionalPrompts/store/regionalPromptsSlice';
import { SDXLConcatButton } from 'features/sdxl/components/SDXLPrompts/SDXLConcatButton';
import { memo, useCallback, useRef } from 'react';
import type { HotkeyCallback } from 'react-hotkeys-hook';
import { useHotkeys } from 'react-hotkeys-hook';
@ -20,7 +18,6 @@ type Props = {
export const RegionalPromptsPrompt = memo((props: Props) => {
const prompt = useLayerPrompt(props.layerId);
const dispatch = useAppDispatch();
const baseModel = useAppSelector((s) => s.generation.model)?.base;
const textareaRef = useRef<HTMLTextAreaElement>(null);
const { t } = useTranslation();
const _onChange = useCallback(
@ -46,7 +43,7 @@ export const RegionalPromptsPrompt = memo((props: Props) => {
return (
<PromptPopover isOpen={isOpen} onClose={onClose} onSelect={onSelect} width={textareaRef.current?.clientWidth}>
<Box pos="relative">
<Box pos="relative" w="full">
<Textarea
id="prompt"
name="prompt"
@ -62,8 +59,6 @@ export const RegionalPromptsPrompt = memo((props: Props) => {
/>
<PromptOverlayButtonWrapper>
<AddPromptTriggerButton isOpen={isOpen} onOpen={onOpen} />
{baseModel === 'sdxl' && <SDXLConcatButton />}
<ShowDynamicPromptsPreviewButton />
</PromptOverlayButtonWrapper>
</Box>
</PromptPopover>

View File

@ -40,7 +40,8 @@ export const RegionalPromptsStage: React.FC = memo(() => {
}, []);
const sx = useMemo(
() => ({
border: '1px solid cyan',
borderRadius: 'base',
borderWidth: 1,
cursor: tool === 'move' ? 'default' : 'none',
}),
[tool]