edreesi commited on
Commit
13573d3
·
verified ·
1 Parent(s): a53a581

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -9
app.py CHANGED
@@ -78,6 +78,7 @@ def json_to_html_table(data, image_keys):
78
  return html
79
 
80
  def check_liveness(frame):
 
81
  if frame is None:
82
  liveness_result = f"""<div class="markdown-fail-container"><p style="text-align: center; font-size: 20px; color: red;">Liveness Check Failed</p></div>"""
83
  return [liveness_result, {"status": "error", "result": "select image file!"}]
@@ -86,23 +87,30 @@ def check_liveness(frame):
86
  Image.open(frame).save(img_bytes, format="JPEG")
87
  img_bytes.seek(0)
88
 
89
- url = "https://recognito-iddocumentlivenessdetection.p.rapidapi.com/process_image"
90
 
91
  try:
92
- files = {'image': img_bytes}
93
- headers = {"X-RapidAPI-Key": os.environ.get("API_KEY")}
 
94
 
 
 
 
 
95
  result = requests.post(url=url, files=files, headers=headers)
96
  except:
97
  liveness_result = f"""<div class="markdown-fail-container"><p style="text-align: center; font-size: 20px; color: red;">Liveness Check Failed</p></div>"""
98
  return [liveness_result, {"status": "error", "result": "failed to open file!"}]
99
-
100
  if result.ok:
101
  json_result = result.json()
102
  if json_result.get("resultCode") == "Error":
103
  liveness_result = f"""<div class="markdown-fail-container"><p style="text-align: center; font-size: 20px; color: red;">Liveness Check Failed</p></div>"""
104
  return [liveness_result, {"status": "error", "result": "server error!"}]
105
-
 
 
106
  process_results = json_result.get("result")
107
  status = process_results.get("status")
108
  if status == "Ok":
@@ -128,12 +136,9 @@ def check_liveness(frame):
128
  return [liveness_result, {"status": "error", "result": f"{result.text}"}]
129
 
130
  def idcard_recognition(frame1):
131
- # url = "https://recognito-iddocumentrecognition.p.rapidapi.com/api/read_idcard"
132
  url = "https://api.cortex.cerebrium.ai/v4/p-4f1d877e/my-first-project/process-image/"
133
  # url = "https://edreesi-ocr-api.hf.space/process-image/"
134
 
135
-
136
-
137
  files = None
138
  if frame1 is not None:
139
  # Open the image using Pillow
@@ -289,7 +294,7 @@ def launch_demo():
289
 
290
  id_recognition_button.click(idcard_recognition, inputs=id_image_input1, outputs=[id_result_output, json_result_output, image_result_output])
291
 
292
- with gr.Tab("ID Document Liveness Detection"):
293
  with gr.Row():
294
  with gr.Column(scale=1):
295
  id_image_input = gr.Image(label="Image", type='filepath', elem_classes="example-image")
 
78
  return html
79
 
80
  def check_liveness(frame):
81
+
82
  if frame is None:
83
  liveness_result = f"""<div class="markdown-fail-container"><p style="text-align: center; font-size: 20px; color: red;">Liveness Check Failed</p></div>"""
84
  return [liveness_result, {"status": "error", "result": "select image file!"}]
 
87
  Image.open(frame).save(img_bytes, format="JPEG")
88
  img_bytes.seek(0)
89
 
90
+ url = "https://api.cortex.cerebrium.ai/v4/p-4f1d877e/my-first-project/check-liveness/"
91
 
92
  try:
93
+ files = [
94
+ ('file', ('image.jpg', img_bytes, 'image/jpeg'))
95
+ ]
96
 
97
+ headers = {
98
+ 'Authorization': 'Bearer eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJwcm9qZWN0SWQiOiJwLTRmMWQ4NzdlIiwiaWF0IjoxNzM5MjM2NjA5LCJleHAiOjIwNTQ4MTI2MDl9.0LH0iOnqHZfKTH4GF5iTZ4qNj5vylCryo8rBnErljsq2qD2cpVTetCqhKtnbstTUEjuv6MAJw9jt58z-QNJfYLK9sJcnBhawTR3iM2Ap_bFyjlzg2LbgkwRPjUVJkkcuCRhBKyebXwvqQBvWyOtMq6UekauumbmYBRbA2-T4u343YD4tO2xIfsTsTXznALp1SechjRuys-3xo3ZQbUs05_p38fOFucKI-abc91Eq6sIOkLFjYEM68yuV0UBWl-OSpPu66e0SClroAVlKFDMPS9MY0Jr7X1pBYX4jew6vozj9D8Y-HS-KkdPFqJ7HrZOfQd0wGUgYJHyC58yReWXaRQ',
99
+ # 'Content-Type': 'application/json'
100
+ }
101
  result = requests.post(url=url, files=files, headers=headers)
102
  except:
103
  liveness_result = f"""<div class="markdown-fail-container"><p style="text-align: center; font-size: 20px; color: red;">Liveness Check Failed</p></div>"""
104
  return [liveness_result, {"status": "error", "result": "failed to open file!"}]
105
+ print("the result is", result)
106
  if result.ok:
107
  json_result = result.json()
108
  if json_result.get("resultCode") == "Error":
109
  liveness_result = f"""<div class="markdown-fail-container"><p style="text-align: center; font-size: 20px; color: red;">Liveness Check Failed</p></div>"""
110
  return [liveness_result, {"status": "error", "result": "server error!"}]
111
+ # if 'data' in r.json():
112
+ # data = r.json()['data']
113
+
114
  process_results = json_result.get("result")
115
  status = process_results.get("status")
116
  if status == "Ok":
 
136
  return [liveness_result, {"status": "error", "result": f"{result.text}"}]
137
 
138
  def idcard_recognition(frame1):
 
139
  url = "https://api.cortex.cerebrium.ai/v4/p-4f1d877e/my-first-project/process-image/"
140
  # url = "https://edreesi-ocr-api.hf.space/process-image/"
141
 
 
 
142
  files = None
143
  if frame1 is not None:
144
  # Open the image using Pillow
 
294
 
295
  id_recognition_button.click(idcard_recognition, inputs=id_image_input1, outputs=[id_result_output, json_result_output, image_result_output])
296
 
297
+ with gr.Tab("Id Card Liveness Detection"):
298
  with gr.Row():
299
  with gr.Column(scale=1):
300
  id_image_input = gr.Image(label="Image", type='filepath', elem_classes="example-image")