From 6e45f69e9be2176dbf349f43cdae967b42034b12 Mon Sep 17 00:00:00 2001
From: Hironsan <hiroki.nakayama.py@gmail.com>
Date: Thu, 1 Jun 2023 16:21:46 +0900
Subject: [PATCH] Add fields to project model

---
 frontend/domain/models/project/project.ts     | 10 ++++++++++
 frontend/pages/projects/_id/dataset/index.vue | 15 +++------------
 2 files changed, 13 insertions(+), 12 deletions(-)

diff --git a/frontend/domain/models/project/project.ts b/frontend/domain/models/project/project.ts
index fd7b7b59..ee2bb5a4 100644
--- a/frontend/domain/models/project/project.ts
+++ b/frontend/domain/models/project/project.ts
@@ -138,4 +138,14 @@ export class Project {
     }
     return `${this.projectType}Project`
   }
+
+  get isImageProject(): boolean {
+    return [ImageClassification, ImageCaptioning, BoundingBox, Segmentation].includes(
+      this.projectType
+    )
+  }
+
+  get isAudioProject(): boolean {
+    return [Speech2text].includes(this.projectType)
+  }
 }
diff --git a/frontend/pages/projects/_id/dataset/index.vue b/frontend/pages/projects/_id/dataset/index.vue
index 03386845..bb890c75 100644
--- a/frontend/pages/projects/_id/dataset/index.vue
+++ b/frontend/pages/projects/_id/dataset/index.vue
@@ -35,7 +35,7 @@
       </v-dialog>
     </v-card-title>
     <image-list
-      v-if="isImageTask"
+      v-if="project.isImageProject"
       v-model="selected"
       :items="item.items"
       :is-loading="isLoading"
@@ -44,7 +44,7 @@
       @click:labeling="movePage"
     />
     <audio-list
-      v-else-if="isAudioTask"
+      v-else-if="project.isAudioProject"
       v-model="selected"
       :items="item.items"
       :is-loading="isLoading"
@@ -122,17 +122,8 @@ export default Vue.extend({
       return this.$route.params.id
     },
 
-    isImageTask(): boolean {
-      const imageTasks = ['ImageClassification', 'ImageCaptioning', 'BoundingBox', 'Segmentation']
-      return imageTasks.includes(this.project.projectType)
-    },
-
-    isAudioTask(): boolean {
-      return this.project.projectType === 'Speech2text'
-    },
-
     itemKey(): string {
-      if (this.isImageTask || this.isAudioTask) {
+      if (this.project.isImageProject || this.project.isAudioProject) {
         return 'filename'
       } else {
         return 'text'