From 4997c3de20a381eac30224a7a550afa66174f07d Mon Sep 17 00:00:00 2001
From: Joanne <zehan_ding@brown.edu>
Date: Mon, 12 May 2025 20:53:12 -0400
Subject: added tutorial tool, still need to integrate with metadatatool

---
 .../nodes/chatbot/chatboxcomponents/ChatBox.tsx    | 45 ++++++++--------------
 1 file changed, 17 insertions(+), 28 deletions(-)

(limited to 'src/client/views/nodes/chatbot/chatboxcomponents')

diff --git a/src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx b/src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx
index 6e9307d37..ad2f3e892 100644
--- a/src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx
+++ b/src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx
@@ -106,7 +106,7 @@ export class ChatBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
             this.vectorstore_id = StrCast(this.dataDoc.vectorstore_id);
         }
         this.vectorstore = new Vectorstore(this.vectorstore_id, this.retrieveDocIds);
-        this.agent = new Agent(this.vectorstore, this.retrieveSummaries, this.retrieveFormattedHistory, this.retrieveCSVData, this.addLinkedUrlDoc, this.createImageInDash, this.createDocInDash, this.createCSVInDash);
+        this.agent = new Agent(this.vectorstore, this.retrieveSummaries, this.retrieveFormattedHistory, this.retrieveCSVData, this.addLinkedUrlDoc, this.createImageInDash, this.createDocInDash, this.createCSVInDash, this.Document);
         this.messagesRef = React.createRef<HTMLDivElement>();
 
         // Reaction to update dataDoc when chat history changes
@@ -309,7 +309,6 @@ export class ChatBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
                             };
                         }
                     });
-                    this.scrollToBottom();
                 };
 
                 const onAnswerUpdate = (answerUpdate: string) => {
@@ -317,41 +316,29 @@ export class ChatBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
                         if (this._current_message) {
                             this._current_message = {
                                 ...this._current_message,
-                                content: [{ text: answerUpdate, type: TEXT_TYPE.NORMAL, index: 0, citation_ids: [] }],
+                                content: [{ index: 0, type: TEXT_TYPE.NORMAL, text: answerUpdate, citation_ids: null }],
                             };
                         }
                     });
                 };
 
-                // Send the user's question to the assistant and get the final message
-                const finalMessage = await this.agent.askAgent(trimmedText, onProcessingUpdate, onAnswerUpdate);
+                // Get the response from the agent
+                const response = await this.agent.askAgent(trimmedText, onProcessingUpdate, onAnswerUpdate);
 
-                // Update the history with the final assistant message
+                // Push the final message to history
                 runInAction(() => {
-                    if (this._current_message) {
-                        this._history.push({ ...finalMessage });
-                        this._current_message = undefined;
-                        this.dataDoc.data = JSON.stringify(this._history);
-                    }
+                    this._history.push(response);
+                    this._isLoading = false;
+                    this._current_message = undefined;
                 });
-            } catch (err) {
-                console.error('Error:', err);
-                // Handle error in processing
-                runInAction(() =>
-                    this._history.push({
-                        role: ASSISTANT_ROLE.ASSISTANT,
-                        content: [{ index: 0, type: TEXT_TYPE.ERROR, text: `Sorry, I encountered an error while processing your request: ${err} `, citation_ids: null }],
-                        processing_info: [],
-                    })
-                );
-            } finally {
+            } catch (error) {
+                console.error('Error in askGPT:', error);
                 runInAction(() => {
                     this._isLoading = false;
+                    this._current_message = undefined;
                 });
-                this.scrollToBottom();
             }
         }
-        this.scrollToBottom();
     };
 
     /**
@@ -408,7 +395,7 @@ export class ChatBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
             if (doc) {
                 LinkManager.Instance.addLink(Docs.Create.LinkDocument(this.Document, doc));
                 this._props.addDocument?.(doc);
-                DocumentManager.Instance.showDocument(doc, { willZoomCentered: true }, () => {}).then(() => this.addCSVForAnalysis(doc, id));
+                DocumentManager.Instance.showDocument(doc, { willZoomCentered: true }, () => this.addCSVForAnalysis(doc, id));
             }
         });
 
@@ -446,7 +433,7 @@ export class ChatBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
         const ndoc = (() => {
             switch (doc.doc_type) {
                 default:           
-                case supportedDocTypes.text:       return Docs.Create.TextDocument(data as string, options);
+                case supportedDocTypes.text:       return Docs.Create.TextDocument(doc.text as string, options);
                 case supportedDocTypes.comparison: return this.createComparison(JSON.parse(data as string) as parsedDoc[], options);
                 case supportedDocTypes.flashcard:  return this.createFlashcard(JSON.parse(data as string) as  parsedDoc[], options);
                 case supportedDocTypes.deck:       return this.createDeck(JSON.parse(data as string) as  parsedDoc[], options);
@@ -825,7 +812,9 @@ export class ChatBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
                         {
                             index: 0,
                             type: TEXT_TYPE.NORMAL,
-                            text: `Hey, ${this.userName()}! Welcome to Your Friendly Assistant. Link a document or ask questions to get started.`,
+                            text: this.dataDoc.is_dash_doc_assistant
+                                ? 'Welcome to your help assistant for Dash. Ask any Dash-related questions to get started.'
+                                : `Hey, ${this.userName()}! Welcome to Your Friendly Assistant. Link a document or ask questions to get started.`,
                             citation_ids: null,
                         },
                     ],
@@ -987,7 +976,7 @@ export class ChatBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
                     </div>
                 )}
                 <div className="chat-header">
-                    <h2>{this.userName()}&apos;s AI Assistant</h2>
+                    <h2>{this.dataDoc.is_dash_doc_assistant ? 'Dash Help Assistant' : `${this.userName()}'s AI Assistant`}</h2>
                 </div>
                 <div className="chat-messages" ref={this.messagesRef}>
                     {this._history.map((message, index) => (
-- 
cgit v1.2.3-70-g09d2