1 files changed, 13 insertions, 23 deletions
diff --git a/src/client/views/nodes/generativeFill/GenerativeFillButtons.tsx b/src/client/views/nodes/generativeFill/GenerativeFillButtons.tsx
index 348e27a16..e8cb61ab5 100644
--- a/src/client/views/nodes/generativeFill/GenerativeFillButtons.tsx
+++ b/src/client/views/nodes/generativeFill/GenerativeFillButtons.tsx
@@ -3,44 +3,34 @@ import { ImageUtility } from './generativeFillUtils/ImageHandler';
 import { canvasSize } from './generativeFillUtils/generativeFillConstants';
 import { Oval } from 'react-loader-spinner';
 import './GenerativeFillButtons.scss';
-import React from 'react';
+import React = require('react');
+import { Doc } from '../../../../fields/Doc';
 
 interface ButtonContainerProps {
     canvasRef: React.RefObject<HTMLCanvasElement>;
-    backgroundref: React.RefObject<HTMLCanvasElement>;
     currImg: React.MutableRefObject<HTMLImageElement | null>;
-    undoStack: React.MutableRefObject<ImageData[]>;
     getEdit: () => Promise<void>;
     loading: boolean;
+    onSave: () => Promise<void>;
+    onReset: () => void;
 }
 
-const Buttons = ({ canvasRef, backgroundref, currImg, undoStack, loading, getEdit }: ButtonContainerProps) => {
-    const handleReset = () => {
-        if (!canvasRef.current || !currImg.current) return;
-        const ctx = ImageUtility.getCanvasContext(canvasRef);
-        if (!ctx) return;
-        ctx.clearRect(0, 0, canvasSize, canvasSize);
-        ImageUtility.drawImgToCanvas(currImg.current, canvasRef, true);
+const Buttons = ({ canvasRef, currImg, loading, getEdit, onSave, onReset }: ButtonContainerProps) => {
+    const handleSave = () => {
+        onSave();
     };
 
     return (
         <div className="generativeFillBtnContainer">
-            <Button onClick={handleReset}>Reset</Button>
-            <Button
+            <Button onClick={onReset}>Reset</Button>
+            <Button onClick={handleSave}>Save</Button>
+            {/* <Button
                 onClick={() => {
                     if (!canvasRef.current) return;
-                    ImageUtility.downloadCanvas(canvasRef.current);
+                    ImageUtility.downloadImageCanvas('/assets/firefly.png');
                 }}>
-                Download
-            </Button>
-            {/* <Button
-        onClick={() => {
-          if (!canvasRef.current) return;
-          ImageUtility.downloadImageCanvas("/assets/firefly.png");
-        }}
-      >
-        Download Original
-      </Button> */}
+                Download Original
+            </Button> */}
             <Button
                 variant="contained"
                 onClick={() => {