From a351fba48343a89245d9a67f3b7999af08301ce6 Mon Sep 17 00:00:00 2001
From: eeee0717 <chentao020717Work@outlook.com>
Date: Thu, 5 Jun 2025 20:32:47 +0800
Subject: [PATCH] feat: add preprocess icon

---
 src/main/services/KnowledgeService.ts         |  5 +++
 .../src/pages/knowledge/KnowledgeContent.tsx  | 32 +++++++++++++++++--
 .../pages/knowledge/components/StatusIcon.tsx | 23 +++++++------
 src/renderer/src/queue/KnowledgeQueue.ts      | 14 +++++++-
 src/renderer/src/store/knowledge.ts           | 19 ++++++++++-
 src/renderer/src/types/index.ts               |  1 +
 6 files changed, 79 insertions(+), 15 deletions(-)

diff --git a/src/main/services/KnowledgeService.ts b/src/main/services/KnowledgeService.ts
index 84362c026..e02fca9fc 100644
--- a/src/main/services/KnowledgeService.ts
+++ b/src/main/services/KnowledgeService.ts
@@ -565,6 +565,11 @@ class KnowledgeService {
         Logger.info(`Starting preprocess processing for scanned PDF: ${file.path}`)
         const { processedFile } = await provider.parseFile(item.id, file)
         fileToProcess = processedFile
+        console.warn('base', base, 'item', item, 'file', file)
+        const mainWindow = windowService.getMainWindow()
+        mainWindow?.webContents.send('file-preprocess-finished', {
+          itemId: item.id
+        })
       } catch (err) {
         Logger.error(`Preprocess processing failed: ${err}`)
         // 如果预处理失败，使用原始文件
diff --git a/src/renderer/src/pages/knowledge/KnowledgeContent.tsx b/src/renderer/src/pages/knowledge/KnowledgeContent.tsx
index eb5fd59a3..6b439ff74 100644
--- a/src/renderer/src/pages/knowledge/KnowledgeContent.tsx
+++ b/src/renderer/src/pages/knowledge/KnowledgeContent.tsx
@@ -39,9 +39,13 @@ const KnowledgeContent: FC<KnowledgeContentProps> = ({ selectedBase }) => {
   const { t } = useTranslation()
   const [expandAll, setExpandAll] = useState(false)
   const [progressMap, setProgressMap] = useState<Map<string, number>>(new Map())
-
+  const [preprocessMap, setPreprocessMap] = useState<Map<string, boolean>>(new Map())
   useEffect(() => {
     const handlers = [
+      window.electron.ipcRenderer.on('file-preprocess-finished', (_, { itemId }) => {
+        setPreprocessMap((prev) => new Map(prev).set(itemId, true))
+      }),
+
       window.electron.ipcRenderer.on('file-preprocess-progress', (_, { itemId, progress }) => {
         setProgressMap((prev) => new Map(prev).set(itemId, progress))
       }),
@@ -78,6 +82,7 @@ const KnowledgeContent: FC<KnowledgeContentProps> = ({ selectedBase }) => {
 
   const providerName = getProviderName(base?.model.provider || '')
   const disabled = !base?.version || !providerName
+  console.log(fileItems)
 
   if (!base) {
     return null
@@ -252,6 +257,16 @@ const KnowledgeContent: FC<KnowledgeContentProps> = ({ selectedBase }) => {
     }
   }
 
+  const showPreprocessIcon = (item: KnowledgeItem) => {
+    if (base.preprocessOrOcrProvider && item.isPreprocessed !== false) {
+      return true
+    }
+    if (!base.preprocessOrOcrProvider && item.isPreprocessed === true) {
+      return true
+    }
+    return false
+  }
+
   return (
     <MainContainer>
       <HeaderContainer>
@@ -369,13 +384,24 @@ const KnowledgeContent: FC<KnowledgeContentProps> = ({ selectedBase }) => {
                               {item.uniqueId && (
                                 <Button type="text" icon={<RefreshIcon />} onClick={() => refreshItem(item)} />
                               )}
+                              {showPreprocessIcon(item) && (
+                                <StatusIconWrapper>
+                                  <StatusIcon
+                                    sourceId={item.id}
+                                    base={base}
+                                    getProcessingStatus={getProcessingStatus}
+                                    type="file"
+                                    isPreprocessed={preprocessMap.get(item.id) || item.isPreprocessed || false}
+                                    progress={progressMap.get(item.id)}
+                                  />
+                                </StatusIconWrapper>
+                              )}
                               <StatusIconWrapper>
                                 <StatusIcon
                                   sourceId={item.id}
                                   base={base}
                                   getProcessingStatus={getProcessingStatus}
                                   type="file"
-                                  progress={progressMap.get(item.id)}
                                 />
                               </StatusIconWrapper>
                               <Button type="text" danger onClick={() => removeItem(item)} icon={<DeleteOutlined />} />
@@ -718,7 +744,7 @@ const ClickableSpan = styled.span`
 `
 
 const StatusIconWrapper = styled.div`
-  width: auto;
+  width: 32px;
   height: 44px;
   display: flex;
   align-items: center;
diff --git a/src/renderer/src/pages/knowledge/components/StatusIcon.tsx b/src/renderer/src/pages/knowledge/components/StatusIcon.tsx
index f1334198d..f2eb5fdf4 100644
--- a/src/renderer/src/pages/knowledge/components/StatusIcon.tsx
+++ b/src/renderer/src/pages/knowledge/components/StatusIcon.tsx
@@ -12,9 +12,17 @@ interface StatusIconProps {
   getProcessingStatus: (sourceId: string) => ProcessingStatus | undefined
   type: string
   progress?: number
+  isPreprocessed?: boolean
 }
 
-const StatusIcon: FC<StatusIconProps> = ({ sourceId, base, getProcessingStatus, type, progress = 0 }) => {
+const StatusIcon: FC<StatusIconProps> = ({
+  sourceId,
+  base,
+  getProcessingStatus,
+  type,
+  progress = 0,
+  isPreprocessed
+}) => {
   const { t } = useTranslation()
   const status = getProcessingStatus(sourceId)
   const item = base.items.find((item) => item.id === sourceId)
@@ -23,16 +31,11 @@ const StatusIcon: FC<StatusIconProps> = ({ sourceId, base, getProcessingStatus,
   const statusDisplay = useMemo(() => {
     if (!status) {
       if (item?.uniqueId) {
-        if (base.preprocessOrOcrProvider && item.type === 'file') {
+        if (isPreprocessed && item.type === 'file') {
           return (
-            <div style={{ gap: '8px', width: 'fit-content', height: '32px', display: 'flex' }}>
-              <Tooltip title={t('knowledge.status_preprocess_completed')} placement="left">
-                <CheckCircleOutlined style={{ color: '#52c41a' }} />
-              </Tooltip>
-              <Tooltip title={t('knowledge.status_embedding_completed')} placement="left">
-                <CheckCircleOutlined style={{ color: '#52c41a' }} />
-              </Tooltip>
-            </div>
+            <Tooltip title={t('knowledge.status_preprocess_completed')} placement="left">
+              <CheckCircleOutlined style={{ color: '#52c41a' }} />
+            </Tooltip>
           )
         }
         return (
diff --git a/src/renderer/src/queue/KnowledgeQueue.ts b/src/renderer/src/queue/KnowledgeQueue.ts
index 07ef1850b..331d9e54d 100644
--- a/src/renderer/src/queue/KnowledgeQueue.ts
+++ b/src/renderer/src/queue/KnowledgeQueue.ts
@@ -3,7 +3,12 @@ import db from '@renderer/databases'
 import { getKnowledgeBaseParams } from '@renderer/services/KnowledgeService'
 import { NotificationService } from '@renderer/services/NotificationService'
 import store from '@renderer/store'
-import { clearCompletedProcessing, updateBaseItemUniqueId, updateItemProcessingStatus } from '@renderer/store/knowledge'
+import {
+  clearCompletedProcessing,
+  updateBaseItemIsPreprocessed,
+  updateBaseItemUniqueId,
+  updateItemProcessingStatus
+} from '@renderer/store/knowledge'
 import { KnowledgeItem } from '@renderer/types'
 import { uuid } from '@renderer/utils'
 import type { LoaderReturn } from '@shared/config/types'
@@ -182,6 +187,13 @@ class KnowledgeQueue {
             uniqueIds: result.uniqueIds
           })
         )
+        store.dispatch(
+          updateBaseItemIsPreprocessed({
+            baseId,
+            itemId: item.id,
+            isPreprocessed: base.preprocessOrOcrProvider ? true : false
+          })
+        )
       }
       Logger.log(`[KnowledgeQueue] Updated uniqueId for item ${item.id} in base ${baseId} `)
 
diff --git a/src/renderer/src/store/knowledge.ts b/src/renderer/src/store/knowledge.ts
index 2facbcab3..dc7c32878 100644
--- a/src/renderer/src/store/knowledge.ts
+++ b/src/renderer/src/store/knowledge.ts
@@ -176,6 +176,7 @@ const knowledgeSlice = createSlice({
       action: PayloadAction<{ baseId: string; itemId: string; uniqueId: string; uniqueIds: string[] }>
     ) {
       const base = state.bases.find((b) => b.id === action.payload.baseId)
+      console.log('base2', base)
       if (base) {
         const item = base.items.find((item) => item.id === action.payload.itemId)
         if (item) {
@@ -183,6 +184,21 @@ const knowledgeSlice = createSlice({
           item.uniqueIds = action.payload.uniqueIds
         }
       }
+    },
+
+    updateBaseItemIsPreprocessed(
+      state,
+      action: PayloadAction<{ baseId: string; itemId: string; isPreprocessed: boolean }>
+    ) {
+      const base = state.bases.find((b) => b.id === action.payload.baseId)
+      console.log('base', base)
+      if (base) {
+        const item = base.items.find((item) => item.id === action.payload.itemId)
+        console.log('item', item)
+        if (item) {
+          item.isPreprocessed = action.payload.isPreprocessed
+        }
+      }
     }
   }
 })
@@ -201,7 +217,8 @@ export const {
   updateItemProcessingStatus,
   clearCompletedProcessing,
   clearAllProcessing,
-  updateBaseItemUniqueId
+  updateBaseItemUniqueId,
+  updateBaseItemIsPreprocessed
 } = knowledgeSlice.actions
 
 export default knowledgeSlice.reducer
diff --git a/src/renderer/src/types/index.ts b/src/renderer/src/types/index.ts
index 493f0fbc5..184353e27 100644
--- a/src/renderer/src/types/index.ts
+++ b/src/renderer/src/types/index.ts
@@ -363,6 +363,7 @@ export type KnowledgeItem = {
   processingProgress?: number
   processingError?: string
   retryCount?: number
+  isPreprocessed?: boolean
 }
 
 export interface KnowledgeBase {