Update app.py

app.py

@@ -18,6 +18,9 @@ import threading
 import uuid
 import time
 from human_text_detect import detect_human_text
+import os
+
+HF_TOKEN = os.getenv("HF_TOKEN")
 
 app = Flask(__name__)
 CORS(app)
@@ -62,6 +65,10 @@ def process_analysis(task_id, text, model_name, topic, threshold):
 
 @app.route('/detectHumanInAIText/checkText', methods=['POST'])
 def check_text():
+    received_token = request.headers.get("HF_TOKEN")
+    if received_token != HF_TOKEN:
+        return jsonify({"error": "Unauthorized"}), 403  # Forbidden
+    
     # Get data
     print('Get data')
     data = request.get_json()
@@ -82,6 +89,10 @@ def check_text():
 
 @app.route('/detectHumanInAIText/getAnalyzeResults', methods=['GET'])
 def get_results():
+    received_token = request.headers.get("HF_TOKEN")
+    if received_token != HF_TOKEN:
+        return jsonify({"error": "Unauthorized"}), 403  # Forbidden
+        
     task_id = request.args.get('taskId')
     
     if not task_id: