diff --git a/app.py b/app.py new file mode 100644 index 0000000000000000000000000000000000000000..64849d369758f76c4897f63fc1ae666d1444aae0 --- /dev/null +++ b/app.py @@ -0,0 +1,29 @@ + +# Copyright 2022, Frank Sauerburger + +"""Movie title suggestion api""" + +import json +from flask import Flask, request +import qgram + +app = Flask(__name__) + +with open("examples/titles.json") as fileobj: + titles = json.load(fileobj) + +index = qgram.QGramIndex() +for title, _ in titles.values(): + index.add_term(title) + +@app.route("/") +def suggest(): + query = request.args.get("q") + if not query: + return {} + + results = index.search(query) + return { + "suggestions": results[:100] + } + diff --git a/requirements.txt b/requirements.txt index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..52b3b9229be8650b39be951a1eca8d0a83aa6dfe 100644 --- a/requirements.txt +++ b/requirements.txt @@ -0,0 +1 @@ +flask~=2.1.2