From ad9740d72d219d01a95ca8a85a03ecbb796d04a7 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Sat, 1 Jun 2024 09:48:08 +1000
Subject: [PATCH] feat(ui): alt-click comparison image exits compare

---
 .../frontend/web/src/features/gallery/store/gallerySlice.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/invokeai/frontend/web/src/features/gallery/store/gallerySlice.ts b/invokeai/frontend/web/src/features/gallery/store/gallerySlice.ts
index fe17f45940..899e5db13d 100644
--- a/invokeai/frontend/web/src/features/gallery/store/gallerySlice.ts
+++ b/invokeai/frontend/web/src/features/gallery/store/gallerySlice.ts
@@ -38,7 +38,11 @@ export const gallerySlice = createSlice({
       state.selection = uniqBy(action.payload, (i) => i.image_name);
     },
     imageToCompareChanged: (state, action: PayloadAction<ImageDTO | null>) => {
-      state.imageToCompare = action.payload;
+      if (state.imageToCompare?.image_name === action.payload?.image_name) {
+        state.imageToCompare = null;
+      } else {
+        state.imageToCompare = action.payload;
+      }
       if (action.payload) {
         state.isImageViewerOpen = true;
       }