Instantly share code, notes, and snippets.

Embed
What would you like to do?
Markov Chain message generator
import random
import sys
def build_chain(text, chain = {}):
words = text.split(' ')
index = 1
for word in words[index:]:
key = words[index - 1]
if key in chain:
chain[key].append(word)
else:
chain[key] = [word]
index += 1
return chain
def generate_message(chain, count = 100):
word1 = random.choice(list(chain.keys()))
message = word1.capitalize()
while len(message.split(' ')) < count:
word2 = random.choice(chain[word1])
word1 = word2
message += ' ' + word2
return message
def read_file(filename):
with open(filename, "r") as file:
contents = file.read().replace('\n\n',' ')
return contents
def write_file(filename, message):
with open(filename, "w") as file:
file.write(message)
if __name__ == '__main__':
message = read_file(sys.argv[1])
chain = build_chain(message)
message = generate_message(chain)
write_file("output.txt", message)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment