From f9ab91042877732a0f64f29bbbfdb2f595b14199 Mon Sep 17 00:00:00 2001
From: Leire <leire@argilla.io>
Date: Mon, 18 Nov 2024 11:26:01 +0100
Subject: [PATCH] [BUGFIX] Show `Import data` if user is admin or owner (#5688)

FIx #5686
---
 argilla-frontend/pages/dataset/_id/annotation-mode/index.vue  | 1 +
 .../dataset/_id/annotation-mode/useAnnotationModeViewModel.ts | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/argilla-frontend/pages/dataset/_id/annotation-mode/index.vue b/argilla-frontend/pages/dataset/_id/annotation-mode/index.vue
index 234eb24306..31f4a01987 100644
--- a/argilla-frontend/pages/dataset/_id/annotation-mode/index.vue
+++ b/argilla-frontend/pages/dataset/_id/annotation-mode/index.vue
@@ -12,6 +12,7 @@
         >
           <template slot="dialog-cta" v-if="dataset && dataset.createdFromUI">
             <ImportData
+              v-if="isAdminOrOwner"
               :snippet="dataset.createCodeSnippetFromHub(getUser())"
             />
           </template>
diff --git a/argilla-frontend/pages/dataset/_id/annotation-mode/useAnnotationModeViewModel.ts b/argilla-frontend/pages/dataset/_id/annotation-mode/useAnnotationModeViewModel.ts
index 087c758267..f553303113 100644
--- a/argilla-frontend/pages/dataset/_id/annotation-mode/useAnnotationModeViewModel.ts
+++ b/argilla-frontend/pages/dataset/_id/annotation-mode/useAnnotationModeViewModel.ts
@@ -9,10 +9,11 @@ import { useDatasetViewModel } from "../useDatasetViewModel";
 import { GetDatasetByIdUseCase } from "@/v1/domain/usecases/get-dataset-by-id-use-case";
 import { useDataset } from "@/v1/infrastructure/storage/DatasetStorage";
 import { RecordCriteria } from "~/v1/domain/entities/record/RecordCriteria";
-import { useRoutes, useUser } from "~/v1/infrastructure/services";
+import { useRoutes, useUser, useRole } from "~/v1/infrastructure/services";
 import { RecordStatus } from "~/v1/domain/entities/record/RecordAnswer";
 
 export const useAnnotationModeViewModel = () => {
+  const { isAdminOrOwner } = useRole();
   const router = useRouter();
   const routes = useRoutes();
   const { getUser } = useUser();
@@ -113,5 +114,6 @@ export const useAnnotationModeViewModel = () => {
     breadcrumbs,
     updateQueryParams,
     getUser,
+    isAdminOrOwner,
   };
 };