make_sentence_with_start() returns error if only one word is used

[Kyu]

It used to just use memory and do nothing, but now it gives:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\Users\Me\AppData\Local\Programs\Python\Python35-32\lib\site-packages\mark
ovify\text.py", line 140, in make_sentence_with_start
    return self.make_sentence(init_state, **kwargs)
  File "C:\Users\Me\AppData\Local\Programs\Python\Python35-32\lib\site-packages\mark
ovify\text.py", line 122, in make_sentence
    words = prefix + self.chain.walk(init_state)
  File "C:\Users\Me\AppData\Local\Programs\Python\Python35-32\lib\site-packages\mark
ovify\chain.py", line 114, in walk
    return list(self.gen(init_state))
  File "C:\Users\Me\AppData\Local\Programs\Python\Python35-32\lib\site-packages\mark
ovify\chain.py", line 103, in gen
    next_word = self.move(state)
  File "C:\Users\Me\AppData\Local\Programs\Python\Python35-32\lib\site-packages\mark
ovify\chain.py", line 89, in move
    choices, weights = zip(*self.model[state].items())
KeyError: ('I',)

This is in Python 3.5.2

Code:

with open("BlocSpeaks.txt", "r", encoding="utf-8") as txt:
    text = text.read()

text_model = markovify.Text(text)

text_model.make_sentence_with_start("I")
#Returns error as seen above

text_model.make_sentence_with_start("I dont")

'''"I dont play BLOC tho fucking homofag I can not longer stand Commy's defiance he is unk
illable He only was because she hadn\u2019t done anything."'''

text_model.make_sentence_with_start("I will")
'''I will ban you without a special kind of dull tbh back If you need to check bloc?''''

[jsvine]

What version of markovify are you using? You'll need v0.4.1 or later to be able to pass one-word beginnings.

[Kyu]

Yep, that was the issue 🤦

[Kyu]

It's now returning ('_BEGIN', 'word') , but I guess that's an issue with not having enough sample to use.

Do you have any tips on managing memory usage? cpu usage spikes up and down 90% when this is used, and memory usage doubles and stays high. This is with a 2mb txt file

[jsvine]

Hmmm, haven't noticed memory problems like that lately, but it's possible I've overlooked an edge case. Have you tried profiling your program?

If you send over the script and corpus, I can also try taking a look.

[Kyu]

async def bloc_speaks(prompt=""):
    try:
        with open(os.getcwd() + '/BlocSim/BlocSpeaks.txt',
                  'r', encoding='utf-8') as f:
            text = f.read()
    except FileNotFoundError:
        print("File not found")
        return
    text_model = markovify.Text(text)
    try:
        if prompt:
            sentence = text_model.make_sentence_with_start(prompt)
            return sentence
        else:
            sentence = text_model.make_sentence()
            return sentence
    except Exception as e:
        return (e)

corpus