rm rotation on homepage, add dataset list

1 files changed, 5 insertions, 3 deletions

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)