Check image 0 bytes and give 400 if invalid request

src/api/app_utils.py

@@ -110,6 +110,11 @@ def extract_request_data() -> Tuple[bytes, str, str, str, list]:
 
     image_content = image_file.read()
 
+    # Check if the image content is empty or None
+    if image_content is None or len(image_content) == 0:
+        raise ValueError(
+            "The uploaded image is empty. Please upload a valid image file.")
+
     # Extract other form data
     group_id = request.form.get('group_id')
     if not group_id:

src/api/routes.py

@@ -51,7 +51,19 @@ def predict() -> flask.Response:
 
     # Add incoming request to queue
     # Here, we're just queuing the raw data.
-    image_file, group_id, identifier, model, whitelist = extract_request_data()
+    try:
+        image_file, group_id, identifier, model, whitelist = extract_request_data()
+    except ValueError as e:
+        response = jsonify({
+            "status": "error",
+            "code": 400,
+            "message": str(e),
+            "timestamp": datetime.datetime.now().isoformat()
+        })
+
+        response.status_code = 400
+        logger.error("Error processing request: %s", str(e))
+        return response
 
     logger.debug("Data received: %s, %s", group_id, identifier)
     logger.debug("Adding %s to queue", identifier)