import collections

def following_words(txt):
    words = txt.split()
    freq = collections.defaultdict(list)
    for word, next_word in zip(words, words[1:]):
        freq[word].append(next_word)
    return freq

def freq_following_word(txt):
    following = following_words(txt)
    for f in following:
        vals = collections.Counter(following[f])
        s = sorted(vals.most_common(), key = lambda x: (-x[1], x[0]))
        following[f] = s[0][0]
    return following

def text(word, freq, num):
    words = []
    for i in range(num):
        words.append(word)
        word = freq[word]
    return ' '.join(words)


import urllib.request
txt = 'danes je lep dan danes sije sonce danes sije dan ki je sonce'
#urllib.request.urlopen('http://squeeb1134.tripod.com/1984.txt').read().decode('utf8')
print (text('danes', freq_following_word(txt), 5))