vrn123/workplace-decision-engine

1from env import WorkplaceEnv2from models import Action3 4 5def choose_action(task):6    desc = task.description.lower()7 8    if "refund" in desc or "login" in desc:9        return Action(action_type="complete", task_id=task.id)10 11    elif "meeting" in desc:12        return Action(action_type="schedule", task_id=task.id)13 14    else:15        return Action(action_type="respond", task_id=task.id)16 17 18def main():19    env = WorkplaceEnv()20    obs = env.reset()21 22    print("START")23 24    done = False25 26    while not done:27        for task in obs.tasks:28            if task.status == "completed":29                continue30 31            action = choose_action(task)32 33            obs, reward, done, info = env.step(action)34 35            print(36                f"STEP | action={action.action_type} | reward={reward:.2f} | score={info['score']:.2f}"37            )38 39            if done:40                break41 42    print("END")43    print(f"FINAL_SCORE: {info['score']:.2f}")44 45 46# ✅ VERY IMPORTANT (required)47if __name__ == "__main__":48    main()