From 8576d3797b22a1f99e7a619d977b7f5b46dfd69e Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Sun, 18 Aug 2024 20:59:43 -0400
Subject: [PATCH] Prefer image for queue task preview (#520)

---
 src/components/sidebar/tabs/queue/ResultItem.vue | 3 +++
 src/components/sidebar/tabs/queue/TaskItem.vue   | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/components/sidebar/tabs/queue/ResultItem.vue b/src/components/sidebar/tabs/queue/ResultItem.vue
index 37b2f44a5..c0e9bbaba 100644
--- a/src/components/sidebar/tabs/queue/ResultItem.vue
+++ b/src/components/sidebar/tabs/queue/ResultItem.vue
@@ -52,6 +52,9 @@ onMounted(() => {
   aspect-ratio: 1 / 1;
   overflow: hidden;
   position: relative;
+  display: flex;
+  justify-content: center;
+  align-items: center;
 }
 
 .task-output-image {
diff --git a/src/components/sidebar/tabs/queue/TaskItem.vue b/src/components/sidebar/tabs/queue/TaskItem.vue
index d7f4110ef..558027f7e 100644
--- a/src/components/sidebar/tabs/queue/TaskItem.vue
+++ b/src/components/sidebar/tabs/queue/TaskItem.vue
@@ -4,7 +4,7 @@
       <template v-if="task.displayStatus === TaskItemDisplayStatus.Completed">
         <ResultItem
           v-if="flatOutputs.length"
-          :result="flatOutputs[0]"
+          :result="task.previewOutput || flatOutputs[0]"
           @preview="handlePreview"
         />
       </template>