Update licensing_client.py

licensing_client.py

@@ -17,7 +17,6 @@ from loguru import logger
 STORAGE_FILE = "Hugging AbuAlone09/AI-Video-Engine-storage"
 
 # ĐỌC BIẾN MÔI TRƯỜNG AN TOÀN TỪ SECRET CỦA HUGGING FACE
-# Ông chỉ cần điền Link gốc và API Key vào ô Secret trên Settings của Space Tool chính, code tự động bốc lên chạy ngầm
 SERVER_URL = os.getenv("LICENSIFY_SERVER_URL", "https://abualone09-my-licensify-server.hf.space").strip()
 SECRET_API_KEY = os.getenv("SECRET_API_KEY", "YOUR_SECRET_API_KEY_HERE").strip()
 
@@ -28,7 +27,6 @@ if not os.path.exists(STORAGE_FILE):
         json.dump({"keys": {}, "free_devices": {}}, f, indent=4)
 
 # --- THÀNH PHẦN QUAN TRỌNG: QUẢN LÝ TRẠNG THÁI 6 LUỒNG TOÀN CỤC (MEMORY LOCK) ---
-# Slot 1 -> 5: Ưu tiên VIP Premium. Slot 6: Dành riêng cho Free (VIP có quyền đẩy tiến trình)
 THREAD_POOL = {
     1: {"status": "idle", "key": None, "type": None, "pid": None, "device": None},
     2: {"status": "idle", "key": None, "type": None, "pid": None, "device": None},
@@ -38,6 +36,11 @@ THREAD_POOL = {
     6: {"status": "idle", "key": None, "type": None, "pid": None, "device": None},
 }
 
+# --- BỔ SUNG HÀM ĐỂ BÊN RUN_APP.PY GỌI ĐỒNG BỘ REALTIME THEO GIÂY MÀ KHÔNG BỊ LỖI ---
+def get_active_threads_count() -> int:
+    """Trả về tổng số luồng thực tế đang bận render trong hệ thống cluster"""
+    return sum(1 for t in THREAD_POOL.values() if t["status"] == "rendering")
+
 # =========================================================
 # CORE 1: XỬ LÝ ĐỌC / GHI & XÁC THỰC API KEY TỪ XA TỚI SERVER
 # =========================================================
@@ -56,14 +59,12 @@ def clean_expired_keys():
     now_ts = time.time()
     changed = False
     
-    # Quét sạch VIP Key hết hạn khỏi hệ thống lưu trữ
     for k, info in list(data["keys"].items()):
         if now_ts > info.get("expiry_timestamp", 0):
             del data["keys"][k]
             changed = True
             logger.warning(f"Key {k} expired and has been automatically purged from storage.")
             
-    # Reset lượt Free nếu hệ thống chuyển dịch sang ngày mới
     today_str = datetime.now().strftime("%Y-%m-%d")
     for dev, info in list(data["free_devices"].items()):
         if info.get("last_date") != today_str:
@@ -75,51 +76,77 @@ def clean_expired_keys():
         _save_storage(data)
 
 def verify_and_get_license_info(key_input: str, device_id: str) -> tuple:
-    """Kiểm tra cục bộ trước, nếu không có gọi API sang Server cổng thanh toán đối soát"""
+    """Kiểm tra cục bộ qua Hugging Face Secret trước, nếu không khớp gọi API sang Server đối soát"""
     clean_expired_keys()
     
-    # 1. Cơ chế đăng nhập bằng mã Admin để Test nhanh hệ thống bypass
-    if key_input.strip() == "ADMIN_SECRET_TEST_2026":
+    token = key_input.strip()
+    
+    # LẤY VIP KEY VÀ ADMIN KEY AN TOÀN TRỰC TIẾP TỪ MỤC SECRET CỦA HUGGING FACE
+    admin_secret = os.getenv("ADMIN_SECRET_KEY", "ADMIN_SECRET_TEST_2026").strip()
+    vip_secret = os.getenv("VIP_SECRET_KEY", "VIP_EARLY_ACCESS_2026").strip()
+    
+    # 1. Cơ chế đăng nhập bằng mã Admin tối cao lấy từ Secret (Không giới hạn, mở Panel Test)
+    if token == admin_secret:
+        return True, {
+            "tier": "ADMIN", 
+            "tx_name": "System Master Administrator", 
+            "amount": "$0.00 (Root Privilege)",
+            "tx_date": datetime.now().strftime("%Y-%m-%d"), 
+            "expiry": "Permanent Access", 
+            "days_left": 9999,
+            "msg": "📋 SYSTEM STATUS: Admin Master Active. Full hardware diagnostic test layer unlocked.",
+            "show_test_panel": True,
+            "remove_watermark": True,
+            "bypass_limits": True
+        }
+        
+    # 2. Cơ chế đặc quyền ưu đãi dành cho VIP Key lấy từ Secret (Bypass giới hạn, không hiện Panel Test)
+    if token == vip_secret:
         return True, {
-            "tier": "ADMIN", "tx_name": "System Internal Tester", "amount": "$0.00",
-            "tx_date": "2026-05-19", "expiry": "2026-12-31", "days_left": 999,
-            "msg": "📋 SYSTEM STATUS: Admin Test Bypass Active. All threads unlocked."
+            "tier": "VIP", 
+            "tx_name": "VIP Promotional Access", 
+            "amount": "$0.00 (Promo Tier)",
+            "tx_date": datetime.now().strftime("%Y-%m-%d"), 
+            "expiry": "2026-12-31", 
+            "days_left": 200,
+            "msg": "👑 VIP STATUS: Promo Key Verified. Watermarks and daily rendering limits removed.",
+            "show_test_panel": False,
+            "remove_watermark": True,
+            "bypass_limits": True
         }
         
     data = _load_storage()
     
-    # 2. Tra cứu kho lưu trữ cục bộ trước để tăng tốc độ phản hồi cho giao diện điện thoại
-    if key_input in data["keys"]:
-        info = data["keys"][key_input]
+    # 3. Tra cứu kho lưu trữ cục bộ trước để tăng tốc độ phản hồi cho giao diện điện thoại
+    if token in data["keys"]:
+        info = data["keys"][token]
         days_left = int((info["expiry_timestamp"] - time.time()) / 86400)
         return True, {
             "tier": "VIP", "tx_name": info["tx_name"], "amount": info["amount"],
             "tx_date": info["tx_date"], "expiry": info["expiry"], "days_left": max(0, days_left),
-            "msg": f"👑 VIP ACTIVE | User: {info['tx_name']} | {max(0, days_left)} days remaining."
+            "msg": f"👑 VIP ACTIVE | User: {info['tx_name']} | {max(0, days_left)} days remaining.",
+            "show_test_panel": False,
+            "remove_watermark": True,
+            "bypass_limits": False
         }
         
-    # 3. Giao tiếp API an toàn từ xa kết nối thẳng tới Kho lưu trữ của Cổng thanh toán dịch vụ
+    # 4. Giao tiếp API an toàn từ xa kết nối thẳng tới Kho lưu trữ của Cổng thanh toán dịch vụ công khai
     try:
-        # Gài mã bảo mật Secret nội bộ vào lớp vỏ Header để thông quan
         headers = {
             "X-API-Key": SECRET_API_KEY,
             "Content-Type": "application/json"
         }
-        payload = {"key": key_input.strip(), "hwid": device_id}
-        
-        # Gọi lệnh POST gửi dữ liệu JSON thông mạch tới cổng dịch vụ của ông
+        payload = {"key": token, "hwid": device_id}
         response = requests.post(f"{SERVER_URL}/api/verify-key", json=payload, headers=headers, timeout=10)
         
         if response.status_code == 200:
             res_data = response.json()
             if res_data.get("status") == "success":
-                # Bóc tách chuẩn xác toàn bộ thông tin giao dịch cần giám sát từ Server dịch vụ trả về
                 tx_info = res_data.get("data", {})
-                expiry_str = tx_info.get("expiry_date") # Định dạng YYYY-MM-DD từ cổng thanh toán
+                expiry_str = tx_info.get("expiry_date")
                 expiry_ts = time.mktime(time.strptime(expiry_str, "%Y-%m-%d"))
                 
-                # Lưu trữ đồng bộ vào file nội bộ để theo dõi tiến trình render
-                data["keys"][key_input] = {
+                data["keys"][token] = {
                     "tx_name": tx_info.get("buyer_name", "Anonymous User"),
                     "amount": tx_info.get("amount_paid", "$2.99"),
                     "tx_date": tx_info.get("payment_date", datetime.now().strftime("%Y-%m-%d")),
@@ -132,19 +159,21 @@ def verify_and_get_license_info(key_input: str, device_id: str) -> tuple:
                 return True, {
                     "tier": "VIP", "tx_name": tx_info.get("buyer_name"), "amount": tx_info.get("amount_paid"),
                     "tx_date": tx_info.get("payment_date"), "expiry": expiry_str, "days_left": max(0, days_left),
-                    "msg": f"👑 VIP KEY VERIFIED | Owner: {tx_info.get('buyer_name')} | Amount: {tx_info.get('amount_paid')} | Expiry: {expiry_str} ({max(0, days_left)} days left)."
+                    "msg": f"👑 VIP KEY VERIFIED | Owner: {tx_info.get('buyer_name')} | Expiry: {expiry_str}.",
+                    "show_test_panel": False,
+                    "remove_watermark": True,
+                    "bypass_limits": False
                 }
-        return False, {"msg": "❌ Invalid License Key or communication failure with payment gateway!"}
+        return False, {"msg": "❌ Invalid Access Key or communication failure with payment gateway!", "show_test_panel": False, "remove_watermark": False, "tier": "FREE"}
     except Exception as e:
-        return False, {"msg": f"⚠️ Connection error to Licensify Server! Verify your Secret URL configs. Info: {str(e)}"}
+        return False, {"msg": f"⚠️ Connection error to Licensify Server! Info: {str(e)}", "show_test_panel": False, "remove_watermark": False, "tier": "FREE"}
 
 # =========================================================
 # CORE 2: CƠ CHẾ ĐIỀU PHỐI 6 LUỒNG ĐỘNG & BIÊN TRỰC VIP / FREE
 # =========================================================
 
 def get_thread_status_json():
-    """Trả về trạng thái phân bổ thời gian thực của 6 luồng theo từng giây lên UI"""
-    busy_count = sum(1 for t in THREAD_POOL.values() if t["status"] == "rendering")
+    busy_count = get_active_threads_count()
     vip_count = sum(1 for t in THREAD_POOL.values() if t["type"] == "VIP")
     free_count = sum(1 for t in THREAD_POOL.values() if t["type"] == "FREE")
     return {
@@ -155,71 +184,56 @@ def get_thread_status_json():
     }
 
 def allocate_render_thread(key_input: str, device_id: str, is_vip: bool) -> tuple:
-    """
-    Thuật toán phân phối 6 luồng động cao cấp:
-    - Mỗi VIP Key chỉ được tạo duy nhất 1 luồng/thiết bị, chống cắm đa luồng trùng lặp.
-    - VIP tự động đẩy tiến trình Free tại Slot 6 nếu hệ thống đầy.
-    - Chặn đứng mọi k���t nối vượt ngưỡng (Người thứ 7 báo đầy luồng).
-    """
     global THREAD_POOL
     
-    # Khóa đơn luồng thiết bị: Kiểm tra trùng lặp tiến trình đang xử lý
     for slot, thread in THREAD_POOL.items():
         if thread["status"] == "rendering":
-            if is_vip and thread["key"] == key_input:
-                return False, "❌ This VIP Key is already running a rendering process on another device! Multi-thread allocation denied."
+            if is_vip and thread["key"] == key_input.strip():
+                return False, "❌ This VIP Key is already running a rendering process! Multi-thread allocation denied."
             if not is_vip and thread["device"] == device_id:
                 return False, "❌ Your device is currently rendering a video. Please wait until it completes!"
 
     target_slot = None
     
     if is_vip:
-        # VIP quét tìm không gian slot trống từ Slot 1 đến Slot 5
         for i in range(1, 6):
             if THREAD_POOL[i]["status"] == "idle":
                 target_slot = i
                 break
         
-        # Nếu 5 slot VIP đã full, VIP tiến hành kiểm tra chiếm dụng Slot số 6
         if not target_slot:
             if THREAD_POOL[6]["status"] == "idle":
                 target_slot = 6
             elif THREAD_POOL[6]["type"] == "FREE":
-                # 🔥 CƠ CHẾ ĐẨY TIẾN TRÌNH: VIP chiếm chỗ, gửi tín hiệu hủy trực tiếp PID của Free
                 free_pid = THREAD_POOL[6]["pid"]
-                logger.warning(f"👑 VIP Eviction Triggered: Terminating Free PID {free_pid} at Slot 6 to reclaim resources.")
+                logger.warning(f"👑 VIP Eviction Triggered: Terminating Free PID {free_pid} at Slot 6.")
                 try:
                     if free_pid:
-                        os.kill(free_pid, signal.SIGKILL) # Chấm dứt ngầm tiến trình FFmpeg của tài khoản Free
+                        os.kill(free_pid, signal.SIGKILL)
                 except ProcessLookupError:
                     pass
                 
                 target_slot = 6
-                # Trả Slot 6 về mặc định trống để nạp đè VIP vào luôn
                 THREAD_POOL[6] = {"status": "idle", "key": None, "type": None, "pid": None, "device": None}
     else:
-        # Tài khoản Free: Chỉ được phép nạp vào Slot số 6 nếu hoàn toàn trống
         if THREAD_POOL[6]["status"] == "idle":
             target_slot = 6
 
-    # BIÊN CHẶN NGƯỜI THỨ 7: Toàn bộ 6 luồng hệ thống đã đạt đỉnh tải
     if not target_slot:
         return False, "⚠️ All rendering channels are currently full. Please try again in a few moments!"
 
     return True, target_slot
 
 def register_process_to_slot(slot: int, key_input: str, device_id: str, is_vip: bool, pid: int):
-    """Ghi nhận định danh tiến trình hệ thống (PID) vào sơ đồ luồng"""
     THREAD_POOL[slot] = {
         "status": "rendering",
-        "key": key_input if is_vip else None,
+        "key": key_input.strip() if is_vip else None,
         "type": "VIP" if is_vip else "FREE",
         "pid": pid,
         "device": device_id
     }
 
 def release_thread_slot(slot: int):
-    """Giải phóng trạng thái slot sau khi tiến trình hoàn tất hoặc xuất file thành công"""
     if slot in THREAD_POOL:
         THREAD_POOL[slot] = {"status": "idle", "key": None, "type": None, "pid": None, "device": None}
 
@@ -228,7 +242,6 @@ def release_thread_slot(slot: int):
 # =========================================================
 
 def check_generation_limits(key_input: str, device_id: str, is_vip: bool) -> tuple:
-    """Kiểm tra hạn mức tạo video sát sao theo thời gian thực"""
     data = _load_storage()
     today_str = datetime.now().strftime("%Y-%m-%d")
     now_ts = time.time()
@@ -238,33 +251,31 @@ def check_generation_limits(key_input: str, device_id: str, is_vip: bool) -> tup
             return False, "License error: Key missing from verified list."
         
         vip_data = data["keys"][key_input]
-        # Khởi tạo bộ đếm mới nếu bước sang ngày mới theo chu kỳ cổng thanh toán cung cấp
         if vip_data.get("last_date_used") != today_str:
             vip_data["last_date_used"] = today_str
-            vip_data["daily_batches"] = 0 # Tối đa 5 lượt / ngày (tổng 10 video)
-            vip_data["videos_this_batch"] = 0 # Tối đa 2 video / lượt tạo
+            vip_data["daily_batches"] = 0 
+            vip_data["videos_this_batch"] = 0 
             vip_data["cooldown_until"] = 0
             
         if now_ts < vip_data.get("cooldown_until", 0):
             wait_min = int((vip_data["cooldown_until"] - now_ts) / 60)
-            return False, f"⏱️ Cooldown active! You have reached your limit for this batch. Please return in {wait_min} minutes."
+            return False, f"⏱️ Cooldown active! Please return in {wait_min} minutes."
             
         if vip_data.get("daily_batches", 0) >= 5:
-            return False, "❌ You have exhausted your daily limit of 5 batches (10 videos maximum). Please return tomorrow!"
+            return False, "❌ You have exhausted your daily limit of 5 batches. Please return tomorrow!"
             
         remaining_batches = 5 - vip_data.get("daily_batches", 0)
         return True, {
             "videos_created": vip_data.get("videos_this_batch", 0),
             "batches_created": vip_data.get("daily_batches", 0),
             "remaining_batches": remaining_batches,
-            "status_str": f"VIP Usage: Batch {vip_data.get('daily_batches', 0)}/5 | Current Batch: {vip_data.get('videos_this_batch', 0)}/2 Videos"
+            "status_str": f"VIP Usage: Batch {vip_data.get('daily_batches', 0)}/5 | Batch Videos: {vip_data.get('videos_this_batch', 0)}/2"
         }
     else:
-        # Cơ chế nhận diện thiết bị chống spam cho khách hàng không dùng key (FREE)
         if device_id not in data["free_devices"]:
             data["free_devices"][device_id] = {
                 "last_date": today_str,
-                "daily_batches": 0, # Tối đa 3 lượt / ngày (1 video/lượt)
+                "daily_batches": 0,
                 "cooldown_until": 0
             }
             
@@ -279,7 +290,7 @@ def check_generation_limits(key_input: str, device_id: str, is_vip: bool) -> tup
             return False, f"⏱️ Free Tier Cooldown: Please wait {wait_hours + 1} hours before generating again, or upgrade to VIP Premium!"
             
         if free_data.get("daily_batches", 0) >= 3:
-            return False, "❌ Free Tier Exhausted! Maximum 3 daily videos reached. Upgrade to VIP to continue rendering immediately!"
+            return False, "❌ Free Tier Exhausted! Maximum 3 daily videos reached."
             
         remaining_free = 3 - free_data.get("daily_batches", 0)
         return True, {
@@ -290,26 +301,22 @@ def check_generation_limits(key_input: str, device_id: str, is_vip: bool) -> tup
         }
 
 def commit_generation_success(key_input: str, device_id: str, is_vip: bool):
-    """
-    HÀM GHI NHẬN HẠN MỨC: Chỉ kích hoạt ghi đè khi video đã thành công và xuất xưởng 100%.
-    Nếu tài khoản Free bị VIP đẩy tiến trình giữa chừng -> Không chạy qua hàm này -> Bảo toàn lượt tạo.
-    """
     data = _load_storage()
     now_ts = time.time()
     
     if is_vip:
-        vip_data = data["keys"][key_input]
-        vip_data["videos_this_batch"] = vip_data.get("videos_this_batch", 0) + 1
-        
-        # Nếu đã render đủ 2 video cho lượt tạo này của VIP
-        if vip_data["videos_this_batch"] >= 2:
-            vip_data["daily_batches"] = vip_data.get("daily_batches", 0) + 1
-            vip_data["videos_this_batch"] = 0 # Reset đếm video lượt mới
-            vip_data["cooldown_until"] = now_ts + 3600 # Bắt đầu tính hàng chờ khóa 1 tiếng
+        if key_input in data["keys"]:
+            vip_data = data["keys"][key_input]
+            vip_data["videos_this_batch"] = vip_data.get("videos_this_batch", 0) + 1
+            if vip_data["videos_this_batch"] >= 2:
+                vip_data["daily_batches"] = vip_data.get("daily_batches", 0) + 1
+                vip_data["videos_this_batch"] = 0 
+                vip_data["cooldown_until"] = now_ts + 3600
     else:
-        free_data = data["free_devices"][device_id]
-        free_data["daily_batches"] = free_data.get("daily_batches", 0) + 1
-        free_data["cooldown_until"] = now_ts + (3 * 3600) # Khóa hàng chờ chặn tạo 3 tiếng đối với Free
+        if device_id in data["free_devices"]:
+            free_data = data["free_devices"][device_id]
+            free_data["daily_batches"] = free_data.get("daily_batches", 0) + 1
+            free_data["cooldown_until"] = now_ts + (3 * 3600)
         
     _save_storage(data)
 
@@ -320,22 +327,16 @@ def commit_generation_success(key_input: str, device_id: str, is_vip: bool):
 def execute_admin_diagnostic_test() -> str:
     """Khi Admin nhấn nút Test trên run_app.py, hàm này gọi ngầm file tester.py để hứng dữ liệu báo cáo trả lên UI"""
     test_script = "tester.py"
-    
-    # Kiểm tra sự tồn tại của file kiểm thử trên phân vùng hệ thống
     if not os.path.exists(test_script):
         return f"❌ FILE MISSING: Tệp '{test_script}' không tồn tại trong thư mục chạy hệ thống!"
             
     try:
-        # Gọi tệp kiểm thử độc lập chạy ngầm thông qua subprocess hệ thống
         result = subprocess.run(["python", test_script], capture_output=True, text=True, timeout=15)
-        
-        # Hứng toàn bộ kết quả đầu ra (stdout) hoặc thông báo lỗi (stderr) để bắn về khung thông báo run_app
         if result.returncode == 0:
             return f"✅ [TESTER REPORT SUCCESS]:\n{result.stdout.strip()}"
         else:
             return f"❌ [TESTER REPORT CRASHED WITH EXIT CODE {result.returncode}]:\n{result.stderr.strip()}"
     except subprocess.TimeoutExpired:
-        return "⚠️ [TESTER TIMEOUT]: Tiến trình kiểm thử của file tester.py chạy quá lâu (vượt ngưỡng 15 giây)!"
+        return "⚠️ [TESTER TIMEOUT]: Tiến trình kiểm thử vượt ngưỡng 15 giây!"
     except Exception as e:
         return f"❌ [SYSTEM CRASH]: Không thể kích hoạt được tiến trình kiểm thử. Lỗi: {str(e)}"
-