From f7b249252d5984c1826fb61ce9802c1365a290ef Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Wed, 25 Jun 2025 13:51:29 +1000
Subject: [PATCH] fix(ui): issues with progress viewer

---
 .../gallery/components/ImageViewer/CurrentImageButtons.tsx       | 1 -
 .../gallery/components/ImageViewer/CurrentImagePreview.tsx       | 1 +
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImageButtons.tsx b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImageButtons.tsx
index 987b6cd911..8e92c37923 100644
--- a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImageButtons.tsx
+++ b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImageButtons.tsx
@@ -30,7 +30,6 @@ export const CurrentImageButtons = memo(() => {
   const ctx = useImageViewerContext();
   const hasProgressImage = useStore(ctx.$hasProgressImage);
   const shouldShowProgressInViewer = useAppSelector(selectShouldShowProgressInViewer);
-
   const isDisabledOverride = hasProgressImage && shouldShowProgressInViewer;
 
   const imageName = useAppSelector(selectLastSelectedImage);
diff --git a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImagePreview.tsx b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImagePreview.tsx
index 1b4ce009f2..c57781b207 100644
--- a/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImagePreview.tsx
+++ b/invokeai/frontend/web/src/features/gallery/components/ImageViewer/CurrentImagePreview.tsx
@@ -88,6 +88,7 @@ export const CurrentImagePreview = memo(({ imageDTO }: { imageDTO: ImageDTO | nu
       return;
     }
     if (progressEvent.session_id === imageDTO.session_id) {
+      setProgressEvent(null);
       setProgressImage(null);
     }
   }, [imageDTO, progressEvent]);