diff --git a/src/stores/queueStore.ts b/src/stores/queueStore.ts index f79d4c373..c522b85f6 100644 --- a/src/stores/queueStore.ts +++ b/src/stores/queueStore.ts @@ -10,7 +10,7 @@ import type { ResultItem } from '@/types/apiTypes' import type { NodeId } from '@/types/comfyWorkflow' -import { instanceToPlain, plainToClass } from 'class-transformer' +import { plainToClass } from 'class-transformer' import _ from 'lodash' import { defineStore } from 'pinia' import { toRaw } from 'vue' @@ -88,7 +88,12 @@ export class TaskItemImpl { } get previewOutput(): ResultItemImpl | undefined { - return this.flatOutputs.find((output) => output.supportsPreview) + return ( + this.flatOutputs.find( + // Prefer saved media files over the temp previews + (output) => output.type === 'output' && output.supportsPreview + ) ?? this.flatOutputs.find((output) => output.supportsPreview) + ) } get apiTaskType(): APITaskType {