Update

app.py

@@ -52,8 +52,12 @@ def ocr_idcard():
 
 
   ocrResult = TTVOcrProcess(file_path1.encode('utf-8'), file_path2.encode('utf-8'))
-  ocrResDict = json.loads(ocrResult)
   status = "ok"
+  if not ocrResult:
+    ocrResDict = {}
+    status = "error"
+  else:
+    ocrResDict = json.loads(ocrResult)  
 
   if_check = ttv_if_checker(file_path1.encode('utf-8'))
   response = jsonify({"status": status, "data": ocrResDict, "authenticity": if_check})
@@ -79,8 +83,12 @@ def ocr_idcard_base64():
   file_path = os.path.abspath(save_path)
 
   ocrResult = TTVOcrProcess(file_path.encode('utf-8'))
-  ocrResDict = json.loads(ocrResult)
   status = "ok"
+  if not ocrResult:
+    ocrResDict = {}
+    status = "error"
+  else:
+    ocrResDict = json.loads(ocrResult)  
 
   if_check = ttv_if_checker(file_path.encode('utf-8'))
   response = jsonify({"status": status, "data": ocrResDict, "authenticity": if_check})
@@ -105,8 +113,12 @@ def ocr_credit():
   file_path = os.path.abspath(save_path)
 
   ocrResult = TTVOcrCreditCard(file_path.encode('utf-8'))
-  ocrResDict = json.loads(ocrResult)
   status = "ok"
+  if not ocrResult:
+    ocrResDict = {}
+    status = "error"
+  else:
+    ocrResDict = json.loads(ocrResult)  
 
   response = jsonify({"status": status, "data": ocrResDict})
 
@@ -130,8 +142,12 @@ def ocr_credit_base64():
   file_path = os.path.abspath(save_path)
 
   ocrResult = TTVOcrCreditCard(file_path.encode('utf-8'))
-  ocrResDict = json.loads(ocrResult)
   status = "ok"
+  if not ocrResult:
+    ocrResDict = {}
+    status = "error"
+  else:
+    ocrResDict = json.loads(ocrResult)
 
   response = jsonify({"status": status, "data": ocrResDict})
 
@@ -162,7 +178,6 @@ def ocr_barcode():
     ocrResDict = json.loads(ocrResult)  
 
   response = jsonify({"status": status, "data": ocrResDict})
-  print("@@@@@", response)
 
   os.remove(file_path)
 
@@ -184,8 +199,12 @@ def ocr_barcode_base64():
   print('file_path: ', file_path)
 
   ocrResult = TTVOcrBarCode(file_path.encode('utf-8'))
-  ocrResDict = json.loads(ocrResult)
   status = "ok"
+  if not ocrResult:
+    ocrResDict = {}
+    status = "error"
+  else:
+    ocrResDict = json.loads(ocrResult)
 
   response = jsonify({"status": status, "data": ocrResDict})
 

demo.py

@@ -61,8 +61,6 @@ def barcode_recognition(frame):
     files = {'image': open(frame, 'rb')}
     r = requests.post(url=url, files=files)
 
-    print('#########', r.json())
-
     images = None   
     resultValues = {}
     table_value = ""
@@ -95,7 +93,6 @@ def barcode_recognition(frame):
                     "</table>".format(table_value=table_value))
     
     json_result = json.dumps(resultValues, indent=4)
-    print("#############", json_result, images)
     return [json_result, images]
 
 def credit_recognition(frame):
@@ -162,6 +159,14 @@ with gr.Blocks() as demo:
                 image_result_output = gr.HTML()
 
         id_recognition_button.click(idcard_recognition, inputs=[id_image_input1, id_image_input2], outputs=[id_result_output, image_result_output])
+        gr.Markdown("## Image Examples")
+        gr.Examples(
+            examples=[['examples/11.jpg', 'examples/12.jpg'], ['examples/21.jpg', 'examples/22.jpg'], ['examples/3.jpg', None]],
+            inputs=[id_image_input1, id_image_input2],
+            outputs=[id_result_output, image_result_output],
+            fn=idcard_recognition,
+            cache_examples=True,
+        )
     with gr.TabItem("Barcode Recognition"):
         with gr.Row():
             with gr.Column(scale=3):