docs loading properly into preview

1 files changed, 15 insertions, 9 deletions

diff --git a/src/client/views/nodes/DataVizBox/DocCreatorMenu/DocCreatorMenu.tsx b/src/client/views/nodes/DataVizBox/DocCreatorMenu/DocCreatorMenu.tsx
index 24a05fd71..ad7aa7cb9 100644
--- a/src/client/views/nodes/DataVizBox/DocCreatorMenu/DocCreatorMenu.tsx
+++ b/src/client/views/nodes/DataVizBox/DocCreatorMenu/DocCreatorMenu.tsx
@@ -114,11 +114,6 @@ export class DocCreatorMenu extends ObservableReactComponent<FieldViewProps> {
     };
     @action setGSuggestedTemplates = (docs: Doc[]) => {
         this._suggestedTemplates = docs;
-
-        const mainCollection = this._dataViz?.DocumentView?.().containerViewPath?.().lastElement()?.ComponentView as CollectionFreeFormView;
-        docs.forEach(doc => {
-            mainCollection.addDocument(doc);
-        });
     };
 
     @computed get docsToRender() {
@@ -415,8 +410,10 @@ export class DocCreatorMenu extends ObservableReactComponent<FieldViewProps> {
     };
 
     testTemplate = async () => {
-        
-        console.log(this.templateManager.templates);
+
+        console.log(this._suggestedTemplates
+            ?.map(doc => ({ icon: ImageCast(doc.icon), doc }))
+            .filter(info => info.icon && info.doc));
 
         this.forceUpdate();
 
@@ -672,6 +669,7 @@ export class DocCreatorMenu extends ObservableReactComponent<FieldViewProps> {
                     );
                     brokenDownAssignments.push([template, toObj]);
                 });
+
                 return brokenDownAssignments;
             }
         } catch (err) {
@@ -691,10 +689,18 @@ export class DocCreatorMenu extends ObservableReactComponent<FieldViewProps> {
 
         const renderedTemplatePromises: Promise<Template>[] = assignments.map(([template, assignments]) => this.applyGPTContentToTemplate(template, assignments));
 
-        const renderedTemplates: Template[] = await Promise.all(renderedTemplatePromises);
+        //const renderedTemplates: Template[] = await Promise.all(renderedTemplatePromises);
+
+        const renderedTemplates: Doc[] = templates.map(template => template.mainField.renderedDoc());
+
+        const mainCollection = this._dataViz?.DocumentView?.().containerViewPath?.().lastElement()?.ComponentView as CollectionFreeFormView;
+        
+        renderedTemplates.forEach(doc => {
+            mainCollection.addDocument(doc);
+        });
 
         setTimeout(() => {
-            this.setGSuggestedTemplates(templates.map(template => template.mainField.renderedDoc()));
+            this.setGSuggestedTemplates(renderedTemplates);
             this._GPTLoading = false;
         });
     };