exploration.py

import os
import json
import uuid
import argparse
import time as t
from datetime import datetime
import random

from config.config import Config
from agents.agent import Agent
from envs.desktop_env import DesktopEnv

def main():

    configs = Config.get_instance().config_data

    path = configs["LOG_ROOT"]

    related_apps = ["Cursor","ShareX","Zotero","Evernote","Windows PowerShell ISE"]

    for i in range(1000):

        related_app = random.choice(related_apps)
        try:
            print(f"trajectory {i+1} / {100} is running...")
            id = str(uuid.uuid4())
            if not os.path.exists(f"{configs['LOG_ROOT']}/{id}"):
                os.makedirs(f"{configs['LOG_ROOT']}/{id}")
            env = DesktopEnv(
                path_to_vm=configs["VM_PATH"],
                observation_space=["Screen_A11Y"], 
                action_space="pyautogui",
                snapshot_name="init_state",
                headless=True,
                uuid=id,
                related_app=related_app,
            )
            agent = Agent(
                observation_space=["Screen_A11Y"],
                action_space="pyautogui",
                uuid=id,
                related_app=related_app,
            )
            step = 0
            while True:
                observation = env.get_observation()
                action, status = agent.get_action(observation)
                env.execute_action(action)

                step += 1
                agent.step = step
                env.step = step

                if status == "FINISH" or step > configs["MAX_STEP"]:
                    break
        except Exception as e:
            print(e)

if __name__ == "__main__":
    main()