From 5b1099193d827b5c753c79dd7ab57b94984f0bc9 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Wed, 16 Aug 2023 11:49:00 +1000
Subject: [PATCH] fix(ui): restore reset button in node image component

---
 .../components/fields/fieldTypes/ImageInputField.tsx   | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/invokeai/frontend/web/src/features/nodes/components/fields/fieldTypes/ImageInputField.tsx b/invokeai/frontend/web/src/features/nodes/components/fields/fieldTypes/ImageInputField.tsx
index e5bb06c4c6..f9f9c404d7 100644
--- a/invokeai/frontend/web/src/features/nodes/components/fields/fieldTypes/ImageInputField.tsx
+++ b/invokeai/frontend/web/src/features/nodes/components/fields/fieldTypes/ImageInputField.tsx
@@ -2,6 +2,7 @@ import { Flex } from '@chakra-ui/react';
 import { skipToken } from '@reduxjs/toolkit/dist/query';
 import { useAppDispatch } from 'app/store/storeHooks';
 import IAIDndImage from 'common/components/IAIDndImage';
+import IAIDndImageIcon from 'common/components/IAIDndImageIcon';
 import {
   TypesafeDraggableData,
   TypesafeDroppableData,
@@ -12,6 +13,7 @@ import {
   ImageInputFieldValue,
 } from 'features/nodes/types/types';
 import { memo, useCallback, useMemo } from 'react';
+import { FaUndo } from 'react-icons/fa';
 import { useGetImageDTOQuery } from 'services/api/endpoints/images';
 import { PostUploadAction } from 'services/api/types';
 import { FieldComponentProps } from './types';
@@ -80,7 +82,13 @@ const ImageInputFieldComponent = (
         draggableData={draggableData}
         postUploadAction={postUploadAction}
         useThumbailFallback
-      />
+      >
+        <IAIDndImageIcon
+          onClick={handleReset}
+          icon={imageDTO ? <FaUndo /> : undefined}
+          tooltip="Reset Image"
+        />
+      </IAIDndImage>
     </Flex>
   );
 };