Update app.py

app.py

@@ -75,16 +75,20 @@ client = ZGMApiClient("https://gm.haivl.wtf", ZGM_API_TOKEN)
 def get_key_from_api() -> tuple[Optional[str], str]:
     try:
         response = requests.get(f"{KEY_API_URL}/api/keys/get", timeout=5)
+        print(f"API Response: {response.text}")  # Thêm log debug
         response.raise_for_status()
         data = response.json()
+        print(f"Extracted Key: {data['key']}")  # Thêm log debug
         return data["key"], f"Còn lại {data['remaining']} key."
     except requests.exceptions.RequestException as e:
+        print(f"API Error: {str(e)}")  # Thêm log debug
         return None, f"Lỗi khi lấy key từ API: {str(e)}"
 
 # Hàm xác thực và xóa key (gọi API trên VPS)
 def validate_and_remove_key(access_key: str, result: Dict) -> tuple[bool, str]:
     # Kiểm tra key có hợp lệ không (gọi API để lấy key và so sánh)
     key_from_api, message = get_key_from_api()
+    print(f"Input Key: {access_key}, API Key: {key_from_api}")  # Thêm log debug
     if not key_from_api:
         return False, message
     if access_key != key_from_api:
@@ -100,6 +104,17 @@ def validate_and_remove_key(access_key: str, result: Dict) -> tuple[bool, str]:
         return True, remaining_message
     else:
         return True, f"Lỗi: {result.get('message', 'Không thể thuê Gmail.')}\nKey không bị xóa."
+    
+    # Chỉ xóa key trên VPS khi thuê thành công (đã gọi API xóa trong get_key_from_api)
+    if result.get("code") == 200 and result.get("data") and "orderId" in result["data"] and "email" in result["data"]:
+        remaining_message = message
+        if "Còn lại" in message:
+            remaining_keys = int(message.split("Còn lại ")[1].split(" key")[0])
+            if remaining_keys < 100:
+                remaining_message += " Số lượng key còn ít! Vui lòng tạo thêm key mới."
+        return True, remaining_message
+    else:
+        return True, f"Lỗi: {result.get('message', 'Không thể thuê Gmail.')}\nKey không bị xóa."
 
 # Hàm xử lý bất đồng bộ với chỉ báo tải
 def execute_with_loading(func, *args, **kwargs):