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)