# define environment and agent
receptacles = ['diningtable 1','drawer 2', 'drawer 1', 'sinkbasin 1', 'toilet 1', 'sidetable 2', 'sidetable 1', 'cabinet 1', 'countertop 1', 'microwave 1', 'fridge 1']
agent = Agent(receptacles)

# Your task is to: put soapbar on countertop.
# here is a solution:
def solution(agent, start_from=1):
    # General Plan: I need to get a list of receptacles where the soapbar is likely to appear, and then go to each receptacle in the list until seeing a soapbar. Then I can put get the identifier of the soapbar and take it. Finally I can go to the countertop and put the soapbar.
    if start_from <= 1:
        print("[Step 1] get a list of receptacles where the soapbar is likely to appear")
        # I can ask the assistant to do that.
        answer = ask(f'Given a list of receptacles, please sort them in descending order based on the likelihood of finding a soapbar in each of them. The list of receptacles is: {agent.receptacles}. You should directly return a Python list.')
        recep_to_check = literal_eval(answer)
        # expectation: the returned recep_to_check should not be empty.
        assert recep_to_check, f'Error in [Step 1]: recep_to_check should not be empty. {agent.report()}'

    if start_from <= 2:
        print("[Step 2] go to each receptacle in the list until seeing a soapbar")
        for receptacle in recep_to_check:
            observation = agent.goto(receptacle)
            # check if the receptacle is closed. If so, open it.
            if 'closed' in observation:
                observation = agent.open_receptacle(receptacle)
            # check if a soapbar is in/on the receptacle.
            if 'soapbar' in observation:
                break
        # expectation: I should be able to find a receptacle where a soapbar is in/on it.
        assert 'soapbar' in observation, f'Error in [Step 2]: There is no soapbar in/on {recep_to_check}. {agent.report()}'

    if start_from <= 3:
        print("[Step 3] identify the soapbar I juts found and take it")
        # I need to get the identifier of the soapbar. I can ask the assistant to do that.
        answer = ask(f'From the observation, get the identifier of an object. For example, On the cabinet 1, you see a cloth 2, and a toiletpaper 2. The identifier of cloth is 2. Now, {observation} The identifier of the soap? Only Output a single number without any other words. ')
        found_soapbar = f'soapbar {answer}'
        observation = agent.take(found_soapbar, receptacle)
        # expectation: I should be able to take the soapbar from the receptacle.
        assert agent.holding == found_soapbar, f'Error in [Step 3]: I cannot take {found_soapbar} from the {receptacle}. {agent.report()}'

    if start_from <= 4:
        print("[Step 4] go to a countertop and put the soapbar on it")
        # There are multiple countertops, and I only need to go to one of them.
        observation = agent.goto('countertop 1')
        # check if the countertop is closed. If so, open it.
        if 'closed' in observation:
            observation = agent.open_receptacle('countertop 1')
        observation = agent.put(found_soapbar, 'countertop 1')
        # expectation: I should be able to put the soapbar on the countertop.
        assert f'You put the {found_soapbar} in/on the countertop 1.' in observation, f'Error in [Step 4]: I cannot put the {found_soapbar} on the countertop 1. {agent.report()}'