correlation from gpt

1 files changed, 25 insertions, 0 deletions

diff --git a/src/client/views/nodes/DataVizBox/DataVizBox.tsx b/src/client/views/nodes/DataVizBox/DataVizBox.tsx
index 60c5fdba2..24199a5e3 100644
--- a/src/client/views/nodes/DataVizBox/DataVizBox.tsx
+++ b/src/client/views/nodes/DataVizBox/DataVizBox.tsx
@@ -29,6 +29,7 @@ import { PieChart } from './components/PieChart';
 import { TableBox } from './components/TableBox';
 import { Checkbox } from '@mui/material';
 import { ContextMenu } from '../../ContextMenu';
+import { DragManager } from '../../../util/DragManager';
 
 export enum DataVizView {
     TABLE = 'table',
@@ -417,6 +418,7 @@ export class DataVizBox extends ViewBoxAnnotatableComponent<FieldViewProps>() im
     askGPT = action(async () => {
         GPTPopup.Instance.setSidebarId('data_sidebar');
         GPTPopup.Instance.addDoc = this.sidebarAddDocument;
+        GPTPopup.Instance.createFilteredDoc = this.createFilteredDoc;
         GPTPopup.Instance.setDataJson("");
         GPTPopup.Instance.setMode(GPTPopupMode.DATA);
         let data = DataVizBox.dataset.get(CsvCast(this.dataDoc[this.fieldKey]).url.href);
@@ -425,6 +427,29 @@ export class DataVizBox extends ViewBoxAnnotatableComponent<FieldViewProps>() im
         GPTPopup.Instance.generateDataAnalysis();
     });
 
+    createFilteredDoc = (axes?: any, type?: DataVizView) => {
+
+        const embedding = Doc.MakeEmbedding(this.Document!);
+        embedding._dataViz = DataVizView.HISTOGRAM;
+        embedding._dataViz_axes = new List<string>(axes);
+        embedding._dataViz_parentViz = this.Document;
+        embedding.histogramBarColors = Field.Copy(this.layoutDoc.histogramBarColors);
+        embedding.defaultHistogramColor = this.layoutDoc.defaultHistogramColor;
+        embedding.pieSliceColors = Field.Copy(this.layoutDoc.pieSliceColors);
+        this._props.addDocument?.(embedding);
+        embedding._dataViz_axes = new List<string>([this.axes[1]])
+        this.layoutDoc.dataViz_selectedRows = new List<number>(this.records.map((rec, i) => i))
+
+        console.log(embedding.x);
+        console.log(Number(embedding.x));
+        console.log(Number(embedding.x) + 100.0)
+        embedding.x = Number(embedding.x) + 100.0;
+        console.log(embedding.x);
+        // embedding.y = StrCast(Number(embedding.y) + 100);
+
+        return true;
+    };
+
     render() {
         const scale = this._props.NativeDimScaling?.() || 1;
         return !this.records.length ? (