Update app.py

app.py

@@ -153,10 +153,11 @@ def fig2img(fig):
 
 
 # -------------------- OCR Stub --------------------
-
-def read_plate(crop):
-    # Placeholder OCR logic
-    return "KA01AB1234"
+def read_plate(plate_img):
+    results = reader.readtext(np.array(plate_img))
+    if results:
+        return results[0][1]
+    return "UNKNOWN"
 
 
 # -------------------- Visualization --------------------
@@ -178,10 +179,10 @@ def visualize_prediction(img, output_dict, threshold=0.5, id2label=None):
 
     for score, (xmin, ymin, xmax, ymax), label in zip(scores, boxes, labels):
         if "plate" in label.lower():
-            crop = img.crop((int(xmin), int(ymin), int(xmax), int(ymax)))
+            plate_img = img.crop((int(xmin), int(ymin), int(xmax), int(ymax)))
 
-            plate_text = read_plate(crop)
-            vehicle_type = classify_plate_color(crop)
+            plate_text = read_plate(plate_img)
+            vehicle_type = classify_plate_color(plate_img)
             toll, discount_msg = compute_discount(vehicle_type)
 
             cursor.execute(