Update dependencies and refactor DesktopEnv initialization

- Removed specific versioning for the 'requests' library in requirements.txt and setup.py to allow for more flexible updates.
- Refactored the DesktopEnv class to streamline the emulator initialization process, enhancing error handling and logging during startup.
- Improved retry logic for file uploads in SetupController, ensuring robust handling of network issues and providing clearer error messages.
- Maintained existing code logic while enhancing clarity and reliability in the DesktopEnv and SetupController classes.

desktop_env/desktop_env.py

@@ -172,54 +172,52 @@ class DesktopEnv(gym.Env):
         else:
             self.path_to_vm = self.manager.get_vm_path(os_type=self.os_type, region=region, screen_size=(self.screen_width, self.screen_height))
         
-        try:
-            self.snapshot_name = snapshot_name
-            self.cache_dir_base: str = cache_dir
-            # todo: add the logic to get the screen size from the VM
-            self.headless = headless
-            self.require_a11y_tree = require_a11y_tree
-            self.require_terminal = require_terminal
+        self.snapshot_name = snapshot_name
+        self.cache_dir_base: str = cache_dir
+        # todo: add the logic to get the screen size from the VM
+        self.headless = headless
+        self.require_a11y_tree = require_a11y_tree
+        self.require_terminal = require_terminal
 
-            # Initialize emulator and controller
-            logger.info("Initializing...")
-            self._start_emulator()
+        # Initialize emulator and controller
+        logger.info("Initializing...")
+        self._start_emulator()
 
-            # mode: human or machine
-            self.instruction = None
-            assert action_space in ["computer_13", "pyautogui", "claude_computer_use", "autoglm_computer_use"]
-            self.action_space = action_space  # todo: refactor it to the ActType
+        # mode: human or machine
+        self.instruction = None
+        assert action_space in ["computer_13", "pyautogui", "claude_computer_use", "autoglm_computer_use"]
+        self.action_space = action_space  # todo: refactor it to the ActType
+
+        # episodic stuffs, like counters, will be updated or reset
+        # when calling self.reset()
+        self._traj_no: int = -1
+        self._step_no: int = 0
+        self.action_history: List[Dict[str, any]] = []
 
-            # episodic stuffs, like counters, will be updated or reset
-            # when calling self.reset()
-            self._traj_no: int = -1
-            self._step_no: int = 0
-            self.action_history: List[Dict[str, any]] = []
-        except Exception as e:
-            logger.error(f"Failed to initialize DesktopEnv: {e}")
-            # If initialization fails, we should clean up the VM
-            try:
-                self.close()
-                self.manager.delete_vm(self.path_to_vm, self.region)
-                logger.info(f"Cleaned up VM {self.path_to_vm}.")
-            except Exception as cleanup_error:
-                logger.error(f"Failed to clean up VM {self.path_to_vm}: {cleanup_error}")
-            raise
 
     def _start_emulator(self):
-        # Power on the virtual machine
-        self.provider.start_emulator(self.path_to_vm, self.headless, self.os_type)
+        try:
+            # Power on the virtual machine
+            self.provider.start_emulator(self.path_to_vm, self.headless, self.os_type)
 
-        # Get the ip from the virtual machine, and setup the controller
-        vm_ip_ports = self.provider.get_ip_address(self.path_to_vm).split(':')
-        self.vm_ip = vm_ip_ports[0]
-        # Get the ports from the virtual machine (for Docker provider only)
-        if len(vm_ip_ports) > 1:
-            self.server_port = int(vm_ip_ports[1])
-            self.chromium_port = int(vm_ip_ports[2])
-            self.vnc_port = int(vm_ip_ports[3])
-            self.vlc_port = int(vm_ip_ports[4])
-        self.controller = PythonController(vm_ip=self.vm_ip, server_port=self.server_port)
-        self.setup_controller = SetupController(vm_ip=self.vm_ip, server_port=self.server_port, chromium_port=self.chromium_port, vlc_port=self.vlc_port, cache_dir=self.cache_dir_base, client_password=self.client_password, screen_width=self.screen_width, screen_height=self.screen_height)
+            # Get the ip from the virtual machine, and setup the controller
+            vm_ip_ports = self.provider.get_ip_address(self.path_to_vm).split(':')
+            self.vm_ip = vm_ip_ports[0]
+            # Get the ports from the virtual machine (for Docker provider only)
+            if len(vm_ip_ports) > 1:
+                self.server_port = int(vm_ip_ports[1])
+                self.chromium_port = int(vm_ip_ports[2])
+                self.vnc_port = int(vm_ip_ports[3])
+                self.vlc_port = int(vm_ip_ports[4])
+            self.controller = PythonController(vm_ip=self.vm_ip, server_port=self.server_port)
+            self.setup_controller = SetupController(vm_ip=self.vm_ip, server_port=self.server_port, chromium_port=self.chromium_port, vlc_port=self.vlc_port, cache_dir=self.cache_dir_base, client_password=self.client_password, screen_width=self.screen_width, screen_height=self.screen_height)
+
+        except Exception as e:
+            try:
+                self.provider.stop_emulator(self.path_to_vm)
+            except Exception as stop_err:
+                logger.warning(f"Cleanup after interrupt failed: {stop_err}")
+            raise
 
     def _revert_to_snapshot(self):
         # Revert to certain snapshot of the virtual machine, and refresh the path to vm and ip of vm