Overall update app.py

app.py

@@ -64,6 +64,7 @@ class App:
 
 p = App()
 
+black_list = ["Y985BE152"]
 
 def detect_license_plates(model, image):
     plate_image_np = pil_to_np(image)
@@ -120,7 +121,7 @@ def process_image(image_path: Image):
     ax.imshow(image_np)
 
     plates = detect_license_plates(license_plate_detector, image_path)
-    recognized_text = ""
+    recognized_texts = []
 
     for i, (plate, bbox) in enumerate(plates):
         preprocessed_plate = preprocess_license_plate(plate)
@@ -133,6 +134,8 @@ def process_image(image_path: Image):
             recognized_text = recognized_text.replace('7', 'T', 1) if recognized_text[0] == "7" else recognized_text
             recognized_text = recognized_text.replace('0', 'O', 1) if recognized_text[0] == "0" else recognized_text
             recognized_text = recognized_text[:9] if len(recognized_text) >= 9 else recognized_text
+            if recognized_text not in black_list:
+                recognized_texts.append(recognized_text)
 
         x1, y1, x2, y2 = bbox
         rect = patches.Rectangle((x1, y1), x2 - x1, y2 - y1, linewidth=2, edgecolor='r', facecolor='none')
@@ -146,7 +149,7 @@ def process_image(image_path: Image):
     plt.savefig(output_buffer, bbox_inches='tight')
     plt.close()
 
-    return Image.open(output_buffer), recognized_text.strip()
+    return Image.open(output_buffer), recognized_texts
 
 
 # Gradio UI
@@ -161,7 +164,7 @@ interface = gr.Interface(
     inputs=gr.Image(type="pil", label="Загрузите фото машины с номером 📤"),
     outputs=[
         gr.Image(type="pil", label="📸 Выход 0 - Обработанное изображение"),
-        gr.Text(label="🔍 Выход 1 - Распознанный номер"),
+        gr.JSON(label="🔍 Выход 1 - Распознанный номер"),
     ],
     title="Распознавание российских номеров",
     description="🔎 **Загрузите изображение с автомобильным номером** и модель автоматически **определит госномер!** 🔥\n\n📸 **Форматы:** JPG, PNG, BMP",
@@ -171,4 +174,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch(share=True)
+    interface.launch(share=True)