diff --git a/src/stores/queueStore.ts b/src/stores/queueStore.ts index 70e17ec8d..34a66d5c1 100644 --- a/src/stores/queueStore.ts +++ b/src/stores/queueStore.ts @@ -23,6 +23,7 @@ import { api } from '@/scripts/api' import type { ComfyApp } from '@/scripts/app' import { useExtensionService } from '@/services/extensionService' import { useNodeOutputStore } from '@/stores/imagePreviewStore' +import { getMediaTypeFromFilename } from '@/utils/formatUtil' // Task type used in the API. type APITaskType = 'queue' | 'history' @@ -202,8 +203,12 @@ export class ResultItemImpl { ) } + get is3D(): boolean { + return getMediaTypeFromFilename(this.filename) === '3D' + } + get supportsPreview(): boolean { - return this.isImage || this.isVideo || this.isAudio + return this.isImage || this.isVideo || this.isAudio || this.is3D } }