Update app.py

app.py

@@ -1,11 +1,13 @@
 import edge_tts
+import hmac
+import hashlib
+import time
 from fastapi import FastAPI, Request
 from fastapi.responses import StreamingResponse, JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 
 app = FastAPI()
 
-# Mở toang cửa CORS để Web gọi vào không bị chặn
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -14,15 +16,22 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# 🔐 MẬT MÃ BÍ MẬT (CHỈ HF VÀ VPS CỦA SẾP ĐƯỢC BIẾT)
+SECRET_KEY = "VCLOUD_!@#_SECRET_2026"
+
 @app.get("/")
 def read_root():
+    # Điểm danh cho Cronjob/Worker (Không cần Token)
     return {"status": "VCLOUD Voice API is running optimally on Hugging Face!"}
 
 @app.post("/tts")
 @app.get("/tts")
 async def generate_tts(request: Request):
     text = ""
-    voice = "vi-VN-HoaiMyNeural" # Giọng đọc mặc định
+    voice = "vi-VN-HoaiMyNeural"
+    bot_uid = ""
+    req_time = "0"
+    token = ""
     
     if request.method == "POST":
         try:
@@ -34,22 +43,38 @@ async def generate_tts(request: Request):
     elif request.method == "GET":
         text = request.query_params.get("text", "").strip()
         voice = request.query_params.get("voice", voice)
+        bot_uid = request.query_params.get("bot_uid", "").strip()
+        req_time = request.query_params.get("time", "0").strip()
+        token = request.query_params.get("token", "").strip()
 
+    # Cho phép Admin Ping nút xanh trong Cài đặt (Không cần text)
     if not text:
-        return JSONResponse(
-            {"status": "success", "message": "Ping OK. Trạm Voice Hugging Face đang hoạt động hoàn hảo!"}, 
-            status_code=200
-        )
+        return JSONResponse({"status": "success", "message": "Ping OK."}, status_code=200)
 
+    # 🛡️ HÀNG RÀO BẢO MẬT CHỮ KÝ HMAC
     try:
-        communicate = edge_tts.Communicate(text, voice)
+        # 1. Chống dùng lại Token chôm chỉa (Hết hạn sau 12 tiếng = 43200s)
+        current_time = int(time.time())
+        if current_time - int(req_time) > 43200:
+            return JSONResponse({"error": "Token đã hết hạn!"}, status_code=401)
         
-        # 🟢 NÂNG CẤP STREAMING: Bơm thẳng byte âm thanh từ RAM về Web, không chạm vào ổ cứng
+        # 2. Đối chiếu Chữ ký (Mã hóa SHA-256)
+        message = f"{bot_uid}{req_time}".encode('utf-8')
+        expected_token = hmac.new(SECRET_KEY.encode('utf-8'), message, hashlib.sha256).hexdigest()
+        
+        if not hmac.compare_digest(expected_token, token):
+            return JSONResponse({"error": "Truy cập bị từ chối! Sai Token bảo mật."}, status_code=403)
+            
+    except Exception as e:
+        return JSONResponse({"error": "Lỗi xác thực bảo mật"}, status_code=401)
+
+    # 🚀 VƯỢT RÀO THÀNH CÔNG -> STREAM TỪ RAM
+    try:
+        communicate = edge_tts.Communicate(text, voice)
         async def audio_stream():
             async for chunk in communicate.stream():
                 if chunk["type"] == "audio":
                     yield chunk["data"]
-
         return StreamingResponse(audio_stream(), media_type="audio/mpeg")
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)