haiku.py

def haiku(sdict):
    sdict = dict([(sk, sv) for sk, sv in sdict.items() if len(sk) >= 3])
    first_5 = choose_words(sdict, 5, 0)
    seven = choose_words(sdict, 7, 1)
    sec_5 = choose_words(sdict, 5, 2)
    first_line = ' '.join(first_5)
    second_line = ' '.join(seven)
    third_line = ' '.join(sec_5)
    hai = ', '.join([first_line, second_line, third_line])
    with open('haiku.txt', 'a+') as h:
        h.write(hai+'\n')
    return hai