From 982991ca69cd7f844680949ed160e678f89050fc Mon Sep 17 00:00:00 2001 From: srichman333 Date: Tue, 16 Apr 2024 00:44:43 -0400 Subject: correlation from gpt --- src/client/views/nodes/DataVizBox/DataVizBox.tsx | 25 ++++++++++++++++++++++++ 1 file changed, 25 insertions(+) (limited to 'src/client/views/nodes/DataVizBox/DataVizBox.tsx') 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() 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() im GPTPopup.Instance.generateDataAnalysis(); }); + createFilteredDoc = (axes?: any, type?: DataVizView) => { + + const embedding = Doc.MakeEmbedding(this.Document!); + embedding._dataViz = DataVizView.HISTOGRAM; + embedding._dataViz_axes = new List(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([this.axes[1]]) + this.layoutDoc.dataViz_selectedRows = new List(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 ? ( -- cgit v1.2.3-70-g09d2 From b98f3cb1c0e356c318e4d629e8baa05648c153e2 Mon Sep 17 00:00:00 2001 From: srichman333 Date: Tue, 16 Apr 2024 17:02:23 -0400 Subject: makes histogram on transfer to text --- src/client/views/nodes/DataVizBox/DataVizBox.tsx | 3 +-- src/client/views/pdf/GPTPopup/GPTPopup.tsx | 7 +++++-- 2 files changed, 6 insertions(+), 4 deletions(-) (limited to 'src/client/views/nodes/DataVizBox/DataVizBox.tsx') diff --git a/src/client/views/nodes/DataVizBox/DataVizBox.tsx b/src/client/views/nodes/DataVizBox/DataVizBox.tsx index 24199a5e3..19fde7273 100644 --- a/src/client/views/nodes/DataVizBox/DataVizBox.tsx +++ b/src/client/views/nodes/DataVizBox/DataVizBox.tsx @@ -437,7 +437,7 @@ export class DataVizBox extends ViewBoxAnnotatableComponent() im embedding.defaultHistogramColor = this.layoutDoc.defaultHistogramColor; embedding.pieSliceColors = Field.Copy(this.layoutDoc.pieSliceColors); this._props.addDocument?.(embedding); - embedding._dataViz_axes = new List([this.axes[1]]) + embedding._dataViz_axes = new List(axes) this.layoutDoc.dataViz_selectedRows = new List(this.records.map((rec, i) => i)) console.log(embedding.x); @@ -445,7 +445,6 @@ export class DataVizBox extends ViewBoxAnnotatableComponent() im 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; }; diff --git a/src/client/views/pdf/GPTPopup/GPTPopup.tsx b/src/client/views/pdf/GPTPopup/GPTPopup.tsx index 50835a541..706b44aef 100644 --- a/src/client/views/pdf/GPTPopup/GPTPopup.tsx +++ b/src/client/views/pdf/GPTPopup/GPTPopup.tsx @@ -30,6 +30,7 @@ interface GPTPopupProps {} export class GPTPopup extends ObservableReactComponent { static Instance: GPTPopup; @observable private chatMode: boolean = false; + private correlatedColumns: string[] = [] @observable public visible: boolean = false; @@ -173,7 +174,9 @@ export class GPTPopup extends ObservableReactComponent { console.log(res) let json = JSON.parse(res! as string); const keys = Object.keys(json) - console.log(json[keys[0]], json[keys[1]]) + this.correlatedColumns = [] + this.correlatedColumns.push(json[keys[0]]) + this.correlatedColumns.push(json[keys[1]]) GPTPopup.Instance.setText(json[keys[2]] || 'Something went wrong.'); } catch (err) { console.error(err); @@ -199,7 +202,7 @@ export class GPTPopup extends ObservableReactComponent { }); } - this.createFilteredDoc(); + this.createFilteredDoc(this.correlatedColumns); }; /** -- cgit v1.2.3-70-g09d2 From 17cca384290a98b28c4f4a4d6bc4cf5edaef4573 Mon Sep 17 00:00:00 2001 From: srichman333 Date: Tue, 16 Apr 2024 17:43:06 -0400 Subject: visualize button --- src/client/views/nodes/DataVizBox/DataVizBox.tsx | 5 ----- src/client/views/pdf/GPTPopup/GPTPopup.tsx | 6 ++++++ 2 files changed, 6 insertions(+), 5 deletions(-) (limited to 'src/client/views/nodes/DataVizBox/DataVizBox.tsx') diff --git a/src/client/views/nodes/DataVizBox/DataVizBox.tsx b/src/client/views/nodes/DataVizBox/DataVizBox.tsx index 19fde7273..dbba9c7f3 100644 --- a/src/client/views/nodes/DataVizBox/DataVizBox.tsx +++ b/src/client/views/nodes/DataVizBox/DataVizBox.tsx @@ -439,12 +439,7 @@ export class DataVizBox extends ViewBoxAnnotatableComponent() im this._props.addDocument?.(embedding); embedding._dataViz_axes = new List(axes) this.layoutDoc.dataViz_selectedRows = new List(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); return true; }; diff --git a/src/client/views/pdf/GPTPopup/GPTPopup.tsx b/src/client/views/pdf/GPTPopup/GPTPopup.tsx index 706b44aef..686ef9c28 100644 --- a/src/client/views/pdf/GPTPopup/GPTPopup.tsx +++ b/src/client/views/pdf/GPTPopup/GPTPopup.tsx @@ -201,7 +201,12 @@ export class GPTPopup extends ObservableReactComponent { link_relationship: 'GPT Summary', }); } + }; + /** + * Creates a histogram to show the correlation relationship that was found + */ + private createVisualization = () => { this.createFilteredDoc(this.correlatedColumns); }; @@ -373,6 +378,7 @@ export class GPTPopup extends ObservableReactComponent { :( <>