From 276c16e1055c23350abd3d9d071cfce9b4f1b27f Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Fri, 4 Jan 2019 23:37:34 +0100
Subject: rm rotation on homepage, add dataset list

---
 megapixels/app/server/api.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'megapixels/app/server/api.py')

diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py
index 35862837..3683d5fd 100644
--- a/megapixels/app/server/api.py
+++ b/megapixels/app/server/api.py
@@ -45,18 +45,20 @@ def upload(dataset_name):
   dataset = get_dataset(dataset_name)
   if dataset_name not in faiss_datasets:
     return jsonify({
-      'error': 'invalid dataset'  
+      'error': 'bad_dataset'  
     })
   faiss_dataset = faiss_datasets[dataset_name]
   file = request.files['query_img']
   fn = file.filename
-  if fn.endswith('blob'):
+  if fn.endswith('blob'): # FIX PNG IMAGES?
     fn = 'filename.jpg'
 
   basename, ext = os.path.splitext(fn)
   # print("got {}, type {}".format(basename, ext))
   if ext.lower() not in valid_exts:
-    return jsonify({ 'error': 'not an image' })
+    return jsonify({
+      'error': 'not_an_image'
+    })
 
   im = Image.open(file.stream).convert('RGB')
   im_np = pil2np(im)
-- 
cgit v1.2.3-70-g09d2