Fix 0 labeling

pniedzwiedzinski · Jan 10, 2019 · e0b885f · e0b885f
1 parent 1cd4d2e
commit e0b885f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/main.py b/main.py
@@ -7,6 +7,7 @@
 import os
 import random
 import uuid
+import sys
 
 import boto3
 import botocore
@@ -140,6 +141,7 @@ def report_get():
         session_id = request.args.get("key")
 
     session.set(session_id, 'tmp/' + filename)
+    print("\033[92m-- User {} will label {} --\033[0m".format(session_id, filename))
 
     return add_headers(jsonify({"data": data, "key": session_id}))
 

diff --git a/src/components/AppleOrNot.jsx b/src/components/AppleOrNot.jsx
@@ -39,15 +39,12 @@ export default class AppleOrNot extends Component {
   label = l => {
     fetch(`${this.state.host}/label/?key=${this.state.key}&label=${l}`, {
       method: "POST"
-    }).then(this.setState({ img: null, loading: true }));
+    }).then(() => {
+      this.setState({ img: null, loading: true });
+      this.getPhoto();
+    });
   };
 
-  componentDidUpdate() {
-    if (this.state.loading) {
-      setTimeout(this.getPhoto(), 750);
-    }
-  }
-
   render() {
     return (
       <>