# Dan Richert 2005
# dan.richert@gmail.com    

## usage:
## gen1.py <source text> <output file> <stanzas>
## problem with last word?  'adjacent' when using '/text/bread.txt'


from textgen import *
import sys, random

fp = open(sys.argv[2], 'w')

p = build_pdict()
t = TextGen(open(sys.argv[1]).read())

show_nextwords = 0
stanzas = int(sys.argv[3])
counter = 0
while counter <= stanzas:
	print str(counter)+' of '+str(stanzas)
	f = Formatter(random.randint(2, 20), random.randint(2,20))
	nextwords_length = random.randint(2,6)
	t.build_dict(nextwords_length)
	f.buffer = t.prob_text(25)
	fp.write('\n\n\n\n')
	if show_nextwords: fp.write('nextwords:\t%s\n' % (str(nextwords_length)))
	fp.write(f.format())
	counter += 1	
fp.write('\n\n\n\n\n')
fp.close()