import markovify

with open("sources/games.txt", "r") as f:
    data_games = f.readlines()
with open("sources/toys.txt", "r") as f:
    data_toys = f.readlines()
with open("sources/tv.txt", "r") as f:
    data_tv = f.readlines()

corpus_games = ""
for row in data_games:
     corpus_games += row.strip() + ". "
model_games = markovify.Text(corpus_games, state_size=1)

corpus_toys = ""
for row in data_toys:
     corpus_toys += row.strip() + ". "
model_toys = markovify.Text(corpus_toys, state_size=1)

corpus_tv = ""
for row in data_tv:
     corpus_tv += row.strip() + ". "
model_tv = markovify.Text(corpus_tv, state_size=1)

model = markovify.combine([model_games, model_toys, model_tv], [1, 1, 3])

gen = None
for r in range(1):
    while gen == None:
        gen = model.make_short_sentence(50)
    print(gen.replace(".", "").replace(":", ""))