From 85be2532c6d6567f0149632d153baeae0e68f1c4 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Sat, 17 Aug 2024 16:09:25 +1000
Subject: [PATCH] fix(ui): respect image size in staging preview

---
 .../web/src/features/controlLayers/konva/CanvasStagingArea.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/invokeai/frontend/web/src/features/controlLayers/konva/CanvasStagingArea.ts b/invokeai/frontend/web/src/features/controlLayers/konva/CanvasStagingArea.ts
index 2ce21d239e..e07833ce07 100644
--- a/invokeai/frontend/web/src/features/controlLayers/konva/CanvasStagingArea.ts
+++ b/invokeai/frontend/web/src/features/controlLayers/konva/CanvasStagingArea.ts
@@ -3,7 +3,7 @@ import { CanvasImageRenderer } from 'features/controlLayers/konva/CanvasImage';
 import type { CanvasManager } from 'features/controlLayers/konva/CanvasManager';
 import type { CanvasPreview } from 'features/controlLayers/konva/CanvasPreview';
 import { getPrefixedId } from 'features/controlLayers/konva/util';
-import type { StagingAreaImage } from 'features/controlLayers/store/types';
+import { imageDTOToImageWithDims, type StagingAreaImage } from 'features/controlLayers/store/types';
 import Konva from 'konva';
 import type { Logger } from 'roarr';
 
@@ -75,7 +75,7 @@ export class CanvasStagingArea {
       }
 
       if (!this.image.isLoading && !this.image.isError) {
-        await this.image.updateImageSource(imageDTO.image_name);
+        await this.image.update({...this.image.state, image: imageDTOToImageWithDims(imageDTO)}, true);
         this.manager.stateApi.$lastCanvasProgressEvent.set(null);
       }
       this.image.konva.group.visible(shouldShowStagedImage);