1 files changed, 6 insertions, 6 deletions
diff --git a/src/client/views/nodes/generativeFill/GenerativeFill.tsx b/src/client/views/nodes/generativeFill/GenerativeFill.tsx
index 06819d08c..547a76348 100644
--- a/src/client/views/nodes/generativeFill/GenerativeFill.tsx
+++ b/src/client/views/nodes/generativeFill/GenerativeFill.tsx
@@ -33,7 +33,7 @@ interface GenerativeFillProps {
     addDoc: ((doc: Doc | Doc[], annotationKey?: string) => boolean) | undefined;
 }
 
-// New field on image doc: gen_fill_children => list of children Docs
+// Added field on image doc: gen_fill_children: List of children Docs
 
 const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addDoc }: GenerativeFillProps) => {
     const canvasRef = useRef<HTMLCanvasElement>(null);
@@ -191,6 +191,7 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
         };
     }, [canvasRef, imageEditorSource]);
 
+    // once the appropriate dimensions are set, draw the image to the canvas
     useEffect(() => {
         if (!currImg.current) return;
         ImageUtility.drawImgToCanvas(currImg.current, canvasRef, canvasDims.width, canvasDims.height);
@@ -338,7 +339,6 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
         if (firstDoc) {
             const x = 0;
             const initialY = 0;
-
             const newImg = Docs.Create.ImageDocument(source, {
                 x: x,
                 y: initialY,
@@ -347,7 +347,6 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
                 data_nativeWidth: result.nativeWidth,
                 data_nativeHeight: result.nativeHeight,
             });
-            // add a new doc list field to newimg
             if (isNewCollection && newCollectionRef.current) {
                 Doc.AddDocToList(newCollectionRef.current, undefined, newImg);
             } else {
@@ -389,6 +388,7 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
         }
     };
 
+    // Saves an image to the collection
     const onSave = async (src: string) => {
         const img = new Image();
         img.src = src;
@@ -401,13 +401,13 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
         }
     };
 
+    // Closes the editor view
     const handleViewClose = () => {
         MainView.Instance.setImageEditorOpen(false);
         MainView.Instance.setImageEditorSource('');
         if (newCollectionRef.current) {
             newCollectionRef.current.fitContentOnce = true;
         }
-
         setEdits([]);
     };
 
@@ -419,7 +419,7 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
                     <FormControlLabel
                         control={
                             <Checkbox
-                                // disable once edited has been clicked
+                                // disable once edited has been clicked (doesn't make sense to change after first edit)
                                 disabled={edited}
                                 checked={isNewCollection}
                                 onChange={e => {
@@ -431,7 +431,7 @@ const GenerativeFill = ({ imageEditorOpen, imageEditorSource, imageRootDoc, addD
                         labelPlacement="end"
                         sx={{ whiteSpace: 'nowrap' }}
                     />
-                    <Buttons canvasRef={canvasRef} currImg={currImg} getEdit={getEdit} loading={loading} onReset={handleReset} />
+                    <Buttons getEdit={getEdit} loading={loading} onReset={handleReset} />
                     <IconButton onClick={handleViewClose}>
                         <BsX color={activeColor} />
                     </IconButton>