Markov chain lorem ipsum text

gutenberg_ipsum.py

#!/usr/bin/env python2

# Load libraries
import markovgen
import argparse
import re

# Get command line arguments
parser = argparse.ArgumentParser(description='Generate random text with Markov chains!')
parser.add_argument('corpus', type=argparse.FileType('r+'),
                    help='corpus of training text')
parser.add_argument('num_words', type=int, default=100, nargs='?',
                    help='the number of words to output')
args = parser.parse_args()

# Save arguments
corpus = args.corpus
num_words = args.num_words

# Generate Markov chain
markov = markovgen.Markov(corpus)
output = markov.generate_markov_text(num_words)

# Capitalize first letter of paragraph
# Previously used .capitalize(), but that lowercases everything else in the string
output = re.sub('([a-zA-Z])', lambda x: x.groups()[0].upper(), output, 1)
print(output)