auto name gen

testing/namegen.py

@@ -0,0 +1,71 @@
+# THIS WAS ADDED TO MAIN PROGRAM
+
+import csv
+import random
+from collections import defaultdict
+
+from untitled import rules
+
+NAME_CORPUS = []
+
+with open("nameset.csv", newline="", encoding="utf-8") as file:
+    reader = csv.reader(file)
+
+    header = next(reader)
+    for row in reader:
+        name = row[5].lower()
+        if (
+            name == "name not provided"
+            or name == "untitled"
+            or name == "unknown"
+            or name == "kitten"
+            or not name
+        ):
+            continue
+        if name.isalpha():
+            NAME_CORPUS.append(name)
+
+
+def pad(names):
+    new_names = []
+    for name in names:
+        new_names.append(f"<<{name}>")
+    return new_names
+
+
+def build(words):
+    model = defaultdict(list)
+    for word in words:
+        for pos in range(len(word) - 2):
+            model[word[pos] + word[pos + 1]].append(word[pos + 2])
+    return model
+
+
+def make_name(model):
+    result = "<<"
+    while True:
+        window = result[-2:]
+        nxt = random.choice(model[window])
+        if nxt == ">":
+            break
+        result += nxt
+    return result.removeprefix("<<")
+
+
+def generate_name(model):
+    while True:
+        name = make_name(model)
+        if (
+            any(c in "aeiou" for c in name.lower())
+            and len(name) <= 9
+            and rules.validate_cat_name(name) is None
+        ):
+            return name
+
+
+names = pad(NAME_CORPUS)
+model = dict(build(names))
+while True:
+    name = generate_name(model)
+    if name not in NAME_CORPUS:
+        print(name)