feat: add secret_key validation

main.py

@@ -1,12 +1,22 @@
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, abort
 import os
 import stanza
 
+secret_key = os.getenv("SECRET_KEY", "")
+
 nlp = stanza.Pipeline("en", processors="tokenize", download_method=None)
 
 app = Flask(__name__)
 
 
+def guard_secret_key():
+    if request.args.get("secret_key", "") == secret_key:
+        return
+    if request.headers.get("X-Secret-Key", "") == secret_key:
+        return
+    abort(401)
+
+
 @app.route("/")
 def action_index():
     return "Hello, World!"
@@ -14,5 +24,6 @@ def action_index():
 
 @app.route("/sentence_segmentation", methods=["POST"])
 def action_sentences_segment():
+    guard_secret_key()
     doc = nlp(request.get_json()["document"])
     return jsonify({"sentences": [sentence.text for sentence in doc.sentences]})