from untitled import generation, ui, rules, model


def adoption():
    print("Welcome to the shelter!")

    while True:
        choice = "Reroll"
        while choice == "Reroll":
            choices = generation.generate_cat_choices()
            choice = ui.select(
                "Please choose a cat to adopt:",
                [ui.Choice(cat[0], cat[1]) for cat in choices] + ["Reroll"],
            )

        while True:
            name = ui.text("Please choose a name for your cat:")
            error = rules.validate_cat_name(name)
            if not error:
                break
            print(error)
        if ui.confirm(
            f"Do you want to adopt {name}, {generation.generate_trait_sentence(choice).lower()}?"
        ):
            break

    return model.Cat(name, choice)