mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-01-15 06:18:03 -05:00
feat(ui): tweak staging image display
This commit is contained in:
@@ -62,7 +62,7 @@ export const QueueItemPreviewMini = memo(({ item, isSelected, number }: Props) =
|
||||
>
|
||||
<QueueItemStatusLabel status={item.status} position="absolute" margin="auto" />
|
||||
{imageDTO && <DndImage imageDTO={imageDTO} onLoad={onLoad} asThumbnail />}
|
||||
{!imageLoaded && <QueueItemProgressImage itemId={item.item_id} position="absolute" />}
|
||||
{!imageLoaded && <QueueItemProgressImage itemId={item.item_id} position="absolute" withBg />}
|
||||
<QueueItemNumber number={number} position="absolute" top={0} left={1} />
|
||||
<QueueItemCircularProgress itemId={item.item_id} status={item.status} position="absolute" top={1} right={2} />
|
||||
</Flex>
|
||||
|
||||
@@ -3,14 +3,18 @@ import { Flex, Image } from '@invoke-ai/ui-library';
|
||||
import { useCanvasSessionContext, useProgressData } from 'features/controlLayers/components/SimpleSession/context';
|
||||
import { memo } from 'react';
|
||||
|
||||
type Props = { itemId: number } & ImageProps;
|
||||
type Props = { itemId: number; withBg?: boolean } & ImageProps;
|
||||
|
||||
export const QueueItemProgressImage = memo(({ itemId, ...rest }: Props) => {
|
||||
export const QueueItemProgressImage = memo(({ itemId, withBg, ...rest }: Props) => {
|
||||
const ctx = useCanvasSessionContext();
|
||||
const { progressImage } = useProgressData(ctx.$progressData, itemId);
|
||||
|
||||
if (!progressImage) {
|
||||
return <Flex w="full" h="full" bg="base.900" alignItems="center" justifyContent="center" />;
|
||||
if (withBg) {
|
||||
return <Flex w="full" h="full" bg="base.900" alignItems="center" justifyContent="center" />;
|
||||
} else {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
return (
|
||||
|
||||
@@ -77,7 +77,8 @@ export const DndImage = memo(
|
||||
ref={ref}
|
||||
src={asThumbnail ? imageDTO.thumbnail_url : imageDTO.image_url}
|
||||
fallbackSrc={asThumbnail ? undefined : imageDTO.thumbnail_url}
|
||||
w={imageDTO.width}
|
||||
width={imageDTO.width}
|
||||
height={imageDTO.height}
|
||||
sx={sx}
|
||||
data-is-dragging={isDragging}
|
||||
{...rest}
|
||||
|
||||
Reference in New Issue
Block a user