Update app.py

app.py

@@ -1,6 +1,4 @@
 import sys
-sys.path.append('../')
-
 import os
 import gradio as gr
 import json
@@ -9,15 +7,6 @@ import requests
 import numpy as np
 from PIL import Image
 
-from engine.header import *
-
-file_path = os.path.abspath(__file__)
-dir_path = os.path.dirname(file_path)
-root_path = os.path.dirname(dir_path)
-
-device_id = get_deviceid().decode('utf-8')
-print_info('\t <Hardware ID> \t\t {}'.format(device_id))
-
 css = """
 .example-image img{
     display: flex; /* Use flexbox to align items */
@@ -69,63 +58,25 @@ def json_to_html_table(data, image_keys):
   html += "</table>"
   return html
 
-def activate_sdk():
-  online_key = os.environ.get("LICENSE_KEY")
-  offline_key_path = os.path.join(root_path, "license.txt")
-  
-  dict_path = os.path.join(root_path, "engine/bin")
-
-  ret = -1
-  if online_key is None:
-    print_warning("Online license key not found!")
-  else:
-    print_info(f"LICENSE_KEY: {online_key}")
-    activate_ret = set_activation(online_key.encode('utf-8')).decode('utf-8')
-    ret = json.loads(activate_ret).get("errorCode", None)
-
-  if ret == 0:
-    print_log("Successfully online activation SDK!")
-  else:
-    print_error(f"Failed to online activation SDK, Error code {ret}\n Trying offline activation SDK...");
-    if os.path.exists(offline_key_path) is False:
-      print_warning("Offline license key file not found!")
-      print_error(f"Falied to offline activation SDK, Error code {ret}")
-      return ret
+def idcard_recognition(frame1, frame2):
+  url = "recognito-iddocumentrecognition.p.rapidapi.com/api/read_idcard"
+  try:
+    files = None
+    if frame1 is not None and frame2 is not None:
+        files = {'image': open(frame1, 'rb'), 'image2': open(frame2, 'rb')}
+    elif frame1 is not None and frame2 is None:
+        files = {'image': open(frame1, 'rb')}
+    elif frame1 is None and frame2 is not None:
+        files = {'image': open(frame2, 'rb')}
     else:
-      file=open(offline_key_path,"r")
-      offline_key = file.read()
-      file.close()
-      activate_ret = set_activation(offline_key.encode('utf-8')).decode('utf-8')
-      ret = json.loads(activate_ret).get("errorCode", None)
-      if ret == 0:
-        print_log("Successfully offline activation SDK!")
-      else:
-        print_error(f"Falied to offline activation SDK, Error code {ret}")
-        return ret
-  
-  init_ret = init_sdk(dict_path.encode('utf-8')).decode('utf-8')
-  ret = json.loads(activate_ret).get("errorCode", None)
-  print_log(f"Init SDK: {ret}")
-  return ret
+        return ['', None]
 
-def idcard_recognition(frame1, frame2):
-  global g_activation_result
-  if g_activation_result != 0:
-    gr.Warning("SDK Activation Failed!")
-    return ['', None]
-
-  image1, image2 = '', ''
-  if frame1 is not None and frame2 is not None:
-    image1, image2 = frame1, frame2
-  elif frame1 is not None and frame2 is None:
-    image1 = frame1
-  elif frame1 is None and frame2 is not None:
-    image1 = frame2
-  elif frame1 is None and frame2 is None:
+    headers = {"X-RapidAPI-Key": os.environ.get("API_KEY")}
+    
+    r = requests.post(url=url, files=files, headers=headers)
+  except:
     raise gr.Error("Please select images files!")
 
-  ocrResult = ocr_id_card(image1.encode('utf-8'), image2.encode('utf-8'))
-  ocrResDict = json.loads(ocrResult)  
   images = None   
   rawValues = {}
   image_table_value = ""
@@ -143,14 +94,15 @@ def idcard_recognition(frame1, frame2):
     'documentNumber':'',
     }
 
-  for key, value in ocrResDict.items():
-    if key == 'image':
-      for image_key, image_value in value.items():
-        row_value = ("<tr>"
-                      "<td>{key}</td>"
-                      "<td><img src=""data:image/png;base64,{base64_image} width = '200'  height= '100' /></td>"
-                    "</tr>".format(key=image_key, base64_image=image_value))
-        image_table_value = image_table_value + row_value
+  if 'data' in r.json():
+      for key, value in r.json()['data'].items():
+        if key == 'image':
+          for image_key, image_value in value.items():
+            row_value = ("<tr>"
+                          "<td>{key}</td>"
+                          "<td><img src=""data:image/png;base64,{base64_image} width = '200'  height= '100' /></td>"
+                        "</tr>".format(key=image_key, base64_image=image_value))
+            image_table_value = image_table_value + row_value
 
   images = ("<table>"
             "<tr>"
@@ -160,16 +112,17 @@ def idcard_recognition(frame1, frame2):
             "{image_table_value}"
             "</table>".format(image_table_value=image_table_value))
 
-  value = ocrResDict.copy()
-  if 'image' in value:
-    del value['image']
-    rawValues = value
-  else:
-    rawValues = value
+  for key, value in r.json().items():
+    if key == 'data':
+      if 'image' in value:
+        del value['image']
+        rawValues = value
+    else:
+      rawValues = value
 
   
   for result_key in result_table_dict.keys():
-    result_table_dict[result_key] =  find_key_in_dict(ocrResDict, result_key)
+    result_table_dict[result_key] =  find_key_in_dict(r.json(), result_key)
   
   result = json_to_html_table(result_table_dict, {'portrait'})
   json_result = json.dumps(rawValues, indent=6)
@@ -245,5 +198,4 @@ def launch_demo(activate_result):
   demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False)
 
 if __name__ == '__main__':
-  g_activation_result = activate_sdk()
   launch_demo(g_activation_result)