stuff

untitled/screens/adoption.py

@@ -1,7 +1,12 @@
-from untitled import generation, model, rules, ui
+from untitled import PACKAGE_ROOT, generation, model, namegen, rules, ui
 
 
 def adoption():
+    names = []
+    with open(PACKAGE_ROOT / "assets" / "cat_names.txt") as f:
+        for i in f.readlines():
+            names.append(i.strip())
+    name_generator = namegen.NameGenerator(names, rules.validate_cat_name)
     print("Welcome to the shelter!")
     while True:
         auto_name = ""
@@ -25,7 +30,7 @@ def adoption():
                     break
                 print(error)
             else:
-                auto_name = generation.generate_name()
+                auto_name = name_generator.generate_name().capitalize()
 
         if ui.confirm(
             f"Do you want to adopt {name}, {generation.generate_trait_sentence(choice).lower()}?"