diff --git a/src/platform/assets/components/AssetBrowserModal.vue b/src/platform/assets/components/AssetBrowserModal.vue
index 7ffc043b7b..8a61e261d2 100644
--- a/src/platform/assets/components/AssetBrowserModal.vue
+++ b/src/platform/assets/components/AssetBrowserModal.vue
@@ -34,6 +34,7 @@
type="accent"
size="md"
class="!h-10 [&>span]:hidden md:[&>span]:inline"
+ data-attr="upload-model-button"
:label="$t('assetBrowser.uploadModel')"
:on-click="showUploadDialog"
>
diff --git a/src/platform/assets/components/UploadModelConfirmation.vue b/src/platform/assets/components/UploadModelConfirmation.vue
index d3f6325236..4a0b8edc5e 100644
--- a/src/platform/assets/components/UploadModelConfirmation.vue
+++ b/src/platform/assets/components/UploadModelConfirmation.vue
@@ -26,6 +26,7 @@
"
:options="modelTypes"
:disabled="isLoading"
+ data-attr="upload-model-step2-type-selector"
/>
diff --git a/src/platform/assets/components/UploadModelFooter.vue b/src/platform/assets/components/UploadModelFooter.vue
index 019e68ac2e..7bb4cc966d 100644
--- a/src/platform/assets/components/UploadModelFooter.vue
+++ b/src/platform/assets/components/UploadModelFooter.vue
@@ -5,15 +5,20 @@
class="text-muted-foreground mr-auto underline flex items-center gap-2"
>
-
{{
- $t('How do I find this?')
- }}
+
{{ $t('How do I find this?') }}
@@ -22,6 +27,7 @@
:label="$t('g.back')"
type="transparent"
size="md"
+ :data-attr="`upload-model-step${currentStep}-back-button`"
:disabled="isFetchingMetadata || isUploading"
@click="emit('back')"
/>
@@ -32,6 +38,7 @@
:label="$t('g.continue')"
type="secondary"
size="md"
+ data-attr="upload-model-step1-continue-button"
:disabled="!canFetchMetadata || isFetchingMetadata"
@click="emit('fetchMetadata')"
>
@@ -47,6 +54,7 @@
:label="$t('assetBrowser.upload')"
type="secondary"
size="md"
+ data-attr="upload-model-step2-confirm-button"
:disabled="!canUploadModel || isUploading"
@click="emit('upload')"
>
@@ -62,6 +70,7 @@
:label="$t('assetBrowser.finish')"
type="secondary"
size="md"
+ data-attr="upload-model-step3-finish-button"
@click="emit('close')"
/>
diff --git a/src/platform/assets/components/UploadModelUrlInput.vue b/src/platform/assets/components/UploadModelUrlInput.vue
index e646b09ffe..c7e94a8be1 100644
--- a/src/platform/assets/components/UploadModelUrlInput.vue
+++ b/src/platform/assets/components/UploadModelUrlInput.vue
@@ -17,6 +17,7 @@
autofocus
:placeholder="$t('assetBrowser.civitaiLinkPlaceholder')"
class="w-full bg-secondary-background border-0 p-4"
+ data-attr="upload-model-step1-url-input"
/>