bug creating drawing groups

1 files changed, 8 insertions, 6 deletions

diff --git a/src/client/views/smartdraw/SmartDrawHandler.tsx b/src/client/views/smartdraw/SmartDrawHandler.tsx
index 6d2cc0593..d24cc9d50 100644
--- a/src/client/views/smartdraw/SmartDrawHandler.tsx
+++ b/src/client/views/smartdraw/SmartDrawHandler.tsx
@@ -43,7 +43,7 @@ export class SmartDrawHandler extends ObservableReactComponent<{}> {
     @observable private _size: number = 200;
     @observable private _autoColor: boolean = true;
     @observable private _regenInput: string = '';
-    private _addFunc: (e: React.PointerEvent<Element>, strokeList: [InkData, string, string][], opts: DrawingOptions, gptRes: string) => void = () => {};
+    private _addFunc: (e: React.PointerEvent<Element>, strokeList: [InkData, string, string][], opts: DrawingOptions, gptRes: string, containerDoc?: Doc) => void = () => {};
     private _deleteFunc: (doc?: Doc) => void = () => {};
     private _lastInput: DrawingOptions = { text: '', complexity: 5, size: 300, autoColor: true, x: 0, y: 0 };
     private _lastResponse: string = '';
@@ -86,7 +86,7 @@ export class SmartDrawHandler extends ObservableReactComponent<{}> {
     };
 
     @action
-    displaySmartDrawHandler = (x: number, y: number, addFunc: (e: React.PointerEvent<Element>, strokeData: [InkData, string, string][], opts: DrawingOptions, gptRes: string) => void, deleteFunc: (doc?: Doc) => void) => {
+    displaySmartDrawHandler = (x: number, y: number, addFunc: (e: React.PointerEvent<Element>, strokeData: [InkData, string, string][], opts: DrawingOptions, gptRes: string, containerDoc?: Doc) => void, deleteFunc: (doc?: Doc) => void) => {
         this._pageX = x;
         this._pageY = y;
         this._display = true;
@@ -95,9 +95,9 @@ export class SmartDrawHandler extends ObservableReactComponent<{}> {
     };
 
     @action
-    displayRegenerate = (x: number, y: number, addFunc: (e: React.PointerEvent<Element>, strokeData: [InkData, string, string][], opts: DrawingOptions, gptRes: string) => void, deleteFunc: (doc?: Doc) => void) => {
-        const selectedDoc: Doc = DocumentView.SelectedDocs().lastElement();
-        const docData = selectedDoc[DocData];
+    displayRegenerate = (x: number, y: number, addFunc: (e: React.PointerEvent<Element>, strokeData: [InkData, string, string][], opts: DrawingOptions, gptRes: string, containerDoc?: Doc) => void, deleteFunc: (doc?: Doc) => void) => {
+        this._selectedDoc = DocumentView.SelectedDocs().lastElement();
+        const docData = this._selectedDoc[DocData];
         this._addFunc = addFunc;
         this._deleteFunc = deleteFunc;
         this._pageX = x;
@@ -210,8 +210,10 @@ export class SmartDrawHandler extends ObservableReactComponent<{}> {
             });
             if (regenerate) {
                 this._deleteFunc(this._selectedDoc);
+                this._addFunc(e, strokeData, this._lastInput, svg[0], this._selectedDoc);
+            } else {
+                this._addFunc(e, strokeData, this._lastInput, svg[0]);
             }
-            this._addFunc(e, strokeData, this._lastInput, svg[0]);
         }
     };