Create app.py

app.py

@@ -0,0 +1,274 @@
+import requests
+import json
+from uuid import UUID
+from typing import Dict, Any, Optional
+import gradio as gr
+import os
+import threading
+import time
+import pandas as pd
+
+# Lấy ZGM API Token từ biến môi trường (Secrets trên Hugging Face)
+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!")
+
+# Danh sách dịch vụ cố định
+SERVICES = [
+    "ICLOUD", "ZSK", "FACEBOOK", "TELEGRAM", "OTHERS",
+    "GGPLAY", "INSTAGRAM", "FMCPAY", "AMAZON"
+]
+
+class ZGMApiClient:
+    """Lớp client để tương tác với ZGM API"""
+    
+    def __init__(self, base_url: str, api_token: str):
+        self.base_url = base_url.rstrip('/')
+        self.api_token = api_token
+        self.headers = {
+            'ZGM-API-TOKEN': api_token,
+            'Content-Type': 'application/json',
+            'Accept': 'text/plain'
+        }
+
+    def _make_request(self, method: str, endpoint: str, params: Dict = None, data: Dict = None) -> Dict:
+        url = f"{self.base_url}{endpoint}"
+        try:
+            response = requests.request(
+                method=method,
+                url=url,
+                headers=self.headers,
+                params=params,
+                json=data,
+                timeout=15
+            )
+            response.raise_for_status()
+            return response.json()
+        except requests.exceptions.ConnectTimeout:
+            return {"error": f"Không thể kết nối đến {self.base_url}. Vui lòng kiểm tra URL hoặc mạng.", "status_code": 0}
+        except requests.exceptions.HTTPError as e:
+            if response.status_code == 401:
+                return {"error": "Chưa đăng nhập. Vui lòng kiểm tra API Token.", "status_code": 401}
+            elif response.status_code == 500:
+                return {"error": "Lỗi server từ ZGM API.", "status_code": 500}
+            return {"error": f"Lỗi HTTP: {str(e)}", "status_code": response.status_code}
+        except requests.exceptions.RequestException as e:
+            return {"error": f"Lỗi kết nối: {str(e)}", "status_code": 0}
+
+    def rent_otp(self, service: str) -> Dict:
+        data = {"service": service}
+        return self._make_request('POST', '/api/mail/otp-rent', data=data)
+
+    def get_otp(self, order_id: str) -> Dict:
+        try:
+            UUID(order_id)
+            return self._make_request('GET', '/api/mail/otp-get', params={"orderId": order_id})
+        except ValueError:
+            return {"error": "order_id phải là UUID hợp lệ", "status_code": 400}
+
+# 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)
+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:
+        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ệ
+    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
+    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):
+    result = [None]
+    is_loading = [True]
+
+    def target():
+        result[0] = func(*args, **kwargs)
+        is_loading[0] = False
+
+    api_thread = threading.Thread(target=target)
+    api_thread.start()
+
+    chars = "|/-\\"
+    i = 0
+    while is_loading[0]:
+        print(f"\rĐang tải {chars[i % len(chars)]}", end="")
+        i += 1
+        time.sleep(0.1)
+    print("\r", end="")
+
+    api_thread.join()
+    return result[0]
+
+# Hàm định dạng kết quả "Thuê OTP" thành bảng (chỉ Order ID và Email)
+def format_rent_otp_result(data: Dict) -> pd.DataFrame:
+    if not data or "data" not in data:
+        return pd.DataFrame([["Không có dữ liệu để hiển thị."]], columns=["Message"])
+    order_info = data["data"]
+    df = pd.DataFrame([{
+        "Order ID": order_info.get("orderId", "N/A"),
+        "Email": order_info.get("email", "N/A")
+    }])
+    return df
+
+# Hàm định dạng kết quả "Lấy OTP" thành bảng (Service, Email, OTP Code, và nội dung thư)
+def format_get_otp_result(data: Dict, selected_service: str) -> pd.DataFrame:
+    if not data or "data" not in data:
+        return pd.DataFrame([["Không có dữ liệu để hiển thị."]], columns=["Message"])
+    
+    otp_data = data["data"]
+    
+    # Nếu là đối tượng đơn (không phải danh sách)
+    if isinstance(otp_data, dict):
+        otp_data = [otp_data]  # Chuyển thành danh sách để xử lý thống nhất
+    
+    # Nếu không phải danh sách, dữ liệu không hợp lệ
+    if not isinstance(otp_data, list):
+        return pd.DataFrame([["Dữ liệu không hợp lệ."]], columns=["Message"])
+    
+    df = pd.DataFrame([
+        {
+            "Service": selected_service,  # Sử dụng dịch vụ đã chọn khi thuê
+            "Email": item.get("email", "N/A"),  # Thêm email vào log
+            "OTP Code": item.get("otpCode", "N/A"),
+            "Content": item.get("otpContent", "N/A")  # Nội dung thư từ API
+        }
+        for item in otp_data
+    ])
+    return df
+
+# Hàm xóa tất cả log
+def clear_all_logs():
+    return pd.DataFrame([["Chưa có dữ liệu"]], columns=["Message"]), pd.DataFrame([["Chưa có dữ liệu"]], columns=["Message"]), ""
+
+# Hàm giao diện Gradio
+def gradio_interface(access_key, action, service, order_id, rent_otp_log, get_otp_log):
+    message = ""
+    new_order_id = order_id  # Giữ nguyên order_id nếu không có thay đổi
+    
+    # Khởi tạo log nếu chưa có
+    if rent_otp_log is None:
+        rent_otp_log = pd.DataFrame([["Chưa có dữ liệu"]], columns=["Message"])
+    if get_otp_log is None:
+        get_otp_log = pd.DataFrame([["Chưa có dữ liệu"]], columns=["Message"])
+
+    # Chỉ yêu cầu key cho hành động "Thuê OTP"
+    if action == "Thuê OTP":
+        if not access_key:
+            rent_otp_log = pd.DataFrame([["Vui lòng nhập key để thực hiện Thuê OTP!"]], columns=["Message"])
+            return rent_otp_log, get_otp_log, message, order_id
+        if not service:
+            rent_otp_log = pd.DataFrame([["Vui lòng chọn dịch vụ!"]], columns=["Message"])
+            return rent_otp_log, get_otp_log, message, order_id
+        result = execute_with_loading(client.rent_otp, service)
+        is_valid, message = validate_and_remove_key(access_key, result)
+        if not is_valid:
+            rent_otp_log = pd.DataFrame([[message]], columns=["Message"])
+            return rent_otp_log, get_otp_log, message, order_id
+        if result.get("code") == 200:
+            rent_otp_log = format_rent_otp_result(result)
+            # Lấy Order ID mới để tự động điền
+            new_order_id = result["data"].get("orderId", order_id)
+        else:
+            rent_otp_log = pd.DataFrame([[json.dumps(result, indent=2, ensure_ascii=False)]], columns=["Message"])
+        return rent_otp_log, get_otp_log, message, new_order_id
+    elif action == "Lấy OTP":
+        if not order_id:
+            get_otp_log = pd.DataFrame([["Vui lòng nhập Order ID!"]], columns=["Message"])
+            return rent_otp_log, get_otp_log, "", order_id
+        result = execute_with_loading(client.get_otp, order_id)
+        if result.get("code") == 200 and "data" in result:
+            get_otp_log = format_get_otp_result(result, service)  # Truyền service đã chọn
+        else:
+            get_otp_log = pd.DataFrame([[json.dumps(result, indent=2, ensure_ascii=False)]], columns=["Message"])
+        return rent_otp_log, get_otp_log, "", order_id
+    else:
+        return rent_otp_log, get_otp_log, "Hành động không hợp lệ!", order_id
+
+# 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)")
+    
+    with gr.Row():
+        access_key = gr.Textbox(label="Nhập Key Truy Cập (Chỉ cần cho Thuê OTP)", placeholder="Nhập key 32 ký tự...")
+        action = gr.Dropdown(
+            label="Chọn Hành Động",
+            choices=["Thuê OTP", "Lấy OTP"],
+            value="Thuê OTP"
+        )
+    
+    with gr.Row():
+        service = gr.Dropdown(label="Dịch Vụ (Cho Thuê OTP)", choices=SERVICES)
+        order_id = gr.Textbox(label="Order ID (Cho Lấy OTP)", placeholder="Nhập Order ID")
+    
+    rent_otp_log = gr.State(value=pd.DataFrame([["Chưa có dữ liệu"]], columns=["Message"]))
+    get_otp_log = gr.State(value=pd.DataFrame([["Chưa có dữ liệu"]], columns=["Message"]))
+    
+    with gr.Row():
+        rent_otp_output = gr.Dataframe(
+            label="Log Thuê OTP",
+            headers=["Order ID", "Email"],
+            datatype=["str", "str"],
+            interactive=False
+        )
+        get_otp_output = gr.Dataframe(
+            label="Log Lấy OTP",
+            headers=["Service", "Email", "OTP Code", "Content"],
+            datatype=["str", "str", "str", "str"],
+            interactive=False
+        )
+    
+    message_output = gr.Textbox(label="Thông Báo", interactive=False)
+    
+    with gr.Row():
+        submit_btn = gr.Button("Thực Thi", variant="primary")
+        clear_btn = gr.Button("Xóa Log", variant="secondary")
+    
+    submit_btn.click(
+        fn=gradio_interface,
+        inputs=[access_key, action, service, order_id, rent_otp_log, get_otp_log],
+        outputs=[rent_otp_output, get_otp_output, message_output, order_id]
+    )
+    
+    clear_btn.click(
+        fn=clear_all_logs,
+        inputs=None,
+        outputs=[rent_otp_output, get_otp_output, message_output]
+    )
+
+# Khởi chạy giao diện
+demo.launch()