Skip to content

Commit 87b9176

Browse files
small improvement
1 parent 35f8bf1 commit 87b9176

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

bot.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,8 @@ def generate_convo_text(start_line: str = None)->str:
6565
lines.insert(random_index, start_word)
6666
text = ''.join(lines[6:]) # Join the lines starting from index 6 (line 7)
6767
# Build the model.
68-
text_model = markovify.Text(text, state_size=3)
68+
state = random.choice([2,3])
69+
text_model = markovify.Text(text, state_size=state)
6970

7071
# Generate a random number between 5 and 10
7172
random_number = random.randint(5, 10)
@@ -463,7 +464,8 @@ async def on_message(self, message):
463464

464465
# Handle Replies to the Bot
465466
if message.reference and message.reference.resolved and message.reference.resolved.author == bot.user:
466-
await handle_conversation(message)
467+
words = message.content.split()
468+
await message.reply(generate_convo_text(words))
467469
return
468470

469471
# Handle Mentions of the Bot

0 commit comments

Comments
 (0)