feat: extract file status

OVINC-CN · Jun 27, 2024 · 11d0b22 · 11d0b22
1 parent 82c07c8
commit 11d0b22
Show file tree

Hide file tree

Showing 4 changed files with 47 additions and 9 deletions.
diff --git a/src/api/cos.js b/src/api/cos.js
@@ -14,8 +14,15 @@ export const getCOSUploadTempKeyAPI = (filename, purpose, tcaptcha) => new Promi
   );
 });
 
-export const extractFileAPI = (key) => new Promise((resolve, reject) => {
-  http.post('/cos/extract_file/', {key}).then(
+export const extractFileAPI = (filePath) => new Promise((resolve, reject) => {
+  http.post('/cos/extract_file/', {file_path: filePath}).then(
+      (res) => resolve(res),
+      (err) => reject(err),
+  );
+});
+
+export const extractFileStatusAPI = (filePath) => new Promise((resolve, reject) => {
+  http.get('/cos/extract_file_status/', {params: {file_path: filePath}}).then(
       (res) => resolve(res),
       (err) => reject(err),
   );

diff --git a/src/components/ChatInput.vue b/src/components/ChatInput.vue
@@ -7,7 +7,7 @@ import {useStore} from 'vuex';
 import {useI18n} from 'vue-i18n';
 import globalContext from '../context';
 import {checkTCaptcha} from '@/utils/tcaptcha';
-import {extractFileAPI, getCOSConfigAPI, getCOSUploadTempKeyAPI} from '@/api/cos';
+import {extractFileAPI, extractFileStatusAPI, getCOSConfigAPI, getCOSUploadTempKeyAPI} from '@/api/cos';
 import {loadCOSClient} from '@/utils/cos';
 
 // props
@@ -98,8 +98,8 @@ const doChat = async () => {
       .then((preRes) => {
         key = preRes.data.key;
       }, (err) => {
-        Message.error(err.response.data.message);
         emits('setChatLoading', false);
+        Message.error(err.response.data.message);
       });
   if (!key) {
     return;
@@ -230,22 +230,51 @@ const handleFileChange = (event) => {
               },
               (err) => {
                 if (err) {
-                  Message.error(err.message);
                   emits('setChatLoading', false);
+                  Message.error(err.message);
                 } else {
-                  extractFileAPI(credentials.key);
-                  promptForm.value.file = `${credentials.cos_url}${credentials.key}`;
-                  emits('setChatLoading', false);
+                  const filePath = `${credentials.cos_url}${credentials.key}`;
+                  extractFileAPI(filePath).then(
+                      () => {
+                        checkExtractStatus(filePath);
+                      },
+                      (err) => {
+                        emits('setChatLoading', false);
+                        Message.error(err.response.data.message);
+                      },
+                  );
                 }
               });
         },
         (err) => {
-          Message.error(err.response.data.message);
           emits('setChatLoading', false);
+          Message.error(err.response.data.message);
         },
     );
   });
 };
+const checkExtractStatus = (filePath) => {
+  setTimeout(() => {
+    extractFileStatusAPI(filePath).then(
+        (res) => {
+          if (res.data.is_finished) {
+            if (res.data.is_success) {
+              promptForm.value.file = filePath;
+            } else {
+              Message.error(i18n.t('ExtractFileFailed'));
+            }
+            emits('setChatLoading', false);
+          } else {
+            checkExtractStatus(filePath);
+          }
+        },
+        (err) => {
+          emits('setChatLoading', false);
+          Message.error(err.response.data.message);
+        },
+    );
+  }, 2000);
+};
 
 defineExpose({reGenerate, promptForm});
 </script>

diff --git a/src/locale/en-us.js b/src/locale/en-us.js
@@ -19,6 +19,7 @@ const mEnUS = {
   UploadFile: 'Upload',
   SupportFileTypes: 'Only support for image or pdf',
   ReUploadFile: 'Re-Upload',
+  ExtractFileFailed: 'Extract file failed, please try again',
 };
 
 export default mEnUS;
diff --git a/src/locale/zh-cn.js b/src/locale/zh-cn.js
@@ -19,6 +19,7 @@ const mZhCN = {
   UploadFile: '上传',
   SupportFileTypes: '仅支持图片或PDF',
   ReUploadFile: '覆盖',
+  ExtractFileFailed: '解析文件失败，请重新上传',
 };
 
 export default mZhCN;