github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): alerts display depending on current generation destination

Browse Source
This commit is contained in:
psychedelicious
2024-09-12 22:27:57 +10:00
parent 269db8ae19
commit dd3013d333
4 changed files with 36 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/frontend/web/src/features/system/components/ProgressBar.tsx
View File

@@ -1,15 +1,16 @@
import { Progress } from '@invoke-ai/ui-library';
import { useStore } from '@nanostores/react';
import { $isConnected } from 'app/hooks/useSocketIO';
import { useCurrentDestination } from 'features/queue/hooks/useCurrentDestination';
import { memo, useMemo } from 'react';
import { useTranslation } from 'react-i18next';
import { useGetCurrentQueueItemQuery, useGetQueueStatusQuery } from 'services/api/endpoints/queue';
import { useGetQueueStatusQuery } from 'services/api/endpoints/queue';
import { $lastProgressEvent } from 'services/events/setEventListeners';
const ProgressBar = () => {
const { t } = useTranslation();
const destination = useCurrentDestination();
const { data: queueStatus } = useGetQueueStatusQuery();
const currentQueueItem = useGetCurrentQueueItemQuery().data;
const isConnected = useStore($isConnected);
const lastProgressEvent = useStore($lastProgressEvent);
const value = useMemo(() => {
@@ -26,7 +27,7 @@ const ProgressBar = () => {
isIndeterminate={isConnected && Boolean(queueStatus?.queue.in_progress) && !lastProgressEvent}
h={2}
w="full"
colorScheme={currentQueueItem && currentQueueItem.destination === 'canvas' ? 'invokeGreen' : 'invokeBlue'}
colorScheme={destination === 'canvas' ? 'invokeGreen' : 'invokeBlue'}
/>
);
};