added loading animation

1 files changed, 21 insertions, 13 deletions

diff --git a/src/client/views/nodes/ChatBox/vectorstore/Vectorstore.ts b/src/client/views/nodes/ChatBox/vectorstore/Vectorstore.ts
index 388574bff..408274703 100644
--- a/src/client/views/nodes/ChatBox/vectorstore/Vectorstore.ts
+++ b/src/client/views/nodes/ChatBox/vectorstore/Vectorstore.ts
@@ -77,27 +77,35 @@ export class Vectorstore {
             console.log('Local File Path:', local_file_path);
 
             if (local_file_path) {
+                console.log('Creating AI Document...');
                 // Start the document creation process
-                const response = await Networking.PostToServer('/createDocument', { file_path: local_file_path });
-                const jobId = response.job_id;
+                const { jobId } = await Networking.PostToServer('/createDocument', { file_path: local_file_path });
 
                 // Poll the server for progress updates
+                let inProgress: boolean = true;
                 let result: any = null;
+                while (inProgress) {
+                    await new Promise(resolve => setTimeout(resolve, 2000)); // Polling interval
+
+                    const resultResponse = await Networking.FetchFromServer(`/getResult/${jobId}`);
+                    const resultResponseJson = JSON.parse(resultResponse);
+                    //console.log('Result Response:', resultResponseJson);
+                    if (resultResponseJson.status === 'completed') {
+                        console.log('Result here:', resultResponseJson);
+                        result = resultResponseJson;
+                        break;
+                    }
 
-                while (!result) {
-                    await new Promise(resolve => setTimeout(resolve, 5000)); // Polling interval
+                    const progressResponse = await Networking.FetchFromServer(`/getProgress/${jobId}`);
+                    const progressResponseJson = JSON.parse(progressResponse);
+                    //console.log('Progress Response:', progressResponseJson);
 
-                    const progressResponse = JSON.parse(await Networking.FetchFromServer(`/getProgress/${jobId}`));
-                    if (progressResponse) {
-                        const progress = progressResponse.progress;
-                        const step = progressResponse.step;
+                    if (progressResponseJson) {
+                        console.log('Progress:', progressResponseJson);
+                        const progress = progressResponseJson.progress;
+                        const step = progressResponseJson.step;
                         progressCallback(progress, step);
                     }
-
-                    const resultResponse = JSON.parse(await Networking.FetchFromServer(`/getResult/${jobId}`));
-                    if (resultResponse.status === 200) {
-                        result = resultResponse.data;
-                    }
                 }
 
                 // Process the final document result