#495 bug fix

AFP-Medialab · May 17, 2024 · 6d567cb · 6d567cb
1 parent 9bb26b3
commit 6d567cb
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/src/components/NavItems/tools/Deepfake/DeepfakeImage.jsx b/src/components/NavItems/tools/Deepfake/DeepfakeImage.jsx
@@ -50,6 +50,7 @@ const Deepfake = () => {
 
   const submitUrl = () => {
     UseGetDeepfake(
+      keyword,
       input,
       true,
       selectedMode,

diff --git a/src/components/NavItems/tools/Deepfake/DeepfakeVideo.jsx b/src/components/NavItems/tools/Deepfake/DeepfakeVideo.jsx
@@ -48,6 +48,7 @@ const Deepfake = () => {
 
   const submitUrl = () => {
     UseGetDeepfake(
+      keyword,
       input,
       true,
       selectedMode,

diff --git a/src/components/NavItems/tools/Deepfake/Hooks/useGetDeepfake.jsx b/src/components/NavItems/tools/Deepfake/Hooks/useGetDeepfake.jsx
@@ -11,6 +11,7 @@ import { setError } from "redux/reducers/errorReducer";
 import { isValidUrl } from "../../../../Shared/Utils/URLUtils";
 
 async function UseGetDeepfake(
+  keyword,
   url,
   processURL,
   mode,
@@ -48,7 +49,7 @@ async function UseGetDeepfake(
   let res;
 
   const handleError = (e) => {
-    dispatch(setError(e));
+    dispatch(setError(keyword(e)));
     if (mode === "IMAGE") {
       dispatch(setDeepfakeLoadingImage(false));
     } else if (mode === "VIDEO") {
@@ -82,15 +83,15 @@ async function UseGetDeepfake(
         break;
     }
   } catch (error) {
-    handleError("error_" + error.status);
+    handleError("deepfake_error_" + error.response.status);
   }
 
   const getResult = async (id) => {
     let response;
     try {
       response = await axios.get(baseURL + modeURL + "reports/" + id);
     } catch (error) {
-      handleError("error_" + error.status);
+      handleError("deepfake_error_" + error.status);
     }
 
     if (response.data != null) {
@@ -115,7 +116,7 @@ async function UseGetDeepfake(
     try {
       response = await axios.get(baseURL + modeURL + "jobs/" + id);
     } catch (error) {
-      handleError("error_" + error.status);
+      handleError("deepfake_error_" + error.status);
     }
 
     if (response && response.data && response.data.status === "PROCESSING") {
@@ -127,11 +128,11 @@ async function UseGetDeepfake(
     ) {
       await getResult(id);
     } else {
-      handleError("error_" + response.data.status);
+      handleError("deepfake_error_" + response.data.status);
     }
   };
 
-  waitUntilFinish(res.data.id);
+  res ? waitUntilFinish(res.data.id) : null;
 }
 
 function sleep(fn, param) {