Update app.py

app.py

@@ -67,16 +67,23 @@ def classify_plate_color(plate_img):
         img = np.array(plate_img)
         hsv = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)
 
-        green = np.sum(cv2.inRange(hsv, (35,40,40), (85,255,255)))
-        yellow = np.sum(cv2.inRange(hsv, (15,50,50), (35,255,255)))
+        # Color masks
+        green_mask = cv2.inRange(hsv, (35, 40, 40), (85, 255, 255))
+        yellow_mask = cv2.inRange(hsv, (15, 50, 50), (35, 255, 255))
+        white_mask = cv2.inRange(hsv, (0, 0, 200), (180, 40, 255))
 
-        if green > yellow:
+        green = np.sum(green_mask)
+        yellow = np.sum(yellow_mask)
+        white = np.sum(white_mask)
+
+        if green > yellow and green > white:
             return "EV"
-        elif yellow > green:
+        elif yellow > green and yellow > white:
             return "Commercial"
-        return "Personal"
-    except:
-        return "Unknown"
+        elif white > green and white > yellow:
+            return "Personal"
+        else:
+            return "Unknown"
 
 def read_plate(plate_img):
     try: