| import subprocess |
| import sys |
|
|
| from setuptools import setup, find_packages |
| from setuptools.command.install import install |
|
|
|
|
| class InstallPlaywrightCommand(install): |
| """Customized setuptools install command that runs 'playwright install'.""" |
|
|
| def run(self): |
| |
| install.run(self) |
|
|
| |
| try: |
| subprocess.check_call([sys.executable, "-m", "playwright", "install"]) |
| print("Successfully ran 'playwright install'.") |
| except subprocess.CalledProcessError as e: |
| print("Failed to run 'playwright install'. Please run 'playwright install' manually.") |
| print(e) |
|
|
|
|
| setup( |
| name="desktop_env", |
| version="1.0.1", |
| author="Tianbao Xie, Danyang Zhang, Jixuan Chen, Xiaochuan Li, Siheng Zhao, Ruisheng Cao, Toh Jing Hua, etc.", |
| author_email="tianbaoxiexxx@gmail.com", |
| description="The package provides a desktop environment for setting and evaluating desktop automation tasks.", |
| long_description=open('README.md', encoding="utf-8").read(), |
| long_description_content_type="text/markdown", |
| url="https://github.com/xlang-ai/desktop_env", |
| packages=find_packages(), |
| classifiers=[ |
| "Programming Language :: Python :: 3", |
| "License :: OSI Approved :: Apache Software License", |
| "Operating System :: OS Independent", |
| ], |
| python_requires='>=3.10', |
| install_requires=[ |
| "numpy>=1.26,<3", |
| "Pillow~=11.0.0", |
| "fabric", |
| "gymnasium~=0.28.1", |
| "requests", |
| "pytz~=2024.1", |
| "transformers~=4.35.2", |
| "torch~=2.5.0", |
| "accelerate", |
| "opencv-python-headless~=4.8.1.78", |
| "matplotlib~=3.7.4", |
| "pynput~=1.7.6", |
| "pyautogui~=0.9.54", |
| "psutil~=5.9.6", |
| "tqdm~=4.65.0", |
| "pandas>=2.2,<2.3", |
| "flask~=3.0.0", |
| "requests-toolbelt~=1.0.0", |
| "ag2~=0.9.7", |
| "lxml", |
| "cssselect", |
| "xmltodict", |
| "openpyxl", |
| "python-docx", |
| "python-pptx", |
| "pypdf", |
| "PyGetWindow", |
| "rapidfuzz", |
| "pyacoustid", |
| "pygame", |
| "opencv-python-headless", |
| "ImageHash", |
| "scikit-image", |
| "librosa", |
| "pymupdf", |
| "chardet", |
| "playwright", |
| "backoff", |
| "formulas", |
| "pydrive", |
| "fastdtw", |
| "odfpy", |
| "openai", |
| "func-timeout", |
| "beautifulsoup4", |
| "dashscope", |
| "google-generativeai", |
| "PyYaml", |
| "mutagen", |
| "easyocr", |
| "borb", |
| "pypdf2", |
| "pdfplumber", |
| "wandb", |
| "wrapt_timeout_decorator", |
| "gdown", |
| "tiktoken", |
| "groq", |
| "boto3", |
| "azure-identity", |
| "azure-mgmt-compute", |
| "azure-mgmt-network", |
| "docker", |
| "loguru", |
| "dotenv", |
| "tldextract", |
| "anthropic", |
| |
| "alibabacloud_ecs20140526", |
| "alibabacloud_tea_openapi", |
| "alibabacloud_tea_util", |
| ], |
| cmdclass={ |
| 'install': InstallPlaywrightCommand, |
| }, |
| ) |
