From b28831ec25351b3cdd4cb6fc8621a2963a984b59 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Fri, 3 Oct 2025 10:11:39 -0300
Subject: [PATCH] fixed issues with already ingested files, fixed state issues

---
 frontend/src/app/knowledge/page.tsx    | 11 +++++------
 frontend/src/contexts/task-context.tsx |  9 +++++----
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/frontend/src/app/knowledge/page.tsx b/frontend/src/app/knowledge/page.tsx
index 41485289..c6d254c4 100644
--- a/frontend/src/app/knowledge/page.tsx
+++ b/frontend/src/app/knowledge/page.tsx
@@ -75,14 +75,13 @@ function SearchPage() {
 		};
 	});
 
-	const backendFiles = data as File[];
+	const backendFiles = (data as File[]).filter((file) => !taskFilesAsFiles.some((taskFile) => taskFile.filename === file.filename && taskFile.status === "processing"));
 
 	const filteredTaskFiles = taskFilesAsFiles.filter((taskFile) => {
 		return (
 			taskFile.status !== "active" &&
 			!backendFiles.some(
-				(backendFile) => backendFile.filename === taskFile.filename,
-			)
+				(backendFile) => backendFile.filename === taskFile.filename,)
 		);
 	});
 
@@ -91,7 +90,7 @@ function SearchPage() {
 
 	const gridRef = useRef<AgGridReact>(null);
 
-	const [columnDefs] = useState<ColDef<File>[]>([
+	const columnDefs = [
 		{
 			field: "filename",
 			headerName: "Source",
@@ -189,8 +188,8 @@ function SearchPage() {
 			resizable: false,
 			sortable: false,
 			initialFlex: 0,
-		},
-	]);
+		}
+	];
 
 	const defaultColDef: ColDef<File> = {
 		resizable: false,
diff --git a/frontend/src/contexts/task-context.tsx b/frontend/src/contexts/task-context.tsx
index b3275422..8928ae8f 100644
--- a/frontend/src/contexts/task-context.tsx
+++ b/frontend/src/contexts/task-context.tsx
@@ -212,11 +212,12 @@ export function TaskProvider({ children }: { children: React.ReactNode }) {
               onClick: () => console.log("View task", currentTask.task_id),
             },
           });
+          setTimeout(() => {
           refetchSearch();
-          // Remove files for this completed task from the files list
-          // setFiles((prevFiles) =>
-          //   prevFiles.filter((file) => file.task_id !== currentTask.task_id),
-          // );
+          setFiles((prevFiles) =>
+            prevFiles.filter((file) => file.task_id !== currentTask.task_id && file.status !== "failed"),
+            );
+          }, 500);
         } else if (
           previousTask && previousTask.status !== "failed" &&
           previousTask.status !== "error" &&