Update app.py

app.py

@@ -43,28 +43,28 @@ def detect_and_crop_license_plate(image):
     img_bgr = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
     results = full_plate_model(img_bgr)
 
+    detected_image = img_bgr.copy()
     cropped_images = []
     for result in results:
         if hasattr(result, 'boxes') and result.boxes is not None:
             for box in result.boxes.xyxy:
                 x1, y1, x2, y2 = map(int, box)
+                cv2.rectangle(detected_image, (x1, y1), (x2, y2), (255, 0, 0), 2)  # Draw bounding box
                 cropped_image = img_bgr[y1:y2, x1:x2]
                 cropped_images.append(cropped_image)
 
-    return cropped_images
+    return cropped_images, detected_image
 
-# Function to detect characters from license plates
-def detect_characters(license_plate):
-    results = character_model(license_plate)
+# Function to detect and crop characters
+def detect_and_crop_characters(image):
+    results = character_model(image)
     character_crops = []
-    bounding_boxes = []
     for result in results:
         if hasattr(result, 'boxes') and result.boxes is not None:
             for box in result.boxes.xyxy:
                 x1, y1, x2, y2 = map(int, box)
-                character_crops.append(license_plate[y1:y2, x1:x2])
-                bounding_boxes.append((x1, y1, x2, y2))
-    return character_crops, bounding_boxes
+                character_crops.append(image[y1:y2, x1:x2])
+    return character_crops
 
 # Function to recognize characters
 def recognize_characters(character_crops):
@@ -95,26 +95,29 @@ if uploaded_file is not None:
 
     # Detect license plates
     with st.spinner("Processing image..."):
-        cropped_plates = detect_and_crop_license_plate(image)
+        cropped_plates, detected_image = detect_and_crop_license_plate(image)
 
         if cropped_plates:
+            st.image(cv2.cvtColor(detected_image, cv2.COLOR_BGR2RGB), caption="Detected License Plates", use_container_width=True)
+            st.write(f"Detected {len(cropped_plates)} license plate(s).")
+
             for idx, cropped_plate in enumerate(cropped_plates, 1):
                 st.write(f"Processing License Plate {idx}:")
-                character_crops, bounding_boxes = detect_characters(cropped_plate)
+                character_crops = detect_and_crop_characters(cropped_plate)
 
-                # Draw bounding boxes on the cropped license plate
-                for box in bounding_boxes:
-                    x1, y1, x2, y2 = box
-                    cv2.rectangle(cropped_plate, (x1, y1), (x2, y2), (0, 255, 0), 2)
+                if character_crops:
+                    recognized_characters = recognize_characters(character_crops)
+                    st.write("Recognized Characters:", "".join(recognized_characters))
+                else:
+                    st.write("No characters detected in this license plate.")
+        else:
+            st.write("No license plates detected. Running character detection on the full image.")
+            character_crops = detect_and_crop_characters(cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR))
 
-                # Recognize characters
+            if character_crops:
                 recognized_characters = recognize_characters(character_crops)
+                st.write("Recognized Characters:", "".join(recognized_characters))
+            else:
+                st.write("No characters detected in the full image.")
 
-                # Show the annotated license plate with bounding boxes
-                st.image(cv2.cvtColor(cropped_plate, cv2.COLOR_BGR2RGB), caption=f"License Plate {idx} with Bounding Boxes", use_container_width=True)
-
-                # Display recognized characters
-                st.write("Recognized Characters: ", "".join(recognized_characters))
-        else:
-            st.write("No license plates detected.")
     st.success("Processing complete!")