1 files changed, 11 insertions, 0 deletions
diff --git a/src/client/views/MainView.tsx b/src/client/views/MainView.tsx
index c0c473cfc..1fa1eff84 100644
--- a/src/client/views/MainView.tsx
+++ b/src/client/views/MainView.tsx
@@ -65,6 +65,7 @@ import { PreviewCursor } from './PreviewCursor';
 import { PropertiesView } from './PropertiesView';
 import { DashboardStyleProvider, DefaultStyleProvider } from './StyleProvider';
 import { TopBar } from './topbar/TopBar';
+import GenerativeFill from './nodes/generativeFill/GenerativeFill';
 const _global = (window /* browser */ || global) /* node */ as any;
 
 @observer
@@ -72,6 +73,15 @@ export class MainView extends React.Component {
     public static Instance: MainView;
     public static Live: boolean = false;
     private _docBtnRef = React.createRef<HTMLDivElement>();
+    // for ai image editor
+    @observable public imageEditorOpen: boolean = false;
+    @action public setImageEditorOpen = (open: boolean) => (this.imageEditorOpen = open);
+    @observable public imageEditorSource: string = '';
+    @action public setImageEditorSource = (source: string) => (this.imageEditorSource = source);
+    @observable public imageRootDoc: Doc | undefined;
+    @action public setImageRootDoc = (doc: Doc) => (this.imageRootDoc = doc);
+    @observable public addDoc: ((doc: Doc | Doc[], annotationKey?: string) => boolean) | undefined;
+
     @observable public LastButton: Opt<Doc>;
     @observable private _windowWidth: number = 0;
     @observable private _windowHeight: number = 0;
@@ -1014,6 +1024,7 @@ export class MainView extends React.Component {
                 <InkTranscription />
                 {this.snapLines}
                 <LightboxView key="lightbox" PanelWidth={this._windowWidth} PanelHeight={this._windowHeight} maxBorder={[200, 50]} />
+                <GenerativeFill imageEditorOpen={this.imageEditorOpen} imageEditorSource={this.imageEditorSource} imageRootDoc={this.imageRootDoc} addDoc={this.addDoc} />
             </div>
         );
     }