made regenerate ai show up next to image doc. show ai prompt for firefly in gptpopup .

1 files changed, 6 insertions, 5 deletions

diff --git a/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx b/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx
index b33c267ee..752e8b217 100644
--- a/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx
+++ b/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx
@@ -1230,11 +1230,12 @@ export class CollectionFreeFormView extends CollectionSubView<Partial<collection
     };
 
     @action
-    showSmartDraw = (x: number, y: number) => {
-        SmartDrawHandler.Instance.CreateDrawingDoc = this.createDrawingDoc;
-        SmartDrawHandler.Instance.RemoveDrawing = this.removeDrawing;
-        SmartDrawHandler.Instance.AddDrawing = this.addDrawing;
-        SmartDrawHandler.Instance.displaySmartDrawHandler(x, y, NumCast(this.layoutDoc[this.scaleFieldKey]));
+    showSmartDraw = (x: number, y: number, regenerate?: boolean) => {
+        const sm = SmartDrawHandler.Instance;
+        sm.CreateDrawingDoc = this.createDrawingDoc;
+        sm.RemoveDrawing = this.removeDrawing;
+        sm.AddDrawing = this.addDrawing;
+        (regenerate ? sm.displayRegenerate : sm.displaySmartDrawHandler)(x, y, NumCast(this.layoutDoc[this.scaleFieldKey]));
     };
 
     _drawing: Doc[] = [];