Update app.py

app.py

@@ -8,8 +8,17 @@ app = Flask(__name__)
 model = YOLO("yolov8n.pt")
 
 def readb64(base64_string):
-    nparr = np.frombuffer(base64.b64decode(base64_string.split(',')[1]), np.uint8)
-    return cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+    if ',' not in base64_string:
+        raise ValueError("Invalid base64 image string.")
+    header, encoded = base64_string.split(',', 1)
+    if not encoded:
+        raise ValueError("Empty base64 image data.")
+    nparr = np.frombuffer(base64.b64decode(encoded), np.uint8)
+    img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+    if img is None:
+        raise ValueError("Image decoding failed.")
+    return img
+
 
 @app.route('/')
 def index():