Квест на чистом python. Переключение сцен. Как сделать?

[Lektorfuja]

Две сцены :castle и forest
Сначала мы в castle . можно написать "to forest" и будет надпись :vi v lesu. Vvedi castle. Тогда мы вводим "to castle" и будет надпись:"ti v zamke". Но как реализовать это в коде?
P.s. если я не ошибаюсь квесты можно даже делать в powerpoint.методом гиперссылок

[BDA]

def main():
    place = "castle"
    while True:
        direction = input()
        if place == "castle":
            if direction == "to forest":
                print('In forest. Input "to castle".') 
                place = "forest"
            else:
                raise RuntimeError("Unknown direction")
        elif place == "forest":
            if direction == "to castle":
                print('In castle. Input "to forest".') 
                place = "castle"
            else:
                raise RuntimeError("Unknown direction")
        else:
            raise RuntimeError("Unknown place")

main()

[Lektorfuja]

while True:
  a=input("1 forest 2 castle");
  if a=="1": print ("ti v lesu. 2 v zamok");
  if a=="2": print ("ti v zamke. 1 v les");

Отчасти верный ответ