mindee 1.1.2 → 1.3.1

package/mindee/documents/invoice.js

@@ -50,6 +50,7 @@ class Invoice extends Document {
     customerName = undefined,
     customerAddress = undefined,
     customerCompanyRegistration = undefined,
+    words = undefined,
     pageNumber = 0,
     level = "page",
   }) {
@@ -80,7 +81,7 @@ class Invoice extends Document {
         customerCompanyRegistration,
       });
     } else {
-      this.#initFromApiPrediction(apiPrediction, pageNumber);
+      this.#initFromApiPrediction(apiPrediction, pageNumber, words);
     }
     this.#checklist();
     this.#reconstruct();
@@ -142,8 +143,7 @@ class Invoice extends Document {
     }
   }
 
-  #initFromApiPrediction(apiPrediction, pageNumber) {
-    this.words = [];
+  #initFromApiPrediction(apiPrediction, pageNumber, words) {
     this.locale = new Locale({ prediction: apiPrediction.locale, pageNumber });
     this.totalIncl = new Amount({
       prediction: apiPrediction.total_incl,
@@ -243,8 +243,7 @@ class Invoice extends Document {
         })
       );
     }
-    // document.inference.ocr
-    if ("mvision" in apiPrediction) this.words = apiPrediction.mvision;
+    if (words && words.length > 0) this.words = words;
   }
 
   toString() {

package/mindee/documents/receipt.js

@@ -37,6 +37,7 @@ class Receipt extends Document {
     orientation = undefined,
     totalTax = undefined,
     totalExcl = undefined,
+    words = undefined,
     pageNumber = 0,
     level = "page",
   }) {
@@ -60,7 +61,7 @@ class Receipt extends Document {
         pageNumber,
       });
     } else {
-      this.#initFromApiPrediction(apiPrediction, pageNumber);
+      this.#initFromApiPrediction(apiPrediction, pageNumber, words);
     }
     this.#checklist();
     this.#reconstruct();
@@ -108,8 +109,7 @@ class Receipt extends Document {
     @param apiPrediction: Raw prediction from HTTP response
     @param pageNumber: Page number for multi pages pdf input
    */
-  #initFromApiPrediction(apiPrediction, pageNumber) {
-    this.words = [];
+  #initFromApiPrediction(apiPrediction, pageNumber, words) {
     this.locale = new Locale({ prediction: apiPrediction.locale, pageNumber });
     this.totalIncl = new Amount({
       prediction: apiPrediction.total_incl,
@@ -171,7 +171,7 @@ class Receipt extends Document {
         })
       );
     }
-    if ("mvision" in apiPrediction) this.words = apiPrediction.mvision;
+    if (words && words.length > 0) this.words = words;
   }
 
   toString() {

package/mindee/inputs.js

@@ -52,18 +52,24 @@ class Input {
   async initBase64() {
     this.fileObject = this.file;
     this.filepath = undefined;
-    this.fileExtension = undefined;
 
-    if (this.allowCutPdf == true) {
-      const typeOfFile = await fileType.fromBuffer(
+    if (this.filename === undefined) {
+      const mimeType = await fileType.fromBuffer(
         Buffer.from(this.fileObject, "base64")
       );
-
-      if (typeOfFile === undefined) {
-        console.error(`Cannot detect mime type of: ${this.fileObject}`);
-      } else if (typeOfFile.mime === "application/pdf") {
-        await this.cutPdf();
+      if (mimeType !== undefined) {
+        this.fileExtension = mimeType.mime;
+        this.filename = `from_${this.inputType}.${mimeType.ext}`;
+      } else {
+        throw "Could not determine the MIME type. Please specify the 'filename' option.";
       }
+    } else {
+      const filetype = this.filename.split(".").pop();
+      this.fileExtension = this.MIMETYPES[filetype];
+    }
+
+    if (this.fileExtension === "application/pdf" && this.allowCutPdf == true) {
+      await this.cutPdf();
     }
   }
 

package/package.json

@@ -1,9 +1,9 @@
 {
   "name": "mindee",
-  "version": "1.1.2",
-  "description": "Mindee API SDK for Node.js",
+  "version": "1.3.1",
+  "description": "Mindee Client Library for Node.js",
   "main": "mindee/index.js",
-  "license": "Apache-2.0",
+  "license": "MIT",
   "scripts": {
     "build": "babel mindee -d lib",
     "test": "npm run build && mocha tests/**/*.js",
@@ -34,12 +34,12 @@
     "prettier": "^2.2.0"
   },
   "dependencies": {
-    "base64-arraybuffer": "^1.0.1",
+    "base64-arraybuffer": "^1.0.2",
     "base64-stream": "^1.0.0",
     "concat-stream": "^2.0.0",
     "file-type": "^16.5.3",
     "form-data": "^3.0.1",
-    "pdf-lib": "^1.13.0"
+    "pdf-lib": "^1.17.1"
   },
   "keywords": [
     "javascript",