Merge pull request #94 from sb7-winners/tkDocker

tk - init commit

requirements-service.txt

@@ -7,8 +7,8 @@ tqdm==4.32.2
 neuralcoref==4.0
 argparse
 scikit-learn
-bert-extractive-summarizer==0.4.2
+bert-extractive-summarizer==0.6.1
 Flask
 flask-cors
 nltk
-https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-2.1.0/en_core_web_sm-2.1.0.tar.gz
+https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-2.1.0/en_core_web_sm-2.1.0.tar.gz

server.py

@@ -51,9 +51,12 @@ def convert_to_paragraphs(self) -> str:
         sentences: List[str] = self.run()
         return ' '.join([sentence.strip() for sentence in sentences]).strip()
 
+@app.route('/', methods=['GET'])
+def hello_world():
+    return 'Hello, World!'
 
-@app.route('/summarize', methods=['POST'])
-def convert_raw_text():
+@app.route('/summarize_by_ratio', methods=['POST'])
+def convert_raw_text_by_ratio():
     ratio = float(request.args.get('ratio', 0.2))
     min_length = int(request.args.get('min_length', 25))
     max_length = int(request.args.get('max_length', 500))
@@ -69,6 +72,23 @@ def convert_raw_text():
         'summary': summary
     })
 
+@app.route('/summarize_by_sentence', methods=['POST'])
+def convert_raw_text_by_sent():
+    num_sentences = int(request.args.get('num_sentences', 5))
+    min_length = int(request.args.get('min_length', 25))
+    max_length = int(request.args.get('max_length', 500))
+
+    data = request.data
+    if not data:
+        abort(make_response(jsonify(message="Request must have raw text"), 400))
+
+    parsed = Parser(data).convert_to_paragraphs()
+    summary = summarizer(parsed, num_sentences=num_sentences, min_length=min_length, max_length=max_length)
+
+    return jsonify({
+        'summary': summary
+    })
+
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='')
@@ -81,7 +101,7 @@ def convert_raw_text():
     parser.add_argument('-greediness', dest='greediness', help='', default=0.45)
     parser.add_argument('-reduce', dest='reduce', help='', default='mean')
     parser.add_argument('-hidden', dest='hidden', help='', default=-2)
-    parser.add_argument('-port', dest='port', help='', default=5000)
+    parser.add_argument('-port', dest='port', help='', default=8080)
     parser.add_argument('-host', dest='host', help='', default='0.0.0.0')
 
     args = parser.parse_args()