fix app crashign when document is undefined

2024-08-30 20:32:17 +00:00 · 2024-01-11 17:09:05 -05:00
1 changed files with 3 additions and 5 deletions
--- a/invokeai/frontend/web/src/features/gallery/hooks/useGalleryNavigation.ts
+++ b/invokeai/frontend/web/src/features/gallery/hooks/useGalleryNavigation.ts
@ -29,13 +29,11 @@ import { imagesSelectors } from 'services/api/util';
 */
 const getImagesPerRow = (): number => {
  const widthOfGalleryImage =
-    document
-      .querySelector(`[data-testid="${imageItemContainerTestId}"]`)
+    document?.querySelector(`[data-testid="${imageItemContainerTestId}"]`)
      ?.getBoundingClientRect().width ?? 1;

  const widthOfGalleryGrid =
-    document
-      .querySelector(`[data-testid="${imageListContainerTestId}"]`)
+    document?.querySelector(`[data-testid="${imageListContainerTestId}"]`)
      ?.getBoundingClientRect().width ?? 0;

  const imagesPerRow = Math.round(widthOfGalleryGrid / widthOfGalleryImage);
@ -59,7 +57,7 @@ const scrollToImage = (imageName: string, index: number) => {
    return;
  }

-  const imageElement = document.querySelector(
+  const imageElement = document?.querySelector(
    `[data-testid="${getGalleryImageDataTestId(imageName)}"]`
  );
  const itemRect = imageElement?.getBoundingClientRect();