From c94742bde678a5993f908fb7d61c8fa4528e5a67 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Mon, 20 May 2024 09:35:34 +1000
Subject: [PATCH] feat(ui): add canvas objects to metadata when saving canvas
 to gallery

---
 .../listenerMiddleware/listeners/canvasSavedToGallery.ts      | 4 ++++
 invokeai/frontend/web/src/services/api/endpoints/images.ts    | 4 +++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/canvasSavedToGallery.ts b/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/canvasSavedToGallery.ts
index e3ba988886..7f456e9a68 100644
--- a/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/canvasSavedToGallery.ts
+++ b/invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/canvasSavedToGallery.ts
@@ -1,5 +1,6 @@
 import { logger } from 'app/logging/logger';
 import type { AppStartListening } from 'app/store/middleware/listenerMiddleware';
+import { parseify } from 'common/util/serialize';
 import { canvasSavedToGallery } from 'features/canvas/store/actions';
 import { getBaseLayerBlob } from 'features/canvas/util/getBaseLayerBlob';
 import { addToast } from 'features/system/store/systemSlice';
@@ -43,6 +44,9 @@ export const addCanvasSavedToGalleryListener = (startAppListening: AppStartListe
             type: 'TOAST',
             toastOptions: { title: t('toast.canvasSavedGallery') },
           },
+          metadata: {
+            _canvas_objects: parseify(state.canvas.layerState.objects),
+          },
         })
       );
     },
diff --git a/invokeai/frontend/web/src/services/api/endpoints/images.ts b/invokeai/frontend/web/src/services/api/endpoints/images.ts
index 98c253b479..14edf6fb87 100644
--- a/invokeai/frontend/web/src/services/api/endpoints/images.ts
+++ b/invokeai/frontend/web/src/services/api/endpoints/images.ts
@@ -571,11 +571,13 @@ export const imagesApi = api.injectEndpoints({
         session_id?: string;
         board_id?: string;
         crop_visible?: boolean;
+        metadata?: JSONObject;
       }
     >({
-      query: ({ file, image_category, is_intermediate, session_id, board_id, crop_visible }) => {
+      query: ({ file, image_category, is_intermediate, session_id, board_id, crop_visible, metadata }) => {
         const formData = new FormData();
         formData.append('file', file);
+        formData.append('metadata', JSON.stringify(metadata));
         return {
           url: buildImagesUrl('upload'),
           method: 'POST',