From 71fa71e82c5ea6fa3c378991b2db7bbaa59e4c28 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Wed, 21 Aug 2024 14:19:27 -0400
Subject: [PATCH] Prefer saved media over preview media as task cover media
 (#576)

---
 src/stores/queueStore.ts | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

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 {