Update app.py

app.py

@@ -75,43 +75,30 @@ 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
+        print(f"API Response: {response.text}")  # Debug log
         response.raise_for_status()
         data = response.json()
-        print(f"Extracted Key: {data['key']}")  # Thêm log debug
+        print(f"Extracted Key: {data['key']}")  # Debug log
         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
+        print(f"API Error: {str(e)}")  # Debug log
         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)
+# Hàm xác thực và xóa key (gọi API một lần)
 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
+    key_from_api, message = get_key_from_api()  # Gọi API một lần
+    print(f"Input Key: {access_key}, API Key: {key_from_api}")  # Debug log
     if not key_from_api:
         return False, message
-    if access_key != key_from_api:
+    if access_key.strip() != key_from_api.strip():  # So sánh sau khi loại bỏ whitespace
         return False, "Key không hợp lệ! Vui lòng nhập key hợp lệ để tiếp tục."
     
     # 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."
-    
-    # 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."
+        remaining_keys = int(message.split("Còn lại ")[1].split(" key")[0])
+        remaining_message = f"Còn lại {remaining_keys} key."
+        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."