gooberinc/goober
3
0
Fork 1
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

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) 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')}") @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: if not markov_model:
await send_message(ctx, f"{get_translation(LOCALE, 'command_talk_insufficent_text')}") await send_message(ctx, f"{get_translation(LOCALE, 'command_talk_insufficent_text')}")
return return
response = None response = None
for _ in range(20): # Try to generate a coherent sentence for _ in range(20):
response = markov_model.make_sentence(tries=100) 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: if response and response not in generated_sentences:
# Preprocess the sentence for grammar improvements if sentence_size > 1:
response = improve_sentence_coherence(response) response = improve_sentence_coherence(response)
generated_sentences.add(response) generated_sentences.add(response)
break break