Update app.py

app.py

@@ -2,9 +2,8 @@ import gradio as gr
 import cv2
 import numpy as np
 import easyocr
-from datetime import datetime
 
-reader = easyocr.Reader(['en'])
+reader = easyocr.Reader(['en'], gpu=False)
 
 feedback_data = []
 
@@ -14,22 +13,20 @@ feedback_data = []
 
 def classify_vehicle_by_plate_color(plate_img):
 
-    img = np.array(plate_img)
-    hsv = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)
+    hsv = cv2.cvtColor(plate_img, cv2.COLOR_BGR2HSV)
 
     green = np.sum(cv2.inRange(hsv, (35, 40, 40), (85, 255, 255)))
     yellow = np.sum(cv2.inRange(hsv, (15, 50, 50), (35, 255, 255)))
     white = np.sum(cv2.inRange(hsv, (0, 0, 200), (180, 30, 255)))
 
     if green > yellow and green > white:
-        return "EV"
+        return "EV", True
     elif yellow > green and yellow > white:
-        return "Commercial"
+        return "Commercial", False
     else:
-        return "Personal"
+        return "Personal", False
 
 
-    
 #########################################################
 # 2️⃣ Detection + OCR + EV Benefits
 #########################################################
@@ -49,15 +46,22 @@ def detect_vehicles(image):
     count = 0
 
     for cnt in contours:
+
         x, y, w, h = cv2.boundingRect(cnt)
 
-        # Indian plate shape filter
-        if 2 < w/h < 6 and w > 100:
+        # plate shape filtering
+        if h == 0:
+            continue
+
+        ratio = w / h
+
+        if 2 < ratio < 6 and w > 120 and h > 30:
 
             plate_img = img[y:y+h, x:x+w]
 
-            # OCR for plate number
+            # OCR
             results = reader.readtext(plate_img)
+
             plate_number = "Unknown"
 
             if len(results) > 0:
@@ -71,16 +75,20 @@ def detect_vehicles(image):
             else:
                 benefit = "No EV Benefits"
 
-            # Draw box + label
+            # draw detection
             cv2.rectangle(img, (x,y), (x+w,y+h), (0,255,0), 2)
 
             label = f"{plate_number} | {vehicle_type}"
-            cv2.putText(img, label,
-                        (x, y-10),
-                        cv2.FONT_HERSHEY_SIMPLEX,
-                        0.6,
-                        (0,255,0),
-                        2)
+
+            cv2.putText(
+                img,
+                label,
+                (x, y-10),
+                cv2.FONT_HERSHEY_SIMPLEX,
+                0.6,
+                (0,255,0),
+                2
+            )
 
             detected_summary.append(
                 f"Plate: {plate_number} | Type: {vehicle_type} | {benefit}"
@@ -89,7 +97,7 @@ def detect_vehicles(image):
             count += 1
 
     if count == 0:
-        summary = "No vehicles detected."
+        summary = "No number plate detected."
     else:
         summary = "\n".join(detected_summary)
 
@@ -103,23 +111,21 @@ def detect_vehicles(image):
 #########################################################
 
 def submit_feedback(is_correct):
-    feedback_data.append(is_correct)
-    return generate_summary()
 
+    feedback_data.append(is_correct)
 
-def generate_summary():
     total = len(feedback_data)
-    if total == 0:
-        return "No evaluations yet."
 
     correct = sum(feedback_data)
-    accuracy = (correct/total) * 100
+
+    accuracy = (correct / total) * 100
 
     return f"""
-Evaluation Summary:
-Total Samples: {total}
-Correct: {correct}
-Accuracy: {accuracy:.2f}%
+Evaluation Summary
+-------------------
+Total Samples : {total}
+Correct       : {correct}
+Accuracy      : {accuracy:.2f} %
 """
 
 
@@ -140,6 +146,7 @@ with gr.Blocks() as demo:
             detect_btn = gr.Button("Detect", size="sm")
 
             output_image = gr.Image(label="Output")
+
             output_text = gr.Textbox(label="Detection Summary")
 
         with gr.Column(scale=1):
@@ -147,6 +154,7 @@ with gr.Blocks() as demo:
             gr.Markdown("### Feedback")
 
             correct_btn = gr.Button("Correct", size="sm")
+
             incorrect_btn = gr.Button("Incorrect", size="sm")
 
             summary_box = gr.Textbox(label="Evaluation Summary")
@@ -167,4 +175,4 @@ with gr.Blocks() as demo:
         outputs=summary_box
     )
 
-demo.launch(theme=gr.themes.Soft())
+demo.launch()