@@ -35,6 +35,9 @@ class Markov:
if not words:
return self.generate()
+ if init_state:
+ words = list(init_state) + words
+
text = ""
for word in words:
if word in "-–—" or not all(c in string.punctuation for c in word):
