Merge pull request #151 from IDEA-Research/fix/preview_grounding

cefeng06 · web-flow · commit 1f4c99ac461f · 2024-04-10T10:53:35.000+08:00
fix(app): preview grounding annotations with filter
diff --git a/packages/components/src/Annotator/preview.tsx b/packages/components/src/Annotator/preview.tsx
@@ -358,7 +358,7 @@ const Preview: React.FC<PreviewProps> = (props) => {
   }
 
   const getHighlightWords = () => {
-    const objects = list[current].objects;
+    const objects = objectsFilter ? objectsFilter(list[current]) : list[current].objects;
     return categories
       .filter((category) => objects.find((obj: any) => obj.categoryId === category.id))
       .map((category) => {

Original file line number	Diff line number	Diff line change
`@@ -358,7 +358,7 @@ const Preview: React.FC<PreviewProps> = (props) => {`
`358`	`358`	`}`
`359`	`359`
`360`	`360`	`const getHighlightWords = () => {`
`361`		`- const objects = list[current].objects;`
	`361`	`+ const objects = objectsFilter ? objectsFilter(list[current]) : list[current].objects;`
`362`	`362`	`return categories`
`363`	`363`	`.filter((category) => objects.find((obj: any) => obj.categoryId === category.id))`
`364`	`364`	`.map((category) => {`