def load(self, personality, cacheDir):
"""Load a personality into memory by loading its Markov chains"""
self.brains[personality] = dict()
self.brains[personality]["fwd"] = Markov.MarkovChain(
cacheDir + personality + "-fwd", 1, False)
self.brains[personality]["rev"] = Markov.MarkovChain(
cacheDir + personality + "-rev", 1, False)
def makeChain(name, sourceDir, cacheDir):
"""Parse the source text to generate new Markov chains."""
sourceFile = sourceDir + name + ".txt"
fwd = Markov.MarkovChain(cacheDir + name + "-fwd", 1, True)
rev = Markov.MarkovChain(cacheDir + name + "-rev", 1, True)
try:
for line in open(sourceFile):
words = line.split()
if len(words) > 0:
fwd.add(words)
words.reverse()
rev.add(words)
except IOError:
print "The source named '%s' does not exist" % (name, )
finally:
fwd.close()
rev.close()
def load(self, brain):
self.brain = Markov.MarkovChain(self.cache + brain + "-fwd", 1, False)
