No empty logs and fixes some errors

application/api/image_processing.py

@@ -20,11 +20,15 @@ async def process_image(image: bytes) -> str:
         if r.boxes:
             for box in r.boxes:
                 cls_name = r.names[int(box.cls[0])]
-                if cls_name == "car":
+                if cls_name in ["car", "truck"]:
                     x1, y1, x2, y2 = map(int, box.xyxy[0])
                     size = (x2 - x1) ** 2 + (y2 - y1) ** 2
                     cars.append((size, (x1, y1, x2, y2)))
+        else:
+            return ""
 
+    if cars == []:
+        return ""
     # Get the biggest car box
     size, corners = max(cars, key=lambda x: x[0])
 
@@ -45,5 +49,7 @@ async def process_image(image: bytes) -> str:
                     lp_np = np.array(object=lp_img)
                     result = ocr_reader.readtext(image=lp_np)
                     print(result)
+        else:
+            return ""
 
     return str(result[0][1])  # type: ignore

application/api/views.py

@@ -17,7 +17,8 @@ def data():
     application.last_image.seek(0)
     application.last_image = io.BytesIO(request.data)
     np = asyncio.run(process_image(image=data))
-
+    allowed = False
+    if np != "":
         # Check if the found plate is allowed to exit
         allowed = AllowedPlate.query.filter_by(plate=np).first() is not None
         db.session.add(  # Log the found numberplate and status

application/dashboard/views.py

@@ -24,7 +24,9 @@ def dashboard():
         .all()
     )
     form = npForm()
-    return render_template("dashboard.html", plates=Plates, logs=logs, form=form)
+    return render_template(
+        "dashboard.html", plates=Plates, logs=logs, form=form
+    )
 
 
 @dash_blueprint.route("/add", methods=["GET", "POST"])

pico_files/main.py

@@ -15,19 +15,23 @@ good_pin = Pin(15, Pin.OUT)
 bad_pin = Pin(14, Pin.OUT)
 
 # Send the POST request with the raw image data as the body
-url = f'http://{config.server}:{config.port}/{config.data_path}'
+url = f"http://{config.server}:{config.port}/{config.data_path}"
 
-spi = SPI(0,sck=Pin(18), miso=Pin(16), mosi=Pin(19), baudrate=8000000) # Pins for the Raspberry Pi Pico
+spi = SPI(
+    0, sck=Pin(18), miso=Pin(16), mosi=Pin(19), baudrate=8000000
+)  # Pins for the Raspberry Pi Pico
 cs = Pin(17, Pin.OUT)
 cam = Camera(spi, cs, debug_text_enabled=True)
 headers = {"Content-Type": "image/jpeg"}
 addr = socket.getaddrinfo(config.server, config.port)[0][-1]
+
+
 def send_image(image_file_path):
     cam.capture_jpg()
-    with open(image_file_path, 'wb') as f:
+    with open(image_file_path, "wb") as f:
         f.write(b"")
     cam.save_JPG(image_file_path)
-    with open(image_file_path, 'rb') as f:
+    with open(image_file_path, "rb") as f:
         s = socket.socket()
         s.connect(addr)
         # Send HTTP headers
@@ -57,6 +61,7 @@ def send_image(image_file_path):
 
         return
 
+
 while connection.isconnected():
     good_pin.low()
     bad_pin.low()