small change in gpt prompty

2 files changed, 36 insertions, 4 deletions

diff --git a/src/client/apis/gpt/GPT.ts b/src/client/apis/gpt/GPT.ts
index 027c10e28..cde408382 100644
--- a/src/client/apis/gpt/GPT.ts
+++ b/src/client/apis/gpt/GPT.ts
@@ -19,7 +19,7 @@ const callTypeMap: { [type: string]: GPTCallOpts } = {
     summary: { model: 'gpt-3.5-turbo-instruct', maxTokens: 256, temp: 0.5, prompt: 'Summarize this text in simpler terms: ' },
     edit: { model: 'gpt-3.5-turbo-instruct', maxTokens: 256, temp: 0.5, prompt: 'Reword this: ' },
     completion: { model: 'gpt-3.5-turbo-instruct', maxTokens: 256, temp: 0.5, prompt: '' },
-    mermaid:{model:'gpt-4-turbo',maxTokens:2048,temp:0,prompt:"Write this in mermaid code and only give me the mermaid code (Heres an example of changing color of a pie chart to help you pie title Example \"Red\": 20 \"Blue\": 50 \"Green\": 30 %%{init: {'theme': 'base', 'themeVariables': {'pie1': '#0000FF', 'pie2': '#00FF00', 'pie3': '#FF0000'}}}%% keep in mind that pie1 is the highest since its sorted in descending order. Heres an example of a mindmap: mindmap  root((mindmap))    Origins      Long history      ::icon(fa fa-book)      Popularisation        British popular psychology author Tony Buzan    Research      On effectivness<br/>and features      On Automatic creation       Uses            Creative techniques            Strategic planning            Argument mapping    Tools      Pen and paper     Mermaid.  "}
+    mermaid:{model:'gpt-4-turbo',maxTokens:2048,temp:0,prompt:"(Heres an example of changing color of a pie chart to help you pie title Example \"Red\": 20 \"Blue\": 50 \"Green\": 30 %%{init: {'theme': 'base', 'themeVariables': {'pie1': '#0000FF', 'pie2': '#00FF00', 'pie3': '#FF0000'}}}%% keep in mind that pie1 is the highest since its sorted in descending order. Heres an example of a mindmap: mindmap  root((mindmap))    Origins      Long history      ::icon(fa fa-book)      Popularisation        British popular psychology author Tony Buzan    Research      On effectivness<br/>and features      On Automatic creation       Uses            Creative techniques            Strategic planning            Argument mapping    Tools      Pen and paper     Mermaid.  "}
 };
 
 
diff --git a/src/client/views/nodes/DiagramBox.tsx b/src/client/views/nodes/DiagramBox.tsx
index 1b8290652..1813e6097 100644
--- a/src/client/views/nodes/DiagramBox.tsx
+++ b/src/client/views/nodes/DiagramBox.tsx
@@ -113,11 +113,43 @@ export class DiagramBox extends ViewBoxAnnotatableComponent<FieldViewProps>() im
         // this.htmlCode = modelOutput;
         this.generateMermaidCode();
     };
-    @action generateMermaidCode = async () => {
+    printTitle(){
+        let docArray: Doc[] = DocListCast(this.Document.data);
+        let mermaidCodeDoc = docArray.filter(doc => doc.type == 'rich text')
+        mermaidCodeDoc=mermaidCodeDoc.filter(doc=>(doc.text as RichTextField).Text=='mermaidCodeTitle')
+        console.log(mermaidCodeDoc[0].title)
+        if(mermaidCodeDoc[0]){
+            console.log(mermaidCodeDoc[0].title)
+            if(typeof mermaidCodeDoc[0].title=='string'){
+                console.log(mermaidCodeDoc[0].title)
+                if(mermaidCodeDoc[0].title!=""){
+                    console.log("you have to see me")
+                }
+            }
+        }
+    }
+    @action async generateMermaidCode() {
         console.log('Generating Mermaid Code');
         this.loading = true;
-
-        let res = await gptAPICall(this.inputValue, GPTCallType.MERMAID);
+        let prompt=""
+        // let docArray: Doc[] = DocListCast(this.Document.data);
+        // let mermaidCodeDoc = docArray.filter(doc => doc.type == 'rich text')
+        // mermaidCodeDoc=mermaidCodeDoc.filter(doc=>(doc.text as RichTextField).Text=='mermaidCodeTitle')
+        // if(mermaidCodeDoc[0]){
+        //     console.log(mermaidCodeDoc[0].title)
+        //     if(typeof mermaidCodeDoc[0].title=='string'){
+        //         console.log(mermaidCodeDoc[0].title)
+        //         if(mermaidCodeDoc[0].title!=""){
+        //             prompt="Edit this code "+this.inputValue+": "+mermaidCodeDoc[0].title
+        //             console.log("you have to see me")
+        //         }
+        //     }
+        // }
+        // else{
+            prompt="Write this in mermaid code and only give me the mermaid code: "+this.inputValue
+            console.log("there is no text save")
+        //}
+        let res = await gptAPICall(prompt, GPTCallType.MERMAID);
         this.loading = false;
         if (res == 'Error connecting with API.') {
             // If GPT call failed