Update app.py

app.py

@@ -39,15 +39,17 @@ 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, img_bgr
+    return cropped_images, detected_image
 
 # Function to detect characters
 def detect_characters(image):
@@ -69,7 +71,10 @@ if uploaded_file is not None:
 
     # Detect license plates and characters
     with st.spinner("Processing image..."):
-        cropped_plates, original_image = detect_and_crop_license_plate(image)
+        cropped_plates, detected_image = detect_and_crop_license_plate(image)
+
+        # Show the image with detected license plates
+        st.image(cv2.cvtColor(detected_image, cv2.COLOR_BGR2RGB), caption="Detected License Plates", use_container_width=True)
 
         if cropped_plates:
             st.write(f"Detected {len(cropped_plates)} license plate(s). Showing results:")
@@ -80,7 +85,7 @@ if uploaded_file is not None:
                 st.image(cv2.cvtColor(annotated_plate, cv2.COLOR_BGR2RGB), caption=f"License Plate {idx} with Characters", use_container_width=True)
         else:
             st.write("No license plates detected. Running character detection on the full image.")
-            annotated_image = detect_characters(original_image.copy())
+            annotated_image = detect_characters(detected_image.copy())
             st.image(cv2.cvtColor(annotated_image, cv2.COLOR_BGR2RGB), caption="Full Image with Characters", use_container_width=True)
 
     st.success("Processing complete!")