add project as a workflow category in the front-end

This commit is contained in:
Mary Hipp 2024-01-23 09:49:50 -05:00 committed by psychedelicious
parent b4cf5496b6
commit 3dc13221d8
5 changed files with 630 additions and 170 deletions

View File

@ -1697,6 +1697,7 @@
"workflowLibrary": "Library",
"userWorkflows": "My Workflows",
"defaultWorkflows": "Default Workflows",
"projectWorkflows": "Project Workflows",
"openWorkflow": "Open Workflow",
"uploadWorkflow": "Load from File",
"deleteWorkflow": "Delete Workflow",
@ -1709,6 +1710,7 @@
"workflowSaved": "Workflow Saved",
"noRecentWorkflows": "No Recent Workflows",
"noUserWorkflows": "No User Workflows",
"noWorkflows": "No Workflows",
"noSystemWorkflows": "No System Workflows",
"problemLoading": "Problem Loading Workflows",
"loading": "Loading Workflows",

View File

@ -15,7 +15,7 @@ export type XYPosition = z.infer<typeof zXYPosition>;
export const zDimension = z.number().gt(0).nullish();
export type Dimension = z.infer<typeof zDimension>;
export const zWorkflowCategory = z.enum(['user', 'default']);
export const zWorkflowCategory = z.enum(['user', 'default', 'project']);
export type WorkflowCategory = z.infer<typeof zWorkflowCategory>;
// #endregion

View File

@ -13,6 +13,8 @@ import {
InputRightElement,
Spacer,
} from '@invoke-ai/ui';
import { useStore } from '@nanostores/react';
import { $projectId } from 'app/store/nanostores/projectId';
import {
IAINoContentFallback,
IAINoContentFallbackWithSpinner,
@ -62,6 +64,7 @@ const WorkflowLibraryList = () => {
const [order_by, setOrderBy] = useState<WorkflowRecordOrderBy>('opened_at');
const [direction, setDirection] = useState<SQLiteDirection>('ASC');
const [debouncedQuery] = useDebounce(query, 500);
const projectId = useStore($projectId);
const queryArg = useMemo<Parameters<typeof useListWorkflowsQuery>[0]>(() => {
if (category === 'user') {
@ -142,13 +145,8 @@ const WorkflowLibraryList = () => {
[]
);
const handleSetUserCategory = useCallback(() => {
setCategory('user');
setPage(0);
}, []);
const handleSetDefaultCategory = useCallback(() => {
setCategory('default');
const handleSetCategory = useCallback((category: WorkflowCategory) => {
setCategory(category);
setPage(0);
}, []);
@ -158,21 +156,31 @@ const WorkflowLibraryList = () => {
<ButtonGroup>
<Button
variant={category === 'user' ? undefined : 'ghost'}
onClick={handleSetUserCategory}
onClick={handleSetCategory.bind(null, 'user')}
isChecked={category === 'user'}
>
{t('workflows.userWorkflows')}
</Button>
<Button
variant={category === 'default' ? undefined : 'ghost'}
onClick={handleSetDefaultCategory}
isChecked={category === 'default'}
>
{t('workflows.defaultWorkflows')}
</Button>
{projectId ? (
<Button
variant={category === 'project' ? undefined : 'ghost'}
onClick={handleSetCategory.bind(null, 'project')}
isChecked={category === 'project'}
>
{t('workflows.projectWorkflows')}
</Button>
) : (
<Button
variant={category === 'default' ? undefined : 'ghost'}
onClick={handleSetCategory.bind(null, 'default')}
isChecked={category === 'default'}
>
{t('workflows.defaultWorkflows')}
</Button>
)}
</ButtonGroup>
<Spacer />
{category === 'user' && (
{category !== 'default' && (
<>
<FormControl isDisabled={isFetching} w={64} minW={56}>
<FormLabel>{t('common.orderBy')}</FormLabel>
@ -228,7 +236,7 @@ const WorkflowLibraryList = () => {
</Flex>
</ScrollableContent>
) : (
<IAINoContentFallback label={t('workflows.noUserWorkflows')} />
<IAINoContentFallback label={t('workflows.noWorkflows')} />
)}
<Divider />
{data && (

View File

@ -52,7 +52,7 @@ const WorkflowLibraryListItem = ({ workflowDTO }: Props) => {
{workflowDTO.name || t('workflows.unnamedWorkflow')}
</Heading>
<Spacer />
{workflowDTO.category === 'user' && (
{workflowDTO.category !== 'default' && (
<Text
fontSize="sm"
variant="subtext"
@ -81,7 +81,7 @@ const WorkflowLibraryListItem = ({ workflowDTO }: Props) => {
</Text>
)}
<Spacer />
{workflowDTO.category === 'user' && (
{workflowDTO.category !== 'default' && (
<Text
fontSize="sm"
variant="subtext"
@ -104,7 +104,7 @@ const WorkflowLibraryListItem = ({ workflowDTO }: Props) => {
>
{t('common.load')}
</Button>
{workflowDTO.category === 'user' && (
{workflowDTO.category !== 'default' && (
<Button
flexShrink={0}
colorScheme="error"

File diff suppressed because one or more lines are too long