From 4ea6c0a7f6745d5149526ef4158993d00460e119 Mon Sep 17 00:00:00 2001 From: khuxkm Date: Fri, 3 Aug 2018 01:43:48 -0400 Subject: [PATCH] Add chain loader --- chain.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 chain.py diff --git a/chain.py b/chain.py new file mode 100644 index 0000000..8a9d0b6 --- /dev/null +++ b/chain.py @@ -0,0 +1,19 @@ +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))