Skip to content

Commit 1072b74

Browse files
fix(ui): edge cases in starter models search
1 parent 46dc8c6 commit 1072b74

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StartModelsResultItem.tsx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ export const StarterModelsResultItem = ({ result }: Props) => {
3030
<Flex alignItems="center" justifyContent="space-between" w="100%" gap={3}>
3131
<Flex fontSize="sm" flexDir="column">
3232
<Flex gap={3}>
33-
<Badge h="min-content">{result.type.replace('_', ' ')}</Badge>
33+
<Badge h="min-content">{result.type.replaceAll('_', ' ')}</Badge>
3434
<ModelBaseBadge base={result.base} />
3535
<Text fontWeight="semibold">{result.name}</Text>
3636
</Flex>

invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StarterModelsResults.tsx

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,14 +18,18 @@ export const StarterModelsResults = ({ results }: StarterModelsResultsProps) =>
1818

1919
const filteredResults = useMemo(() => {
2020
return results.filter((result) => {
21+
const trimmedSearchTerm = searchTerm.trim().toLowerCase();
2122
const name = result.name.toLowerCase();
2223
const type = result.type.toLowerCase();
23-
return name.includes(searchTerm.toLowerCase()) || type.includes(searchTerm.toLowerCase());
24+
const description = result.description.toLowerCase();
25+
return (
26+
name.includes(trimmedSearchTerm) || type.includes(trimmedSearchTerm) || description.includes(trimmedSearchTerm)
27+
);
2428
});
2529
}, [results, searchTerm]);
2630

2731
const handleSearch: ChangeEventHandler<HTMLInputElement> = useCallback((e) => {
28-
setSearchTerm(e.target.value.trim());
32+
setSearchTerm(e.target.value);
2933
}, []);
3034

3135
const clearSearch = useCallback(() => {

0 commit comments

Comments
 (0)