import random


class MLPlay:
    def __init__(self, *args, **kwargs):
        print("Initial ml script")
        self.count=0

    def update(self, scene_info: dict, *args, **kwargs):
        """
        Generate the command according to the received scene information
        """
        # print("AI received data from game :", json.dumps(scene_info))
        # print(scene_info)
        actions = ["UP", "DOWN", "LEFT", "RIGHT"]
        self.count+=1
        if self.count >100:
            a = actions[100]
        return random.sample(actions, 1)

    def reset(self):
        """
        Reset the status
        """
        print("reset ml script")
        pass