| | 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, |
| | }, |
| | ) |
