Update app.py

app.py

@@ -53,9 +53,9 @@ def idcard_recognition():
             j.update({"Status": "Ok"})
             response = jsonify(j)
         else:
-            response = jsonify({"Status": "Error"})
+            response = jsonify({"Status": "Processing Failed"})
     except:
-        response = jsonify({"Status": "Error"})
+        response = jsonify({"Status": "Image Error"})
 
     response.status_code = 200
     response.headers["Content-Type"] = "application/json; charset=utf-8"
@@ -63,22 +63,32 @@ def idcard_recognition():
 
 @app.route('/idcard_recognition_multi_page', methods=['POST'])
 def idcard_recognition_multi_page():
+    base64_image1 = ""
+    base64_image2 = ""
     try:
         file1 = request.files['file1']
-        file2 = request.files['file2']
-
         base64_image1 = base64.b64encode(file1.read()).decode('utf-8')
+    except:
+        response = jsonify({"Status": "Front Image Error"})
+
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    try:
+        file2 = request.files['file2']
         base64_image2 = base64.b64encode(file2.read()).decode('utf-8')
+    except:
+        base64_image2 = ""
+        
         ret = idcardRecognition(base64_image1.encode('utf-8'), base64_image2.encode('utf-8'))
 
-        if ret != None:
-            j = json.loads(ret)
-            j.update({"Status": "Ok"})
-            response = jsonify(j)
-        else:
-            response = jsonify({"Status": "Error"})
-    except:
-        response = jsonify({"Status": "Error"})
+    if ret != None:
+        j = json.loads(ret)
+        j.update({"Status": "Ok"})
+        response = jsonify(j)
+    else:
+        response = jsonify({"Status": "Processing Failed"})
 
     response.status_code = 200
     response.headers["Content-Type"] = "application/json; charset=utf-8"
@@ -97,9 +107,9 @@ def idcard_recognition_base64():
             j.update({"Status": "Ok"})
             response = jsonify(j)
         else:
-            response = jsonify({"Status": "Error"})
+            response = jsonify({"Status": "Processing Failed"})
     except:
-        response = jsonify({"Status": "Error"})
+        response = jsonify({"Status": "Image Error"})
 
     response.status_code = 200
     response.headers["Content-Type"] = "application/json; charset=utf-8"
@@ -107,21 +117,33 @@ def idcard_recognition_base64():
 
 @app.route('/idcard_recognition_base64_multi_page', methods=['POST'])
 def idcard_recognition_base64_multi_page():
+    base64_image1 = ""
+    base64_image2 = ""
+
     try:
         content = request.get_json()
         base64_image1 = content['base64_1']
-        base64_image2 = content['base64_2']
+    except:
+        response = jsonify({"Status": "Front Image Error"})
 
-        ret = idcardRecognition(base64_image1.encode('utf-8'), base64_image2.encode('utf-8'))
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
 
-        if ret != None:
-            j = json.loads(ret)
-            j.update({"Status": "Ok"})
-            response = jsonify(j)
-        else:
-            response = jsonify({"Status": "Error"})
+    try:
+        content = request.get_json()
+        base64_image2 = content['base64_2']
     except:
-        response = jsonify({"Status": "Error"})
+        base64_image2 = ""
+        
+    ret = idcardRecognition(base64_image1.encode('utf-8'), base64_image2.encode('utf-8'))
+
+    if ret != None:
+        j = json.loads(ret)
+        j.update({"Status": "Ok"})
+        response = jsonify(j)
+    else:
+        response = jsonify({"Status": "Processing Failed"})
 
     response.status_code = 200
     response.headers["Content-Type"] = "application/json; charset=utf-8"