import random class WorkspaceEnv: def __init__(self): self.state = 'cluttered' self.actions = ['organize', 'clean', 'sort', 'relax'] def reset(self): self.state = 'cluttered' return self.state def step(self, action): if action == 'organize': self.state = 'organized' reward = 1.0 done = True else: reward = -0.1 done = False return self.state, reward, done