Update app.py

app.py

@@ -13,23 +13,13 @@ ZGM_API_TOKEN = os.getenv("ZGM_API_TOKEN")
 if not ZGM_API_TOKEN:
     raise ValueError("ZGM_API_TOKEN không được tìm thấy trong biến môi trường!")
 
-# Tải danh sách key từ nhiều biến môi trường
-VALID_KEYS = set()
-for i in range(1, 11):  # Giả sử chia thành 10 phần, mỗi phần 1,000 key
-    key_part = os.getenv(f"ZGM_VALID_KEYS_{i}")
-    if key_part:
-        try:
-            VALID_KEYS.update(json.loads(key_part))
-        except json.JSONDecodeError:
-            raise ValueError(f"ZGM_VALID_KEYS_{i} không phải là chuỗi JSON hợp lệ!")
-if not VALID_KEYS:
-    raise ValueError("Không tìm thấy key nào trong biến môi trường!")
+# Lấy địa chỉ IP của VPS từ Secrets
+VPS_IP = os.getenv("VPS_IP")
+if not VPS_IP:
+    raise ValueError("VPS_IP không được tìm thấy trong biến môi trường!")
 
-# Danh sách dịch vụ cố định
-SERVICES = [
-    "ICLOUD", "ZSK", "FACEBOOK", "TELEGRAM", "OTHERS",
-    "GGPLAY", "INSTAGRAM", "FMCPAY", "AMAZON"
-]
+# URL của API trên VPS (dùng port 3000)
+KEY_API_URL = f"http://{VPS_IP}:3000"
 
 class ZGMApiClient:
     """Lớp client để tương tác với ZGM API"""
@@ -81,33 +71,33 @@ class ZGMApiClient:
 # Khởi tạo client
 client = ZGMApiClient("https://gm.haivl.wtf", ZGM_API_TOKEN)
 
-# Hàm xác thực và xóa key (chỉ xóa khi thuê thành công)
+# Hàm lấy key từ API trên VPS
+def get_key_from_api() -> tuple[Optional[str], str]:
+    try:
+        response = requests.get(f"{KEY_API_URL}/api/keys/get", timeout=5)
+        response.raise_for_status()
+        data = response.json()
+        return data["key"], f"Còn lại {data['remaining']} key."
+    except requests.exceptions.RequestException as e:
+        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]:
-    if len(VALID_KEYS) == 0:
-        return False, "Hết key! Vui lòng liên hệ quản trị viên để tạo thêm key."
-    
-    if access_key not in VALID_KEYS:
+    # 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()
+    if not key_from_api:
+        return False, message
+    if access_key != key_from_api:
         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 khi API trả về code 200 và có dữ liệu hợp lệ
+    # 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"]:
-        VALID_KEYS.remove(access_key)
-        # Cập nhật lại các biến môi trường (chia nhỏ để tránh vượt giới hạn)
-        keys_list = list(VALID_KEYS)
-        chunk_size = 1000  # Mỗi phần 1,000 key
-        for i in range(1, 11):
-            start = (i - 1) * chunk_size
-            end = i * chunk_size
-            chunk = keys_list[start:end]
-            if chunk:
-                os.environ[f"ZGM_VALID_KEYS_{i}"] = json.dumps(chunk)
-            else:
-                os.environ.pop(f"ZGM_VALID_KEYS_{i}", None)
-        remaining_keys = len(VALID_KEYS)
-        warning = f"\nCòn lại {remaining_keys} key."
-        if remaining_keys < 100:
-            warning += " Số lượng key còn ít! Vui lòng tạo thêm key mới."
-        return True, warning
+        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."
 
@@ -219,6 +209,12 @@ def gradio_interface(access_key, action, service, order_id, rent_otp_log, get_ot
     else:
         return rent_otp_log, get_otp_log, "Hành động không hợp lệ!", order_id
 
+# Danh sách dịch vụ cố định
+SERVICES = [
+    "ICLOUD", "ZSK", "FACEBOOK", "TELEGRAM", "OTHERS",
+    "GGPLAY", "INSTAGRAM", "FMCPAY", "AMAZON"
+]
+
 # Tạo giao diện Gradio
 with gr.Blocks() as demo:
     gr.Markdown("## ZGM API Client - Nhập Key để Thuê OTP (Key chỉ cần thiết cho Thuê OTP, mỗi key dùng 1 lần)")
@@ -271,4 +267,4 @@ with gr.Blocks() as demo:
     )
 
 # Khởi chạy giao diện
-demo.launch()
+demo.launch()