From 28fa95b09b5e69c5c3b25768befa9b30a3f80aab Mon Sep 17 00:00:00 2001
From: Kai Labusch <labusch@gmail.com>
Date: Tue, 18 Aug 2020 11:49:16 +0200
Subject: [PATCH] add cache

---
 README.md                     |  3 ++-
 qurator/sbb_ner/webapp/app.py | 11 +++++++++++
 requirements.txt              |  1 +
 3 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index e14f620..cf996d0 100644
--- a/README.md
+++ b/README.md
@@ -168,7 +168,8 @@ Output:
   ]
 ]
 ```
-The JSON above is the expected input format of the [SBB named entity disambiguation system](sbb_ner).
+The JSON above is the expected input format of the 
+[SBB named entity linking and disambiguation system](https://github.com/qurator-spk/sbb_ned).
 # Model-Training 
 
 ***
diff --git a/qurator/sbb_ner/webapp/app.py b/qurator/sbb_ner/webapp/app.py
index ea5f4c4..b1d566f 100644
--- a/qurator/sbb_ner/webapp/app.py
+++ b/qurator/sbb_ner/webapp/app.py
@@ -1,6 +1,8 @@
 import os
 import logging
 from flask import Flask, send_from_directory, redirect, jsonify, request
+from flask_caching import Cache
+from hashlib import sha256
 import html
 import json
 import torch
@@ -16,6 +18,8 @@ app = Flask(__name__)
 
 app.config.from_json('config.json' if not os.environ.get('CONFIG') else os.environ.get('CONFIG'))
 
+cache = Cache(app)
+
 logger = logging.getLogger(__name__)
 
 
@@ -149,6 +153,10 @@ predictor_store = PredictorStore()
 tokenizer = NERTokenizer()
 
 
+def key_prefix():
+    return "{}:{}".format(request.path, sha256(str(request.json).encode('utf-8')).hexdigest())
+
+
 @app.route('/')
 def entry():
     return redirect("/index.html", code=302)
@@ -160,6 +168,7 @@ def get_models():
 
 
 @app.route('/tokenized', methods=['GET', 'POST'])
+@cache.cached(key_prefix=key_prefix)
 def tokenized():
 
     raw_text = request.json['text']
@@ -173,6 +182,7 @@ def tokenized():
 
 @app.route('/ner-bert-tokens', methods=['GET', 'POST'])
 @app.route('/ner-bert-tokens/<model_id>', methods=['GET', 'POST'])
+@cache.cached(key_prefix=key_prefix)
 def ner_bert_tokens(model_id=None):
 
     raw_text = request.json['text']
@@ -198,6 +208,7 @@ def ner_bert_tokens(model_id=None):
 
 @app.route('/ner', methods=['GET', 'POST'])
 @app.route('/ner/<model_id>', methods=['GET', 'POST'])
+@cache.cached(key_prefix=key_prefix)
 def ner(model_id=None):
 
     raw_text = request.json['text']
diff --git a/requirements.txt b/requirements.txt
index 6b47ac8..2bd4239 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -14,5 +14,6 @@ cloudpickle
 pytest
 pytest-cov
 flask
+Flask-Caching
 gunicorn
 somajo