From 240997b17ef023fb6652cf0aa39fc7d85432ad1f Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Fri, 12 Mar 2021 10:07:24 +0900
Subject: [PATCH] Add auto labeling form to the ToolbarLabeling.vue

---
 .../tasks/toolbar/ToolbarLabeling.vue         | 36 +++++++++++++++++--
 .../tasks/toolbar/forms/FormAutoLabeling.vue  |  2 +-
 2 files changed, 35 insertions(+), 3 deletions(-)

diff --git a/frontend/components/tasks/toolbar/ToolbarLabeling.vue b/frontend/components/tasks/toolbar/ToolbarLabeling.vue
index 9f5f0cad..acbc55f8 100644
--- a/frontend/components/tasks/toolbar/ToolbarLabeling.vue
+++ b/frontend/components/tasks/toolbar/ToolbarLabeling.vue
@@ -37,8 +37,16 @@
         </v-dialog>
 
         <button-auto-labeling
-          @click:auto="$emit('click:auto')"
+          @click:auto="dialogAutoLabeling=true"
         />
+        <v-dialog v-model="dialogAutoLabeling">
+          <form-auto-labeling
+            :is-enabled="enableAutoLabeling"
+            :error-message="errorMessage"
+            @click:cancel="dialogAutoLabeling=false"
+            @input="updateAutoLabeling"
+          />
+        </v-dialog>
 
         <button-clear
           @click:clear="dialogClear=true"
@@ -73,6 +81,7 @@ import ButtonFilter from './buttons/ButtonFilter.vue'
 import ButtonGuideline from './buttons/ButtonGuideline.vue'
 import ButtonPagination from './buttons/ButtonPagination.vue'
 import ButtonReview from './buttons/ButtonReview.vue'
+import FormAutoLabeling from './forms/FOrmAutoLabeling.vue'
 import FormClearLabel from './forms/FormClearLabel.vue'
 import FormComment from './forms/FormComment.vue'
 import FormGuideline from './forms/FormGuideline.vue'
@@ -86,6 +95,7 @@ export default Vue.extend({
     ButtonGuideline,
     ButtonPagination,
     ButtonReview,
+    FormAutoLabeling,
     FormClearLabel,
     FormComment,
     FormGuideline
@@ -96,6 +106,11 @@ export default Vue.extend({
       type: Number,
       required: true
     },
+    enableAutoLabeling: {
+      type: Boolean,
+      default: false,
+      required: true
+    },
     filterOption: {
       type: String,
       default: ''
@@ -117,9 +132,11 @@ export default Vue.extend({
 
   data() {
     return {
+      dialogAutoLabeling: false,
       dialogClear: false,
       dialogComment: false,
-      dialogGuideline: false
+      dialogGuideline: false,
+      errorMessage: ''
     }
   },
 
@@ -133,6 +150,21 @@ export default Vue.extend({
   methods: {
     updatePage(page: number) {
       this.$router.push({ query: { page: page.toString() }})
+    },
+    
+    async updateAutoLabeling(isEnable: boolean) {
+      if (isEnable) {
+        try {
+          const projectId = this.$route.params.id
+          await this.$services.textClassification.autoLabel(projectId, this.docId)
+          this.$emit('update:enable-auto-labeling', true)
+          this.errorMessage = ''
+        } catch (e) {
+          this.errorMessage = e.response.data.detail
+        }
+      } else {
+        this.$emit('update:enable-auto-labeling', false)
+      }
     }
   }
 })
diff --git a/frontend/components/tasks/toolbar/forms/FormAutoLabeling.vue b/frontend/components/tasks/toolbar/forms/FormAutoLabeling.vue
index 462cb031..3b990bb6 100644
--- a/frontend/components/tasks/toolbar/forms/FormAutoLabeling.vue
+++ b/frontend/components/tasks/toolbar/forms/FormAutoLabeling.vue
@@ -18,7 +18,7 @@
       <v-switch
         :value="isEnabled"
         :error-messages="errorMessage"
-        @change="$('input', $event)"
+        @change="$emit('input', $event)"
       />
     </template>
   </base-card>