all the sql stuff communicating nicely

1 files changed, 72 insertions, 0 deletions

diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py
new file mode 100644
index 00000000..e7db11f1
--- /dev/null
+++ b/megapixels/app/server/api.py
@@ -0,0 +1,72 @@
+from flask import Blueprint, jsonify
+
+from app.models.sql_factory import list_datasets, get_dataset, get_table
+
+# from jinja2 import TemplateNotFound
+
+# import os
+# import sys
+# import json
+# import time
+# import argparse
+# import cv2 as cv
+# import numpy as np
+# from datetime import datetime
+# from flask import Flask, request, render_template, jsonify
+# from PIL import Image  # todo: try to remove PIL dependency
+# import re
+
+# sanitize_re = re.compile('[\W]+')
+# valid_exts = ['.gif', '.jpg', '.jpeg', '.png']
+
+# from dotenv import load_dotenv
+# load_dotenv()
+
+# from feature_extractor import FeatureExtractor
+
+# DEFAULT_LIMIT = 50
+
+api = Blueprint('api', __name__)
+
+@api.route('/')
+def index():
+  return jsonify({ 'datasets': list_datasets() })
+
+@api.route('/dataset/<dataset>/test', methods=['POST'])
+def test(dataset='test'):
+  dataset = get_dataset(dataset)
+  print('hiiiiii')
+  return jsonify({ 'test': 'OK', 'dataset': dataset })
+
+# @router.route('/<dataset>/face', methods=['POST'])
+# def upload(name):
+#   file = request.files['query_img']
+#   fn = file.filename
+#   if fn.endswith('blob'):
+#     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' })
+
+#   uploaded_fn = datetime.now().isoformat() + "_" + basename
+#   uploaded_fn = sanitize_re.sub('', uploaded_fn)
+#   uploaded_img_path = "static/uploaded/" + uploaded_fn + ext
+#   uploaded_img_path = uploaded_img_path.lower()
+#   print('query: {}'.format(uploaded_img_path))
+
+#   img = Image.open(file.stream).convert('RGB')
+#   # img.save(uploaded_img_path)
+#   # vec = db.load_feature_vector_from_file(uploaded_img_path)
+#   vec = fe.extract(img)
+#   # print(vec.shape)
+
+#   results = db.search(vec, limit=limit)
+#   query = {
+#     'timing': time.time() - start,
+#   }
+#   print(results)
+#   return jsonify({
+#     'results': results,
+#   })