From 8f306a9e178c202f0366f6b45e30a0057a331e51 Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Fri, 18 Oct 2019 15:27:35 +0900
Subject: [PATCH] Add named entity recognition demo

---
 .../demo/named-entity-recognition/index.vue   | 150 ++++++++++++++++++
 1 file changed, 150 insertions(+)
 create mode 100644 frontend/pages/demo/named-entity-recognition/index.vue

diff --git a/frontend/pages/demo/named-entity-recognition/index.vue b/frontend/pages/demo/named-entity-recognition/index.vue
new file mode 100644
index 00000000..4b28c56f
--- /dev/null
+++ b/frontend/pages/demo/named-entity-recognition/index.vue
@@ -0,0 +1,150 @@
+<template>
+  <v-content>
+    <v-container fluid>
+      <v-row justify="center">
+        <v-col cols="12" md="9">
+          <v-card>
+            <v-card-text class="title">
+              <entity-item-box
+                :labels="items"
+                :text="currentDoc.text"
+                :entities="currentDoc.annotations"
+                :delete-annotation="removeEntity"
+                :update-entity="updateEntity"
+                :add-entity="addEntity"
+              />
+            </v-card-text>
+          </v-card>
+        </v-col>
+        <v-col cols="12" md="3">
+          <metadata-box :metadata="JSON.parse(currentDoc.meta)" />
+        </v-col>
+      </v-row>
+    </v-container>
+  </v-content>
+</template>
+
+<script>
+import EntityItemBox from '~/components/organisms/EntityItemBox'
+import MetadataBox from '@/components/organisms/MetadataBox'
+
+export default {
+  layout: 'annotation',
+
+  components: {
+    EntityItemBox,
+    MetadataBox
+  },
+
+  data() {
+    return {
+      items: [
+        {
+          id: 4,
+          text: 'LOC',
+          prefix_key: null,
+          suffix_key: 'l',
+          background_color: '#7c20e0',
+          text_color: '#ffffff'
+        },
+        {
+          id: 5,
+          text: 'MISC',
+          prefix_key: null,
+          suffix_key: 'm',
+          background_color: '#fbb028',
+          text_color: '#000000'
+        },
+        {
+          id: 6,
+          text: 'ORG',
+          prefix_key: null,
+          suffix_key: 'o',
+          background_color: '#e6d176',
+          text_color: '#000000'
+        },
+        {
+          id: 7,
+          text: 'PER',
+          prefix_key: null,
+          suffix_key: 'p',
+          background_color: '#6a74b9',
+          text_color: '#ffffff'
+        }
+      ],
+      currentDoc: {
+        id: 8,
+        text: 'After bowling Somerset out for 83 on the opening morning at Grace Road , Leicestershire extended their first innings by 94 runs before being bowled out for 296 with England discard Andy Caddick taking three for 83 .',
+        annotations: [
+          {
+            id: 17,
+            prob: 0.0,
+            label: 4,
+            start_offset: 60,
+            end_offset: 70,
+            user: 1,
+            document: 8
+          },
+          {
+            id: 19,
+            prob: 0.0,
+            label: 4,
+            start_offset: 165,
+            end_offset: 172,
+            user: 1,
+            document: 8
+          },
+          {
+            id: 16,
+            prob: 0.0,
+            label: 6,
+            start_offset: 14,
+            end_offset: 22,
+            user: 1,
+            document: 8
+          },
+          {
+            id: 18,
+            prob: 0.0,
+            label: 6,
+            start_offset: 73,
+            end_offset: 87,
+            user: 1,
+            document: 8
+          },
+          {
+            id: 20,
+            prob: 0.0,
+            label: 7,
+            start_offset: 181,
+            end_offset: 193,
+            user: 1,
+            document: 8
+          }
+        ],
+        meta: '{"wikiPageId":2}',
+        annotation_approver: null
+      }
+    }
+  },
+
+  methods: {
+    removeEntity(annotationId) {
+      this.currentDoc.annotations = this.currentDoc.annotations.filter(item => item.id !== annotationId)
+    },
+    updateEntity(labelId, annotationId) {
+      const index = this.currentDoc.annotations.findIndex(item => item.id === annotationId)
+      this.currentDoc.annotations[index].label = labelId
+    },
+    addEntity(startOffset, endOffset, labelId) {
+      const payload = {
+        id: this.currentDoc.annotations.reduce((x, y) => { return x.id > y.id ? x : y }).id + 1,
+        start_offset: startOffset,
+        end_offset: endOffset,
+        label: labelId
+      }
+      this.currentDoc.annotations.push(payload)
+    }
+  }
+}
+</script>