chore: adjust gen dual pdf display

haiwen · Jan 18, 2025 · 0e6d63c · 0e6d63c
1 parent e6d4426
commit 0e6d63c
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 6 deletions.
diff --git a/frontend/src/hooks/metadata-ai-operation.js b/frontend/src/hooks/metadata-ai-operation.js
@@ -70,16 +70,20 @@ export const MetadataAIOperationsProvider = ({
 
   const genDualLayerPDF = useCallback(({ parentDir, fileName }, { success_callback, fail_callback } = {}) => {
     const filePath = Utils.joinPath(parentDir, fileName);
-    const inProgressToaster = toaster.notifyInProgress(gettext('Making PDF searchable by AI...'), { duration: null });
+    const inProgressToaster = toaster.notifyInProgress(gettext('Making PDF searchable by AI...'), { duration: 3 });
     metadataAPI.genDualLayerPDF(repoID, filePath).then(res => {
       inProgressToaster.close();
       if (res.data.task_status === 'processing') {
         toaster.notifyInProgress(gettext('The task has been started, perhaps by another user'), { duration: 5 });
       } else if (res.data.task_status === 'already_ocr') {
         const userConfirmed = window.confirm(gettext('The text has already been OCR processed. Do you want to proceed with OCR again?'));
         if (userConfirmed) {
-          metadataAPI.genDualLayerPDF(repoID, filePath, { force: true }).then(() => {
-            success_callback && success_callback();
+          metadataAPI.genDualLayerPDF(repoID, filePath, { force: true }).then(res => {
+            if (res.data.task_status === 'processing') {
+              toaster.notifyInProgress(gettext('The task has been started, perhaps by another user'), { duration: 5 });
+            } else {
+              success_callback && success_callback();
+            }
           }).catch(() => {
             const errorMessage = gettext('Failed to make PDF searchable');
             toaster.danger(errorMessage);

diff --git a/frontend/src/metadata/api.js b/frontend/src/metadata/api.js
@@ -265,7 +265,7 @@ class MetadataManagerAPI {
   };
 
   genDualLayerPDF = (repoID, filePath, options = {}) => {
-    const url = this.server + '/api/v2.1/ai/pdf/to-dual-layer/';
+    const url = this.server + '/api/v2.1/ai/pdf/generate-text-layer/';
     const params = {
       path: filePath,
       repo_id: repoID,

diff --git a/seahub/ai/utils.py b/seahub/ai/utils.py
@@ -37,7 +37,7 @@ def generate_summary(params):
 
 def generate_dual_layer_pdf(params):
     headers = gen_headers()
-    url = urljoin(SEAFILE_AI_SERVER_URL, '/api/v1/pdf/generate-dual-layer-pdf/')
+    url = urljoin(SEAFILE_AI_SERVER_URL, '/api/v1/pdf/generate-text-layer/')
     resp = requests.post(url, json=params, headers=headers)
     return resp
 

diff --git a/seahub/urls.py b/seahub/urls.py
@@ -1061,5 +1061,5 @@
     re_path(r'^api/v2.1/ai/ocr/$', OCR.as_view(), name='api-v2.1-ocr'),
     re_path(r'^api/v2.1/ai/translate/$', Translate.as_view(), name='api-v2.1-translate'),
     re_path(r'^api/v2.1/ai/writing-assistant/$', WritingAssistant.as_view(), name='api-v2.1-writing-assistant'),
-    re_path(r'^api/v2.1/ai/pdf/to-dual-layer/$', GenDualLayerPDF.as_view(), name='api-v2.1-pdf-to-dual-layer'),
+    re_path(r'^api/v2.1/ai/pdf/generate-text-layer/$', GenDualLayerPDF.as_view(), name='api-v2.1-pdf-generate-text-layer'),
 ]