From a2d4ad0499e4ee4f3f528b2ed7bc162d61026d09 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 17 Dec 2018 01:13:32 +0100
Subject:  build

---
 megapixels/app/server/api.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

(limited to 'megapixels/app/server')

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]
-- 
cgit v1.2.3-70-g09d2