diff --git a/botbotbot/tts.py b/botbotbot/tts.py index dc39a1f..e3df4b4 100644 --- a/botbotbot/tts.py +++ b/botbotbot/tts.py @@ -32,6 +32,7 @@ class CambAI: return self.get_iostream(text, run_id) def gen_task(self, text: str) -> str | None: + logger.info(f"Generating task for {text}.") tts_payload = { "text": text, "voice_id": 20299, @@ -51,6 +52,7 @@ class CambAI: return task_id def get_runid(self, task_id: str | None) -> int | None: + logger.info(f"Getting runid for {task_id}.") if task_id is None: return None @@ -68,6 +70,7 @@ class CambAI: return run_id def get_iostream(self, text: str, run_id: int | None) -> pathlib.Path | None: + logger.info(f"Getting stream for {text} [{run_id}].") if run_id is None: return None diff --git a/botbotbot/voice.py b/botbotbot/voice.py index 4a3083c..0eebbb6 100644 --- a/botbotbot/voice.py +++ b/botbotbot/voice.py @@ -40,12 +40,12 @@ class VoiceBot: async def random_connect(self) -> None: while True: - logger.info("Random connect.") for g_id in self.guild_ids: guild = self.bot.get_guild(g_id) if guild is None or random.random() > 10 / 100: continue + logger.info("Random connect.") voice_chan = random.choice(guild.voice_channels) await self.connect_voice(voice_chan) await asyncio.sleep(60) @@ -63,12 +63,14 @@ class VoiceBot: await vo.disconnect() return elif len(channel.members) == 1: - member = channel.members[0].display_name + member = self.rnd_name(channel.members[0]) + elif random.random() < 50 / 100: + member = self.rnd_name(random.choice(channel.members)) else: member = ( - ", ".join(member.display_name for member in channel.members[:-1]) + ", ".join(self.rnd_name(member) for member in channel.members[:-1]) + " et " - + channel.members[-1].display_name + + self.rnd_name(channel.members[-1]) ) script = None @@ -88,6 +90,10 @@ class VoiceBot: await vo.play(source, wait_finish=True) await vo.disconnect() + @classmethod + def rnd_name(cls, user: discord.Member) -> str: + return random.choice([user.display_name, user.name]) + async def connect_voice_chan( self, channel: discord.VoiceChannel | discord.StageChannel ) -> discord.VoiceClient: