Update demo.py

demo.py

@@ -5,40 +5,53 @@ import io
 import cv2
 import numpy as np
 
-palmprint_count = 0
-
-def compare_palmprint(frame1, frame2):
-    global palmprint_count
+alpr_count = 0
+def plot_one_box(x, img, color=None, label=None, line_thickness=3):
+	# Plots one bounding box on image img
+    tl = line_thickness or round(0.002 * (img.shape[0] + img.shape[1]) / 2) + 1  # line/font thickness
+    color = color
+    c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3]))
+    cv2.rectangle(img, c1, c2, color, thickness=tl, lineType=cv2.LINE_AA)
+    if label:
+        tf = max(tl - 1, 1)  # font thickness
+        t_size = cv2.getTextSize(label, 0, fontScale=tl / 3, thickness=tf)[0]
+        c2 = c1[0] + t_size[0], c1[1] - t_size[1] - 3
+        cv2.rectangle(img, c1, c2, color, -1, cv2.LINE_AA)  # filled
+        cv2.putText(img, label, (c1[0], c1[1] - 2), 0, tl / 3, [0, 0, 0], thickness=tf, lineType=cv2.LINE_AA)
+    return img
     
-    palmprint_count = palmprint_count + 1
-
-    url = "http://127.0.0.1:8080/compare_palmprint"
-    files = {'file1': open(frame1, 'rb'), 'file2': open(frame2, 'rb')}
-
-    r = requests.post(url=url, files=files)
-
-    html = None
-
-    compare_result = r.json().get('compare_result')
-    compare_similarity = r.json().get('compare_similarity')
-
-    html = ("<table>"
-                "<tr>"
-                    "<th>Compare Result</th>"
-                    "<th>Value</th>"
-                "</tr>"
-                "<tr>"
-                    "<td>Result</td>"
-                    "<td>{compare_result}</td>"
-                "</tr>"
-                "<tr>"
-                    "<td>Similarity</td>"
-                    "<td>{compare_similarity}</td>"
-                "</tr>"
-                "</table>".format(compare_result=compare_result, compare_similarity=compare_similarity))
+def alpr(frame):
+    global alpr_count
+    
+    alpr_count = alpr_count + 1
+    print("alpr_count", alpr_count) 
+    url = "http://127.0.0.1:8080/alpr"
+    file = {'file': open(frame, 'rb')}
 
-    return [html]
+    r = requests.post(url=url, files=file)
+   
+    alpr_output = None
+    result = r.json().get('result')
+    plate_number = r.json().get('plate number')
+    box = r.json().get('coordinate')
+    try:
+        image = cv2.imread(frame, cv2.IMREAD_COLOR)
+        if image is None:
+            print('image is null')
+            sys.exit()
+        image = cv2.resize(image, (1024, 640))
+        for alpr in plate_number:
+            print(plate_number)
+            # image = plot_one_box(alpr.get('recog_boxes'), image, label=alpr.get('recog_results'), color=[0, 255, 0], line_thickness=1)
+        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+        
+        alpr_output = image.copy()
+        
+    except:
+        pass
 
+    return alpr_output
+    
 with gr.Blocks() as demo:
     gr.Markdown(
         """