episodegen/chain.py

20 lines
547 B
Python

import markovify,os
# on by default, if it lags turn it off
RETAIN_ORIGINAL = True
combined_model = None
for corpus in os.listdir("corpora"):
with open(os.path.join("corpora",corpus)) as f:
model = markovify.Text(f.read(),retain_original=RETAIN_ORIGINAL)
if combined_model:
combined_model = markovify.combine(models=[combined_model,model])
else:
combined_model = model
def make_sentence(*args,**kwargs):
return combined_model.make_sentence(*args,**kwargs)
if __name__=="__main__":
print(combined_model.make_sentence(tries=100))