from numpy.lib.type_check import real


states = []
rewards= []

def bot1(states_rewards):
    def bot2(states_rewards):
        def get_key(val):
            for key, value in states_rewards.items():
                if val == value:
                    return key
        current_state = 0
        for x in states_rewards:
            states.append(x)
            rewards.append(states_rewards[x])
        for y in range(len(states)):
            current_state += 1
            current_reward = rewards[y]
            return get_key(max(rewards))
    return "The state I'm gonna go to: {}".format(bot2(states_rewards))

print(bot1({'s1': 1, 's2': 0, 's3': 0, 's4': 0, 's5': 0, 's6': 0, 's7': 10}))