mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
added status to import queue model
This commit is contained in:
parent
ea364bdf82
commit
68f24d9f0d
@ -1,9 +1,4 @@
|
|||||||
import {
|
import { Box, Flex, IconButton, Progress, Text } from '@invoke-ai/ui-library';
|
||||||
Flex,
|
|
||||||
IconButton,
|
|
||||||
Progress,
|
|
||||||
Text,
|
|
||||||
} from '@invoke-ai/ui-library';
|
|
||||||
import { useAppDispatch } from 'app/store/storeHooks';
|
import { useAppDispatch } from 'app/store/storeHooks';
|
||||||
import { addToast } from 'features/system/store/systemSlice';
|
import { addToast } from 'features/system/store/systemSlice';
|
||||||
import { makeToast } from 'features/system/util/makeToast';
|
import { makeToast } from 'features/system/util/makeToast';
|
||||||
@ -50,12 +45,31 @@ export const ImportQueueModel = (props: ModelListItemProps) => {
|
|||||||
});
|
});
|
||||||
}, [deleteImportModel, model, dispatch]);
|
}, [deleteImportModel, model, dispatch]);
|
||||||
|
|
||||||
|
const formatBytes = (bytes: number) => {
|
||||||
|
const units = ['b', 'kb', 'mb', 'gb', 'tb'];
|
||||||
|
|
||||||
|
let i = 0;
|
||||||
|
|
||||||
|
for (i; bytes >= 1024 && i < 4; i++) {
|
||||||
|
bytes /= 1024;
|
||||||
|
}
|
||||||
|
|
||||||
|
return `${bytes.toFixed(2)} ${units[i]}`;
|
||||||
|
};
|
||||||
|
|
||||||
const progressValue = useMemo(() => {
|
const progressValue = useMemo(() => {
|
||||||
return (model.bytes / model.total_bytes) * 100;
|
return (model.bytes / model.total_bytes) * 100;
|
||||||
}, [model.bytes, model.total_bytes]);
|
}, [model.bytes, model.total_bytes]);
|
||||||
|
|
||||||
|
const progressString = useMemo(() => {
|
||||||
|
if (model.status !== 'downloading') {
|
||||||
|
return '--';
|
||||||
|
}
|
||||||
|
return `${formatBytes(model.bytes)} / ${formatBytes(model.total_bytes)}`;
|
||||||
|
}, [model.bytes, model.total_bytes, model.status]);
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<Flex gap="3" w="full" alignItems="center" textAlign="center">
|
<Flex gap="2" w="full" alignItems="center" textAlign="center">
|
||||||
<Text w="20%" whiteSpace="nowrap" overflow="hidden" textOverflow="ellipsis">
|
<Text w="20%" whiteSpace="nowrap" overflow="hidden" textOverflow="ellipsis">
|
||||||
{model.source.repo_id}
|
{model.source.repo_id}
|
||||||
</Text>
|
</Text>
|
||||||
@ -66,7 +80,12 @@ export const ImportQueueModel = (props: ModelListItemProps) => {
|
|||||||
h={2}
|
h={2}
|
||||||
w="50%"
|
w="50%"
|
||||||
/>
|
/>
|
||||||
<Text w="20%">{model.status}</Text>
|
<Text minW="20%" fontSize="xs" w="20%">
|
||||||
|
{progressString}
|
||||||
|
</Text>
|
||||||
|
<Text w="15%">{model.status[0].toUpperCase() +
|
||||||
|
model.status.slice(1)}</Text>
|
||||||
|
<Box w="10%">
|
||||||
{(model.status === 'downloading' || model.status === 'waiting') && (
|
{(model.status === 'downloading' || model.status === 'waiting') && (
|
||||||
<IconButton
|
<IconButton
|
||||||
isRound={true}
|
isRound={true}
|
||||||
@ -77,6 +96,7 @@ export const ImportQueueModel = (props: ModelListItemProps) => {
|
|||||||
onClick={handleDeleteModelImport}
|
onClick={handleDeleteModelImport}
|
||||||
/>
|
/>
|
||||||
)}
|
)}
|
||||||
|
</Box>
|
||||||
</Flex>
|
</Flex>
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
Loading…
Reference in New Issue
Block a user