setup controller

desktop_env/envs/desktop_env.py

@@ -9,6 +9,7 @@ from typing import List
 import gymnasium as gym
 
 from desktop_env.controllers.python import PythonController
+from desktop_env.controllers.setup import SetupController
 
 
 def _execute_command(command: List[str]) -> None:
@@ -34,8 +35,9 @@ class DesktopEnv(gym.Env):
         # Initialize emulator and controller
         print("Initializing...")
         self._start_emulator()
-        self.host = self._get_vm_ip()
+        self.host = f"http://{self._get_vm_ip()}:5000"
         self.controller = PythonController(http_server=self.host)
+        self.setup_controller = SetupController(http_server=self.host)
 
         # mode: human or machine
         assert action_space in ["computer_13", "pyautogui"]
@@ -118,6 +120,9 @@ class DesktopEnv(gym.Env):
         done = False  # todo: Define episode termination condition for each example
         info = {}
         return observation, reward, done, info
+    
+    def setup(self, config: dict):
+        self.setup_controller.setup(config) 
 
     def render(self, mode='rgb_array'):
         if mode == 'rgb_array':