import nltk
from random import choice
from nltk.corpus import genesis

g = genesis.words('english-kjv.txt')
cfd = nltk.ConditionalFreqDist(nltk.bigrams(g))

def next_word(w):
    return choice([k for k in cfd[w].keys() for i in range(cfd[w][k])])
def nwords(n, w):
    for i in range(n):
        print(w, end=' ')
        w = next_word(w)
    print()
def nwords2(n, w):		# A more useful function returning a list of words
    result = []
    for i in range(n):
        result.append(w)
        w = next_word(w)
    return result