From 6770425a857271c3853e01b3b4add2168c5d1ff9 Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Fri, 19 Feb 2021 11:39:35 +0900
Subject: [PATCH] Add error handling when user try to activate auto labeling on
 the annotated text

---
 frontend/layouts/annotation.vue | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/frontend/layouts/annotation.vue b/frontend/layouts/annotation.vue
index 006a0cf6..76caf859 100644
--- a/frontend/layouts/annotation.vue
+++ b/frontend/layouts/annotation.vue
@@ -22,6 +22,21 @@
       <v-overlay :value="loading">
         <v-progress-circular indeterminate size="64" />
       </v-overlay>
+      <v-snackbar
+        v-model="snackbar"
+      >
+        {{ text }}
+        <template v-slot:action="{ attrs }">
+          <v-btn
+            color="pink"
+            text
+            v-bind="attrs"
+            @click="snackbar = false"
+          >
+            Close
+          </v-btn>
+        </template>
+      </v-snackbar>
       <v-container fluid>
         <v-row
           no-gutters
@@ -127,7 +142,9 @@ export default {
       },
       errors: {
         'autoLabelingConfig': ''
-      }
+      },
+      snackbar: false,
+      text: ''
     }
   },
 
@@ -200,9 +217,15 @@ export default {
       async handler() {
         this.setCurrent(this.current)
         if (this.options.onAutoLabeling) {
-          this.setLoading(true)
-          await this.autoLabeling({ projectId: this.$route.params.id })
-          this.setLoading(false)
+          try {
+            this.setLoading(true)
+            await this.autoLabeling({ projectId: this.$route.params.id })
+          } catch (e) {
+            this.snackbar = true
+            this.text = e.response.data.detail
+          } finally {
+            this.setLoading(false)
+          }
         }
       },
       immediate: true