Update app.py

app.py

@@ -38,8 +38,8 @@ def load_models():
 # Load models
 full_plate_model, character_model, recognition_model = load_models()
 
-# Function to detect license plates and prepare for character recognition
-def process_license_plate(image):
+# Function to detect license plates
+def detect_license_plate(image):
     img_bgr = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
     results = full_plate_model(img_bgr)
 
@@ -52,24 +52,18 @@ def process_license_plate(image):
 
     return detected_image, img_bgr
 
-# Function to detect and recognize characters directly from the original image
-def detect_and_recognize_characters(image):
-    # Detect characters
+# Function to detect characters
+def detect_characters(image):
     results = character_model(image)
+    detected_image = image.copy()
     character_crops = []
     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), (0, 255, 0), 2)  # Draw bounding box
                 character_crops.append(image[y1:y2, x1:x2])
-    
-    # Recognize characters
-    if character_crops:
-        recognized_characters = recognize_characters(character_crops)
-    else:
-        recognized_characters = []
-
-    return recognized_characters
+    return detected_image, character_crops
 
 # Function to recognize characters
 def recognize_characters(character_crops):
@@ -98,17 +92,22 @@ if uploaded_file is not None:
     # Load image
     image = Image.open(uploaded_file)
 
-    # Process the image for license plate detection
+    # Detect license plates
     with st.spinner("Processing image..."):
-        detected_image, original_image = process_license_plate(image)
+        detected_image, original_image = detect_license_plate(image)
 
-        # Show the detected image with bounding boxes
+        # Show the detected license plate image with bounding boxes
         st.image(cv2.cvtColor(detected_image, cv2.COLOR_BGR2RGB), caption="Detected License Plates", use_container_width=True)
 
-        # Perform character detection and recognition on the original image
-        recognized_characters = detect_and_recognize_characters(original_image)
+        # Detect characters on the original image
+        character_detected_image, character_crops = detect_characters(original_image)
+
+        # Show the detected characters with bounding boxes
+        st.image(cv2.cvtColor(character_detected_image, cv2.COLOR_BGR2RGB), caption="Detected Characters", use_container_width=True)
 
-        if recognized_characters:
+        # Recognize characters from the detected character crops
+        if character_crops:
+            recognized_characters = recognize_characters(character_crops)
             st.write("Recognized Characters:", "".join(recognized_characters))
         else:
             st.write("No characters detected in the image.")