class BasicLevelTranslator:
    def __init__(self):
        pass

    def translate(self, state):
        car_position, car_velocity = state
        car_direction = "right" if car_velocity > 0 else "left"
        res = (f"The car is positioned at {car_position:.3f}, with a velocity of {abs(car_velocity):.3f} towards the {car_direction}.")
       
        return res

class GameDescriber:
    def __init__(self, args):
        self.is_only_local_obs = args.is_only_local_obs == 1
        self.max_episode_len = args.max_episode_len
        self.action_desc_dict = {
        }
        self.reward_desc_dict = {
        }

    def describe_goal(self):
        return "The goal is to reach the flag placed on top of the right hill as quickly as possible."

    def translate_terminate_state(self, state, episode_len, max_episode_len): 
        return ""
    
    def translate_potential_next_state(self, state, action):
        return ""
        
    def describe_game(self):
        return ("In the Mountain Car game, you control a car placed stochastically at the bottom of a sinusoidal valley. "
                "The only possible actions are the accelerations that can be applied to the car in either direction. "
                "The goal of the game is to strategically accelerate the car to reach the goal state on top of the right hill "
                "as quickly as possible. The episode ends if either the car reaches the goal position on top of the right hill "
                f"or the length of the episode is {self.max_episode_len}.")

    def describe_action(self):
        return ("Your Next Move:" 
                "\n Please choose an action. Type '1' to accelerate to the left, '2' to not accelerate, or '3' to accelerate to the right."
                "Ensure you only provide the action number from the valid action list, i.e., [1, 2, 3].")

class BasicStateSequenceTranslator(BasicLevelTranslator):
    def translate(self, infos, is_current=False):
        descriptions = []
        if is_current:
            state_desc = BasicLevelTranslator().translate(infos[-1]['state'])
            return state_desc
        for i, info in enumerate(infos):
            assert 'state' in info, "info should contain state information"

            state_desc = BasicLevelTranslator().translate(info['state'])
            action_desc = f"Take Action: {'Accelerate to the left' if info['action'] == 1 else ('Don’t accelerate' if info['action'] == 2 else 'Accelerate to the right')} ({info['action']})."
            reward_desc = f"Result: Reward of {info['reward']}, "
            next_state_desc = BasicLevelTranslator().translate(info['next_state'])
            descriptions.append(f"{state_desc}.\n {action_desc} \n {reward_desc} \n Transit to {next_state_desc}")
        return descriptions