CodeMirror/Brandon209-Red-bot-Cogs
1
0
Fork 0
mirror of https://github.com/brandons209/Red-bot-Cogs.git synced 2024-06-02 02:24:41 +12:00
Code Issues Projects Releases Wiki Activity

fix actual bugs in markov

Browse source
This commit is contained in:
brandons209 2020-07-22 14:15:14 -04:00
parent d065fb8224
commit 79fda9d5f3
1 changed files with 20 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
markov/markov.py
View file

@ -55,13 +55,17 @@ class Markov(commands.Cog):
""" """
Set max characters of generated text. Set max characters of generated text.
Max size limited by discord is 3000. Max size is 2800.
""" """
if not length: if not length:
curr = await self.config.guild(ctx.guild).max_len() curr = await self.config.guild(ctx.guild).max_len()
await ctx.send(f"Current max length of generated text is `{curr}` characters.") await ctx.send(f"Current max length of generated text is `{curr}` characters.")
return return
if length > 2800:
await ctx.send(error("Max length is 2800."))
return
await self.config.guild(ctx.guild).max_len.set(length) await self.config.guild(ctx.guild).max_len.set(length)
await ctx.tick() await ctx.tick()
@ -81,20 +85,22 @@ class Markov(commands.Cog):
await ctx.send(error("This channel has no data, try talking in it for a bit first!")) await ctx.send(error("This channel has no data, try talking in it for a bit first!"))
return return
last_word = starting_text.split(" ")[-1] if starting_text else None starting_text = starting_text.split(" ") if starting_text else None
last_word = starting_text[-1] if starting_text else None
if not starting_text: if not starting_text:
markov_text = [random.choice(list(model.keys()))] markov_text = [random.choice(list(model.keys()))]
elif last_word in model and not model[last_word]: elif last_word not in model or not model[last_word]:
markov_text = [last_word, random.choice(list(model.keys()))] markov_text = starting_text + [random.choice(list(model.keys()))]
else: else:
markov_text = [last_word, random.choice(model[last_word])] markov_text = starting_text + [random.choice(model[last_word])]
max_len = await self.config.guild(ctx.guild).max_len() max_len = await self.config.guild(ctx.guild).max_len()
tries = 0 tries = 0
max_tries = 20 max_tries = 20
while len(markov_text) < max_len and tries < max_tries: num_chars = len(" ".join(markov_text))
while num_chars < max_len and tries < max_tries:
if "?" in markov_text[-1]: if "?" in markov_text[-1]:
break break
if "\r" in markov_text[-1]: if "\r" in markov_text[-1]:
@ -106,12 +112,18 @@ class Markov(commands.Cog):
# make sure word is in the model and there is data for the word # make sure word is in the model and there is data for the word
if markov_text[-1] in model and model[markov_text[-1]]: if markov_text[-1] in model and model[markov_text[-1]]:
markov_text.append(random.choice(model[markov_text[-1]])) choice = random.choice(model[markov_text[-1]])
num_chars += len(choice)
markov_text.append(choice)
else: else:
markov_text.append(random.choice(list(model.keys()))) choice = random.choice(list(model.keys()))
num_chars += len(choice)
markov_text.append(choice)
tries += 1 tries += 1
markov_text = " ".join(markov_text) markov_text = " ".join(markov_text)
if num_chars > max_len:
markov_text = markov_text[:max_len]
member = ctx.author member = ctx.author
embed = discord.Embed(title="Generated Text", description=markov_text, colour=member.colour) embed = discord.Embed(title="Generated Text", description=markov_text, colour=member.colour)