From 0b1990f8b3755a2e4e3177a471278fd32befd4a0 Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Fri, 28 Mar 2025 18:24:01 +0800
Subject: [PATCH] refactor: lark

---
 ui/src/locales/lang/en-US/views/document.ts   |  2 +-
 ui/src/locales/lang/zh-CN/views/document.ts   |  2 +-
 ui/src/locales/lang/zh-Hant/views/document.ts |  2 +-
 .../views/dataset/ImportDocumentDataset.vue   | 47 ++++++++++++++++++-
 4 files changed, 48 insertions(+), 5 deletions(-)

diff --git a/ui/src/locales/lang/en-US/views/document.ts b/ui/src/locales/lang/en-US/views/document.ts
index c2b8a40b4..3b3d59110 100644
--- a/ui/src/locales/lang/en-US/views/document.ts
+++ b/ui/src/locales/lang/en-US/views/document.ts
@@ -171,7 +171,7 @@ export default {
   },
   feishu: {
     selectDocument: 'Select Document',
-    tip1: 'Only documents and tables are supported. Documents will be segmented based on titles, and tables will be converted to Markdown format before segmentation.',
+    tip1: 'Supports document and table types such as Feishu documents, TXT, Markdown, PDF, DOCX, HTML, XLS, XLSX, CSV, ZIP Feishu files. Documents will be segmented according to titles, and tables will be converted to Markdown format before segmentation.',
     tip2: 'The system does not store the original document. Before importing the document, it is recommended to standardize the document segmentation markers.',
     allCheck: 'Select All',
     errorMessage1: 'Please select a document'
diff --git a/ui/src/locales/lang/zh-CN/views/document.ts b/ui/src/locales/lang/zh-CN/views/document.ts
index 8f00840ac..b3e6f1213 100644
--- a/ui/src/locales/lang/zh-CN/views/document.ts
+++ b/ui/src/locales/lang/zh-CN/views/document.ts
@@ -168,7 +168,7 @@ export default {
   },
   feishu: {
     selectDocument: '选择文档',
-    tip1: '仅支持文档和表格类型，文档会根据标题分段，表格会转为Markdown格式后再分段。',
+    tip1: '支持文档和表格类型飞书文档、支持TXT、Markdown、PDF、DOCX、HTML、XLS、XLSX、CSV、ZIP飞书文件，文档会根据标题分段，表格会转为Markdown格式后再分段。',
     tip2: '系统不存储原始文档，导入文档前，建议规范文档的分段标识。',
     allCheck: '全选',
     errorMessage1: '请选择文档'
diff --git a/ui/src/locales/lang/zh-Hant/views/document.ts b/ui/src/locales/lang/zh-Hant/views/document.ts
index a2c9b1634..1942d5ebc 100644
--- a/ui/src/locales/lang/zh-Hant/views/document.ts
+++ b/ui/src/locales/lang/zh-Hant/views/document.ts
@@ -167,7 +167,7 @@ export default {
   },
   feishu: {
     selectDocument: '選擇文檔',
-    tip1: '僅支持文檔和表格類型，文檔會根據標題分段，表格會轉為Markdown格式後再分段。',
+    tip1: '支持檔案和表格類型飛書檔案、支持TXT、Markdown、PDF、DOCX、HTML、XLS、XLSX、CSV、ZIP飛書檔案，檔案會根據標題分段，表格會轉為Markdown格式後再分段。',
     tip2: '系統不存儲原始文檔，導入文檔前，建議規範文檔的分段標識。',
     allCheck: '全選',
     errorMessage1: '請選擇文檔'
diff --git a/ui/src/views/dataset/ImportDocumentDataset.vue b/ui/src/views/dataset/ImportDocumentDataset.vue
index ef41a9bdb..d512fd55b 100644
--- a/ui/src/views/dataset/ImportDocumentDataset.vue
+++ b/ui/src/views/dataset/ImportDocumentDataset.vue
@@ -63,13 +63,55 @@
                         src="@/assets/fileType/docx-icon.svg"
                         alt=""
                         height="22"
-                        v-else-if="data.type === 'docx'"
+                        v-else-if="data.type === 'docx' || data.name.endsWith('.docx')"
                       />
                       <img
                         src="@/assets/fileType/xlsx-icon.svg"
                         alt=""
                         height="22"
-                        v-else-if="data.type === 'sheet'"
+                        v-else-if="data.type === 'sheet' || data.name.endsWith('.xlsx')"
+                      />
+                      <img
+                        src="@/assets/fileType/xls-icon.svg"
+                        alt=""
+                        height="22"
+                        v-else-if="data.name.endsWith('xls')"
+                      />
+                      <img
+                        src="@/assets/fileType/csv-icon.svg"
+                        alt=""
+                        height="22"
+                        v-else-if="data.name.endsWith('csv')"
+                      />
+                      <img
+                        src="@/assets/fileType/pdf-icon.svg"
+                        alt=""
+                        height="22"
+                        v-else-if="data.name.endsWith('.pdf')"
+                      />
+                      <img
+                        src="@/assets/fileType/html-icon.svg"
+                        alt=""
+                        height="22"
+                        v-else-if="data.name.endsWith('.html')"
+                      />
+                      <img
+                        src="@/assets/fileType/txt-icon.svg"
+                        alt=""
+                        height="22"
+                        v-else-if="data.name.endsWith('.txt')"
+                      />
+                      <img
+                        src="@/assets/fileType/zip-icon.svg"
+                        alt=""
+                        height="22"
+                        v-else-if="data.name.endsWith('.zip')"
+                      />
+                      <img
+                        src="@/assets/fileType/md-icon.svg"
+                        alt=""
+                        height="22"
+                        v-else-if="data.name.endsWith('.md')"
                       />
 
                       <span class="ml-4">{{ node.label }}</span>
@@ -236,6 +278,7 @@ function back() {
     margin-bottom: 20px;
   }
 }
+
 .xlsx-icon {
   svg {
     width: 24px;