import time
from kl_mcp_client.tools import MCPTools

def main():
    # 1. Khởi tạo và Kết nối Server
    tools = MCPTools()
    tools.connect_mcp("http://localhost:8080")
    print("🚀 Đã lập kết nối MCP")

    # 2. Tạo phiên hoạt động (Cần CDP URL)
    resp = tools.create_session("ws://localhost:9222")
    sid = resp["sessionId"]
    print(f"📦 Khởi tạo phiên: {sid}")

    # 3. Điều hướng đến Google
    tools.open_page(sid, "https://www.google.com")
    print("🌐 Đang truy cập Google...")

    # 4. Tìm ô tìm kiếm bằng AI và nhập từ khóa
    tools.find_element_by_prompt(sid, "ô tìm kiếm")
    tools.type(sid, "textarea[name='q']", "AI Browser Automation with MCP")
    tools.send_keys(sid, "Enter")
    print("⌨️ Đang tìm kiếm từ khóa...")

    # Đợi trang kết quả tải xong
    time.sleep(2)

    # 5. Lấy nội dung trang đã làm sạch để phân tích
    clean_text = tools.get_clean_text(sid)
    print(f"📄 Nội dung trang sơ bộ: {clean_text[:200]}...")

    # 6. Chụp ảnh kết quả
    tools.screenshot(sid)
    print("📸 Đã chụp ảnh màn hình kết quả")

    # 7. Giải phóng tài nguyên
    tools.close_session(sid)
    print("✅ Hoàn tất quy trình")

if __name__ == "__main__":
    main()
