From 8ee4d8ff68431dcb7022e5d5730a8256d8d01011 Mon Sep 17 00:00:00 2001
From: Sean Perry <sean.hyatt.perry@gmail.com>
Date: Thu, 21 Jul 2022 14:17:03 -0700
Subject: [PATCH] Added a count for how many files a user has left for time
 mangement

---
 .../components/annotate/annotationWindow.js   |  3 ++
 audino/frontend/src/pages/annotate.js         | 34 ++++++++++++++++---
 2 files changed, 32 insertions(+), 5 deletions(-)

diff --git a/audino/frontend/src/components/annotate/annotationWindow.js b/audino/frontend/src/components/annotate/annotationWindow.js
index f7952d70..1c2709e6 100644
--- a/audino/frontend/src/components/annotate/annotationWindow.js
+++ b/audino/frontend/src/components/annotate/annotationWindow.js
@@ -19,11 +19,13 @@ const AnnotationWindow = props => {
     isRendering,
     original_filename,
     navButtonsEnabled,
+    count
   } = state;
 
   return (
     <div>
       <div className="h-100 mt-5 text-center">
+      
         <AlertSection
           messages={[
             { message: errorUnsavedMessage, type: 'danger' },
@@ -44,6 +46,7 @@ const AnnotationWindow = props => {
                 : <Button id="add-edit-toggle-button" className="editRegion" variant="secondary" onClick={() => {setAddRegionMode(!state.addRegionMode)}}>Add Regions: Off</Button>
               }
             </div>
+            {`                 labeled ` + count.completed + ` out of ` + count.all}
           </div>
         }
 
diff --git a/audino/frontend/src/pages/annotate.js b/audino/frontend/src/pages/annotate.js
index 0fa78660..a9996578 100644
--- a/audino/frontend/src/pages/annotate.js
+++ b/audino/frontend/src/pages/annotate.js
@@ -21,6 +21,12 @@ class Annotate extends React.Component {
 
     this.initalState = {
       colorChange: 0,
+      count: {
+        pending: 0,
+        completed: 0,
+        all: 0,
+        marked_review: 0
+      },
       next_data_url: '',
       next_data_id: -1,
       isPlaying: false,
@@ -30,6 +36,7 @@ class Annotate extends React.Component {
       labelsUrl: `/api/projects/${projectId}/labels`,
       dataUrl: `/api/projects/${projectId}/data/${dataId}`,
       segmentationUrl: `/api/projects/${projectId}/data/${dataId}/segmentations_batch`,
+      apiUrl: `/api/current_user/projects/${projectId}/data`,
       isDataLoading: false,
       wavesurfer: null,
       zoom: 100,
@@ -118,11 +125,27 @@ class Annotate extends React.Component {
           isDataLoading: false
         });
       });
+
+      this.getData()
   }
 
-  /*componentWillUnmount() {
-    handleAllSegmentSave(this)
-  }*/
+  getData() {
+    let { apiUrl} = this.state;
+    apiUrl = `${apiUrl}?page=${1}&active=pending`;
+    axios({
+      method: 'get',
+      url: apiUrl
+    })
+      .then(response => {
+        const { count } = response.data;
+        this.setState({
+          count
+        });
+      })
+      .catch(error => {
+        console.error(error);
+      });
+  }
 
   handleAlertDismiss(e) {
     e.preventDefault(e);
@@ -207,13 +230,14 @@ class Annotate extends React.Component {
   }
 
   nextPage(nextDataId) {
-    const { wavesurfer, projectId } = this.state;
+    const { wavesurfer, projectId, count } = this.state;
     const newState = this.initalState;
     newState.labelsUrl = `/api/projects/${projectId}/labels`;
     newState.dataUrl = `/api/projects/${projectId}/data/${nextDataId}`;
     newState.segmentationUrl = `/api/projects/${projectId}/data/${nextDataId}/segmentations_batch`;
     newState.dataId = nextDataId;
     newState.wavesurfer = null;
+    newState.count = count
 
     wavesurfer.destroy();
     this.setState(newState, () => {
@@ -253,7 +277,7 @@ class Annotate extends React.Component {
     }
     
     //console.log(maxHeight)
-    const { wavesurferMethods, disappear, showActiveForm, sideMenuEnabled, sideMenuOn } =
+    const { wavesurferMethods, disappear, showActiveForm, sideMenuEnabled, sideMenuOn, count } =
       this.state;
 
     if (wavesurferMethods) {