--- a +++ b/src/api.py @@ -0,0 +1,28 @@ +from flask import Flask, redirect, url_for, request +from clustering.search import search_query +import json + +print("ll") +app = Flask(__name__) + +print("ll") + +@app.route("/hello") +def hello(): + return "<h1 style='color:blue'>Hello There!</h1>" + +print("jj") +@app.route("/indexes/train-index/docs/search", methods=["POST"]) +def main(): + request_data = request.get_json() + print("filters", request_data.get("filters")) + filters =json.loads(request_data.get("filters")) + records, count_filtered = search_query(query=request_data["query"], filters=filters, top=request_data["top"]) + assert len(set([record["id"] for record in records])) == len(records), "ids of results are not unique" + res = {"value": records, "count": count_filtered} + return json.dumps(res, indent=2) + + +if __name__ == "__main__": + print("mmmm") + app.run(host="localhost", port=5000)