Hugging Face's logo

Spaces:
kby-ai
/
IDCardRecognition
Running

App Files Community
justin2341 commited on
Commit
3c12d87
·
verified ·
1 Parent(s): a75c8d4

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +107 -23
demo.py CHANGED
@@ -32,41 +32,125 @@ def idcard_recognition(frame1, frame2):
32
  mrz_value = mrz[key]
33
  del mrz[key]
34
 
35
- row_value = ("<tr>"
36
- "<td>{key}</td>"
37
- "<td>{value}</td>"
38
- "<td>{mrz_value}</td>"
39
- "</tr>".format(key=key, value=value, mrz_value=mrz_value))
40
- table_value = table_value + row_value
41
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
42
 
43
- if mrz is not None:
44
  for key, value in mrz.items():
45
  if key == 'MRZ':
46
  value = value.replace('<', '&lt;')
47
  value = value.replace(',', '<p>')
48
-
49
  row_value = ("<tr>"
50
  "<td>{key}</td>"
51
  "<td>{value}</td>"
52
  "<td>{mrz_value}</td>"
 
53
  "</tr>".format(key=key, value='', mrz_value=value))
54
  table_value = table_value + row_value
55
-
56
-
57
- html = ("<table>"
58
- "<tr>"
59
- "<th style=""width:20%"">Field</th>"
60
- "<th style=""width:40%"">Value</th>"
61
- "<th style=""width:40%"">MRZ</th>"
62
- "</tr>"
63
- "<tr>"
64
- "<td>Status</td>"
65
- "<td>{status}</td>"
66
- "<td></td>"
67
- "</tr>"
68
- "{table_value}"
69
- "</table>".format(status=status, table_value=table_value))
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
70
 
71
  table_value = ""
72
  for key, value in r.json().items():
 
32
  mrz_value = mrz[key]
33
  del mrz[key]
34
 
35
+ barcode_value = ''
36
+ if barcode is not None and barcode.get(key) is not None:
37
+ barcode_value = barcode[key]
38
+ del barcode[key]
 
 
39
 
40
+ if mrz is not None and barcode is not None:
41
+ row_value = ("<tr>"
42
+ "<td>{key}</td>"
43
+ "<td>{value}</td>"
44
+ "<td>{mrz_value}</td>"
45
+ "<td>{barcode_value}</td>"
46
+ "</tr>".format(key=key, value=value, mrz_value=mrz_value, barcode_value=barcode_value))
47
+ elif mrz is not None:
48
+ row_value = ("<tr>"
49
+ "<td>{key}</td>"
50
+ "<td>{value}</td>"
51
+ "<td>{mrz_value}</td>"
52
+ "</tr>".format(key=key, value=value, mrz_value=mrz_value))
53
+ elif barcode is not None:
54
+ row_value = ("<tr>"
55
+ "<td>{key}</td>"
56
+ "<td>{value}</td>"
57
+ "<td>{barcode_value}</td>"
58
+ "</tr>".format(key=key, value=value, barcode_value=barcode_value))
59
+ else:
60
+ row_value = ("<tr>"
61
+ "<td>{key}</td>"
62
+ "<td>{value}</td>"
63
+ "</tr>".format(key=key, value=value))
64
+
65
+ table_value = table_value + row_value
66
 
67
+ if mrz is not None and barcode is not None:
68
  for key, value in mrz.items():
69
  if key == 'MRZ':
70
  value = value.replace('<', '&lt;')
71
  value = value.replace(',', '<p>')
72
+
73
  row_value = ("<tr>"
74
  "<td>{key}</td>"
75
  "<td>{value}</td>"
76
  "<td>{mrz_value}</td>"
77
+ "<td></td>"
78
  "</tr>".format(key=key, value='', mrz_value=value))
79
  table_value = table_value + row_value
80
+
81
+ for key, value in barcode.items():
82
+ if key == 'Barcode':
83
+ value = value.replace('<', '&lt;')
84
+ value = value.replace(',', '<p>')
85
+
86
+ row_value = ("<tr>"
87
+ "<td>{key}</td>"
88
+ "<td>{value}</td>"
89
+ "<td></td>"
90
+ "<td>{barcode_value}</td>"
91
+ "</tr>".format(key=key, value='', barcode_value=value))
92
+ table_value = table_value + row_value
93
+ elif mrz is not None:
94
+ for key, value in mrz.items():
95
+ if key == 'MRZ':
96
+ value = value.replace('<', '&lt;')
97
+ value = value.replace(',', '<p>')
98
+
99
+ row_value = ("<tr>"
100
+ "<td>{key}</td>"
101
+ "<td>{value}</td>"
102
+ "<td>{mrz_value}</td>"
103
+ "</tr>".format(key=key, value='', mrz_value=value))
104
+ table_value = table_value + row_value
105
+ elif barcode is not None:
106
+ for key, value in barcode.items():
107
+ if key == 'Barcode':
108
+ value = value.replace('<', '&lt;')
109
+ value = value.replace(',', '<p>')
110
+
111
+ row_value = ("<tr>"
112
+ "<td>{key}</td>"
113
+ "<td>{value}</td>"
114
+ "<td>{barcode_value}</td>"
115
+ "</tr>".format(key=key, value='', barcode_value=value))
116
+ table_value = table_value + row_value
117
+
118
+ if mrz is not None and barcode is not None:
119
+ html = ("<table>"
120
+ "<tr>"
121
+ "<th>Field</th>"
122
+ "<th>OCR</th>"
123
+ "<th>MRZ</th>"
124
+ "<th>Barcode</th>"
125
+ "</tr>"
126
+ "{table_value}"
127
+ "</table>".format(table_value=table_value))
128
+ elif mrz is not None:
129
+ html = ("<table>"
130
+ "<tr>"
131
+ "<th style=""width:20%"">Field</th>"
132
+ "<th style=""width:40%"">OCR</th>"
133
+ "<th style=""width:40%"">MRZ</th>"
134
+ "</tr>"
135
+ "{table_value}"
136
+ "</table>".format(table_value=table_value))
137
+ elif barcode is not None:
138
+ html = ("<table>"
139
+ "<tr>"
140
+ "<th style=""width:20%"">Field</th>"
141
+ "<th style=""width:40%"">OCR</th>"
142
+ "<th style=""width:40%"">Barcode</th>"
143
+ "</tr>"
144
+ "{table_value}"
145
+ "</table>".format(table_value=table_value))
146
+ else:
147
+ html = ("<table>"
148
+ "<tr>"
149
+ "<th>Field</th>"
150
+ "<th>OCR</th>"
151
+ "</tr>"
152
+ "{table_value}"
153
+ "</table>".format(table_value=table_value))
154
 
155
  table_value = ""
156
  for key, value in r.json().items():