from untitled import generation, model, rules, ui


def adoption():
    print("Welcome to the shelter!")
    while True:
        auto_name = ""
        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 or type "idk" to autofill a generated one:',
                auto_name,
            )
            if name.lower() != "idk":
                auto_name = ""
                error = rules.validate_cat_name(name)
                if not error:
                    break
                print(error)
            else:
                auto_name = generation.cat_name_generator.generate_name().capitalize()

        if ui.confirm(
            f"Do you want to adopt {name}, {generation.generate_trait_sentence(choice).lower()}?"
        ):
            break

    return model.Cat(name, choice)