Update app.py

app.py

@@ -5,6 +5,7 @@ from datasets import load_dataset
 from werkzeug.security import generate_password_hash, check_password_hash
 
 app = Flask(__name__)
+auth = HTTPBasicAuth()
 
 # Thiết lập HF_HOME trước khi import huggingface_hub hoặc datasets
 os.environ["HF_HOME"] = "/app/hf_cache"
@@ -41,6 +42,12 @@ data = load_data()
 ADMIN_USERNAME = "admin"
 ADMIN_PASSWORD_HASH = generate_password_hash("password")  # Đổi mật khẩu tại đây
 
+@auth.verify_password
+def verify_password(username, password):
+    if username == 'admin' and check_password_hash(ADMIN_PASSWORD_HASH, password):
+        return username
+    return None
+
 def check_auth():
     auth = request.authorization
     if not auth or auth.username != ADMIN_USERNAME or not check_password_hash(ADMIN_PASSWORD_HASH, auth.password):
@@ -88,6 +95,7 @@ def blog():
 
 # Route cho trang CMS
 @app.route('/cms')
+@auth.login_required
 def cms():
     if not check_auth():
         abort(401)  # Trả về lỗi 401 nếu không xác thực