default keys bug fix + image to drag

3 files changed, 38 insertions, 6 deletions

diff --git a/src/client/apis/gpt/GPT.ts b/src/client/apis/gpt/GPT.ts
index 6bde7989b..d1606e7f0 100644
--- a/src/client/apis/gpt/GPT.ts
+++ b/src/client/apis/gpt/GPT.ts
@@ -1,3 +1,5 @@
+import { config } from "dotenv";
+config();
 import { Configuration, OpenAIApi } from 'openai';
 
 enum GPTCallType {
@@ -65,4 +67,33 @@ const gptImageCall = async (prompt: string, n?: number) => {
     }
 };
 
-export { gptAPICall, gptImageCall, GPTCallType };
+const gptCSVCall = async (inputText: string, callType: GPTCallType) => {
+    const opts: GPTCallOpts = callTypeMap[callType];
+    try {
+        const configuration = new Configuration({
+            apiKey: process.env.OPENAI_KEY,
+        });
+        const openai = new OpenAIApi(configuration);
+        // const client = OpenAIApi;
+        // const response = await openai.createCompletion({
+        //     model: opts.model,
+        //     max_tokens: opts.maxTokens,
+        //     temperature: opts.temp,
+        //     prompt: `${opts.prompt}${inputText}`,
+        // });
+        // return response.data.choices[0].text;
+
+        const responseGpt = await openai.createChatCompletion({
+            model: "gpt-3.5-turbo-16k",
+            messages: [{role: "user", content: inputText}],
+        })
+        const generatedResponse = responseGpt.data.choices[0].message?.content;
+        console.log(generatedResponse);
+        return generatedResponse;
+    } catch (err) {
+        console.log(err);
+        return 'Error connecting with API.';
+    }
+};
+
+export { gptAPICall, gptImageCall, gptCSVCall, GPTCallType };
diff --git a/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx b/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx
index 19689f972..3a1ea766a 100644
--- a/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx
+++ b/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx
@@ -1976,9 +1976,10 @@ ScriptingGlobals.add(function sendToBack(doc: Doc) {
 });
 ScriptingGlobals.add(function datavizFromSchema(doc: Doc) {
     SelectionManager.Views().forEach(view => {
-        const keys = Cast(view.rootDoc.schema_columnKeys, listSpec('string'))?.filter(key => key!="text");
-       
-        if (!keys) return;
+        var keys = Cast(view.layoutDoc.schema_columnKeys, listSpec('string'))?.filter(key => key!="text");
+        const defaultColumnKeys: string[] = ['title', 'type', 'author', 'author_date'];
+        if (!keys) keys = Cast(view.layoutDoc.schema_columnKeys, listSpec('string'), defaultColumnKeys);;
+
         const children = DocListCast(view.rootDoc[Doc.LayoutFieldKey(view.rootDoc)]);
         let csvRows = [];
         csvRows.push(keys.join(','));
diff --git a/src/client/views/nodes/DataVizBox/SchemaCSVPopUp.tsx b/src/client/views/nodes/DataVizBox/SchemaCSVPopUp.tsx
index 0114721a2..309297307 100644
--- a/src/client/views/nodes/DataVizBox/SchemaCSVPopUp.tsx
+++ b/src/client/views/nodes/DataVizBox/SchemaCSVPopUp.tsx
@@ -55,9 +55,9 @@ export class SchemaCSVPopUp extends React.Component<SchemaCSVPopUpProps> {
                 {this.heading('Schema Table as Data Visualization Doc')}
                 <div className="image-content-wrapper">
                         <div className="img-wrapper">
-                            <div className="img-container" style={{background: "blue"}} onPointerDown={e => this.drag(e)}>
+                            <div className="img-container" onPointerDown={e => this.drag(e)}>
                                 <img
-                                    width={150} height={150} 
+                                    width={150} height={150} src={"/assets/dataVizBox.png"}
                                 />
                             </div>
                         </div>