build

1 files changed, 10 insertions, 4 deletions

diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py
index 2f78ecd3..bc60118c 100644
--- a/megapixels/app/server/api.py
+++ b/megapixels/app/server/api.py
@@ -58,6 +58,10 @@ def upload(name):
 
   # get detection as BBox object
   bboxes = detector.detect(im_np, largest=True)
+  if not len(bboxes):
+    return jsonify({
+      'error': 'bbox'
+    })
   bbox = bboxes[0]
   dim = im_np.shape[:2][::-1]
   bbox = bbox.to_dim(dim)  # convert back to real dimensions
@@ -73,16 +77,18 @@ def upload(name):
   distances, indexes = faiss_dataset.search(query, 10)
 
   if len(indexes) == 0:
-    print("weird, no results!")
-    return []
+    return jsonify({
+      'error': 'nomatch'
+    })
 
   # get the results for this single query...
   distances = distances[0]
   indexes = indexes[0]
 
   if len(indexes) == 0:
-    print("no results!")
-    return []
+    return jsonify({
+      'error': 'nomatch'
+    })
 
   lookup = {}
   ids = [i+1 for i in indexes]