From e1a66e22e9015360366bcc04f0843d5ceec9b448 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Fri, 6 Sep 2024 18:44:46 +1000
Subject: [PATCH] feat(ui): fit bbox to stage on canvas reset

---
 .../components/Settings/CanvasSettingsResetButton.tsx        | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/invokeai/frontend/web/src/features/controlLayers/components/Settings/CanvasSettingsResetButton.tsx b/invokeai/frontend/web/src/features/controlLayers/components/Settings/CanvasSettingsResetButton.tsx
index 47c5a38084..32a409a0e0 100644
--- a/invokeai/frontend/web/src/features/controlLayers/components/Settings/CanvasSettingsResetButton.tsx
+++ b/invokeai/frontend/web/src/features/controlLayers/components/Settings/CanvasSettingsResetButton.tsx
@@ -1,5 +1,6 @@
 import { Button } from '@invoke-ai/ui-library';
 import { useAppDispatch } from 'app/store/storeHooks';
+import { useCanvasManager } from 'features/controlLayers/contexts/CanvasManagerProviderGate';
 import { canvasReset } from 'features/controlLayers/store/canvasSlice';
 import { memo, useCallback } from 'react';
 import { useTranslation } from 'react-i18next';
@@ -7,9 +8,11 @@ import { useTranslation } from 'react-i18next';
 export const CanvasSettingsResetButton = memo(() => {
   const { t } = useTranslation();
   const dispatch = useAppDispatch();
+  const canvasManager = useCanvasManager();
   const onClick = useCallback(() => {
     dispatch(canvasReset());
-  }, [dispatch]);
+    canvasManager.stage.fitLayersToStage();
+  }, [canvasManager.stage, dispatch]);
   return (
     <Button onClick={onClick} colorScheme="error" size="sm">
       {t('controlLayers.resetCanvas')}