ctih1/goober
1
0
Fork 0
forked from gooberinc/goober
Code Pull requests Activity

uh yeah

Browse source
This commit is contained in:
WhatDidYouExpect 2025-03-23 20:18:52 +01:00
parent c31f5aa852
commit 41bad42bbb
1 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

14
bot.py
View file

@ -428,16 +428,22 @@ async def retrain(ctx):
await send_message(ctx, f"{get_translation(LOCALE, 'command_markov_retrain_successful').format(data_size=data_size)}", edit=True, message_reference=processing_message_ref)
@bot.hybrid_command(description=f"{get_translation(LOCALE, 'command_desc_talk')}")
async def talk(ctx):
async def talk(ctx, sentence_size: int = 5):
if not markov_model:
await send_message(ctx, f"{get_translation(LOCALE, 'command_talk_insufficent_text')}")
return
response = None
for _ in range(20): # Try to generate a coherent sentence
response = markov_model.make_sentence(tries=100)
for _ in range(20):
if sentence_size == 1:
response = markov_model.make_short_sentence(max_chars=100, tries=100)
if response:
response = response.split()[0]
else:
response = markov_model.make_sentence(tries=100, max_words=sentence_size)
if response and response not in generated_sentences:
# Preprocess the sentence for grammar improvements
if sentence_size > 1:
response = improve_sentence_coherence(response)
generated_sentences.add(response)
break