change emotions prediction api, change ui

api/package.json

@@ -43,6 +43,7 @@
     "@paralect/email-service": "0.2.0",
     "@paralect/node-mongo": "2.0.0-beta.1",
     "app-module-path": "2.2.0",
+    "axios": "0.19.2",
     "bcrypt": "4.0.1",
     "google-auth-library": "6.0.0",
     "ibm-watson": "5.5.0",

api/src/resources/processing/analyze/index.js

@@ -1,4 +1,5 @@
 const validate = require('middlewares/validate');
+
 const NLPService = require('services/nlp.service');
 
 const validator = require('./validator');
@@ -9,9 +10,11 @@ const handler = async (ctx) => {
 
     const result = await NLPService.analyzeText(text);
     const vehicleColors = await NLPService.recognizeVehicleColor(image);
+    const emotions = await NLPService.analyzeEmotions(text);
 
     ctx.body = {
       ...result,
+      emotions,
       entities: [
         ...result.entities,
         ...vehicleColors,

api/src/services/nlp.service.js

@@ -1,8 +1,11 @@
+/* eslint-disable class-methods-use-this */
 const fs = require('fs');
 const sharp = require('sharp');
+const axios = require('axios');
 const vision = require('@google-cloud/vision');
 const { NaturalLanguageUnderstandingV1 } = require('ibm-watson/sdk');
 const { IamAuthenticator } = require('ibm-watson/auth');
+
 const { rgbToHex } = require('helpers/base.helper');
 
 class NaturalLanguageProcessingService {
@@ -34,14 +37,31 @@ class NaturalLanguageProcessingService {
         concepts: {},
         entities: {},
         keywords: {},
-        emotion: {},
         semantic_roles: {},
       },
     });
 
     return response.result;
   }
 
+  async analyzeEmotions(text) {
+    const sentencesArray = text.replace(/([.?!])\s*(?=[A-Z])/g, '$1|').split('|');
+
+    const predictionsArray = (await Promise.all(sentencesArray.map((sentence) => (
+      axios.post('http://demo-nlp.paralect.net/predict', { text: sentence })
+    )))).map((element) => (element.data.predictions.flat(1)));
+
+    const averageEmotions = predictionsArray[0].map((_col, i) => predictionsArray.map(
+      (row) => row[i],
+    ).reduce((acc, c) => acc + c, 0) / predictionsArray.length);
+
+    if (!averageEmotions) {
+      throw new Error('Not recognized');
+    }
+
+    return averageEmotions;
+  }
+
   async localizeVehicle(image) {
     const imageBuffer = fs.readFileSync(image.path);
     const sharpImage = sharp(imageBuffer);