edpibu/botbotbot
1
Fork 0
Code Issues Releases Activity

Enhance VoiceBot to support AIBot integration for dynamic TTS script generation

Browse source
This commit is contained in:
Edgar P. Burkhart 2025-03-22 23:21:07 +01:00
parent e6fc040d78
commit 31f10b8438
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
2 changed files with 15 additions and 8 deletions

2
botbotbot/__init__.py
View file

@ -59,7 +59,7 @@ def main() -> None:
text_bot = TextBot(bot, wl, aibot=aibot, shuffler=shf) text_bot = TextBot(bot, wl, aibot=aibot, shuffler=shf)
text_bot.init_events() text_bot.init_events()
voice_bot = VoiceBot(bot, cambai, shuffler=shf) voice_bot = VoiceBot(bot, cambai, aibot=aibot, shuffler=shf)
voice_bot.init_events() voice_bot.init_events()
@bot.listen("on_ready") @bot.listen("on_ready")

21
botbotbot/voice.py
View file

@ -3,6 +3,7 @@ import random
import discord import discord
from botbotbot.ai import AIBot
from botbotbot.shuffle import Shuffler from botbotbot.shuffle import Shuffler
from botbotbot.tts import CambAI from botbotbot.tts import CambAI
@ -14,11 +15,13 @@ class VoiceBot:
self, self,
bot: discord.Bot, bot: discord.Bot,
cambai: CambAI | None = None, cambai: CambAI | None = None,
aibot: AIBot | None = None,
shuffler: Shuffler | None = None, shuffler: Shuffler | None = None,
) -> None: ) -> None:
self.bot = bot self.bot = bot
self.cambai = cambai self.cambai = cambai
self.shf = shuffler self.shf = shuffler
self.aibot = aibot
def init_events(self) -> None: def init_events(self) -> None:
self.bot.add_listener(self.on_voice_state_update, "on_voice_state_update") self.bot.add_listener(self.on_voice_state_update, "on_voice_state_update")
@ -45,13 +48,17 @@ class VoiceBot:
and random.random() < 5 / 100 and random.random() < 5 / 100
): ):
logger.info("Generating tts") logger.info("Generating tts")
script = random.choice( script = None
[ if self.aibot is not None and random.random() < 20 / 100:
"Salut la jeunesse !", script = self.aibot.answer("Dis un truc aléatoire et original.")
f"Salut {member.display_name}, ça va bien ?", if script is None:
"Allo ? À l'huile !", script = random.choice(
] [
) "Salut la jeunesse !",
f"Salut {member.display_name}, ça va bien ?",
"Allo ? À l'huile !",
]
)
source = await discord.FFmpegOpusAudio.from_probe(self.cambai.tts(script)) source = await discord.FFmpegOpusAudio.from_probe(self.cambai.tts(script))
vo: discord.VoiceClient = await after.channel.connect() vo: discord.VoiceClient = await after.channel.connect()