Update app.py

app.py

@@ -33,6 +33,9 @@ except Exception as e:
     raise
 
 def detect_and_read_plate(image):
+    # Initialize an empty list to store detected plate texts
+    detected_texts = []
+    
     results = model(image)
     for result in results:
         boxes = result.boxes.xyxy.cpu().numpy()  # [x1, y1, x2, y2]
@@ -46,18 +49,27 @@ def detect_and_read_plate(image):
             ocr_result = reader.readtext(plate_img)
             if ocr_result:
                 text = " ".join([res[1] for res in ocr_result])
+                detected_texts.append(text)
                 print(f"Detected Plate: {text}")
-                # Draw bounding box + text
+                # Draw bounding box (optional, kept for visualization)
                 cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
-                cv2.putText(image, text, (x1, y1 - 10),
-                            cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 255, 0), 2)
-    return image
+                # Optionally remove text on image
+                # cv2.putText(image, text, (x1, y1 - 10),
+                #             cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 255, 0), 2)
+    
+    # Join detected texts (in case multiple plates are detected)
+    output_text = "\n".join(detected_texts) if detected_texts else "No license plate detected"
+    return image, output_text
 
+# Create Gradio interface
 demo = gr.Interface(
     fn=detect_and_read_plate,
-    inputs="image",
-    outputs="image",
+    inputs=gr.Image(type="numpy", label="Upload an image of a car"),
+    outputs=[
+        gr.Image(type="numpy", label="Detected License Plate Image"),
+        gr.Textbox(label="Detected License Plate Number")
+    ],
     title="Automatic Number Plate Recognition (ANPR)",
-    description="Upload an image of a car to detect and read its license plate."
+    description="Upload an image of a car to detect and read its license plate. The detected plate number will be shown below the image."
 )
-demo.launch(server_name="0.0.0.0", server_port=7860) 
+demo.launch(server_name="0.0.0.0", server_port=7860)