mirror of
https://github.com/trycua/computer.git
synced 2026-01-02 03:20:22 -06:00
58 lines
1.5 KiB
Python
Executable File
58 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Setup script for CUA Desktop Extension
|
|
Installs required dependencies if not available
|
|
"""
|
|
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
def check_and_install_package(package_name, import_name=None):
|
|
"""Check if a package is available, install if not."""
|
|
if import_name is None:
|
|
import_name = package_name
|
|
|
|
try:
|
|
__import__(import_name)
|
|
print(f"✓ {package_name} is available")
|
|
return True
|
|
except ImportError:
|
|
print(f"⚠ {package_name} not found, installing...")
|
|
try:
|
|
subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
|
|
print(f"✓ {package_name} installed successfully")
|
|
return True
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"✗ Failed to install {package_name}: {e}")
|
|
return False
|
|
|
|
|
|
def main():
|
|
"""Install required packages."""
|
|
print("Setting up CUA Desktop Extension dependencies...")
|
|
|
|
# Required packages
|
|
packages = [
|
|
("mcp", "mcp"),
|
|
("anyio", "anyio"),
|
|
("cua-agent[all]", "agent"),
|
|
("cua-computer", "computer"),
|
|
]
|
|
|
|
all_installed = True
|
|
for package, import_name in packages:
|
|
if not check_and_install_package(package, import_name):
|
|
all_installed = False
|
|
|
|
if all_installed:
|
|
print("✓ All dependencies are ready!")
|
|
return 0
|
|
else:
|
|
print("✗ Some dependencies failed to install")
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|