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

1 files changed, 4 insertions, 2 deletions

diff --git a/src/client/documents/DocUtils.ts b/src/client/documents/DocUtils.ts
index f1de6fe36..18c8d97d4 100644
--- a/src/client/documents/DocUtils.ts
+++ b/src/client/documents/DocUtils.ts
@@ -33,7 +33,6 @@ import { TaskCompletionBox } from '../views/nodes/TaskCompletedBox';
 import { DocumentType } from './DocumentTypes';
 import { Docs, DocumentOptions } from './Documents';
 import { DocumentView } from '../views/nodes/DocumentView';
-import { CollectionFreeFormView } from '../views/collections/collectionFreeForm';
 
 export namespace DocUtils {
     function HasFunctionFilter(val: string) {
@@ -393,7 +392,10 @@ export namespace DocUtils {
             })) as ContextMenuProps[];
         documentList.push({
             description: ':Smart Drawing',
-            event: e => (DocumentView.Selected().lastElement().ComponentView as CollectionFreeFormView)?.showSmartDraw(e?.x || 0, e?.y || 0),
+            event: e =>
+                DocumentView.Selected()
+                    .lastElement()
+                    .ComponentView?.showSmartDraw?.(e?.x || 0, e?.y || 0),
             icon: 'file',
         });
         ContextMenu.Instance.addItem({