From 15ca2786cc0fa2240bbe57fb639fdac70f78678e Mon Sep 17 00:00:00 2001 From: salmonik Date: Mon, 5 Jan 2026 11:12:52 +0200 Subject: better ai --- cogs/ai.py | 22 +++++++++++++++++++--- 1 file changed, 19 insertions(+), 3 deletions(-) diff --git a/cogs/ai.py b/cogs/ai.py index 984e5f7..a3c8201 100644 --- a/cogs/ai.py +++ b/cogs/ai.py @@ -5,7 +5,9 @@ from discord.ext import commands import logging import requests import json - +from gtts import gTTS +from .music.play_utils import PlayUtils +import io logger = logging.getLogger(__name__) @@ -20,7 +22,8 @@ class AI(commands.Cog): async def senator( self, ctx: discord.ApplicationContext, - text=discord.Option(str, "message", required=True) + text=discord.Option(str, "message", required=True), + tts=discord.Option(bool, "face?", default=False) ): await ctx.defer() @@ -42,7 +45,20 @@ class AI(commands.Cog): }) ) logger.info(response.json()) - await ctx.send_followup(response.json()["choices"][0]["message"]["content"]) + if not response.json()["choices"]: + await ctx.send_followup("bot siem zesroĊ‚") + else: + response_text = response.json()["choices"][0]["message"]["content"] + if tts: + # Generate TTS + tts_obj = gTTS(text=response_text, lang='pl') + tts_buffer = io.BytesIO() + tts_obj.write_to_fp(tts_buffer) + tts_buffer.seek(0) + await PlayUtils.play(ctx, discord.FFmpegPCMAudio(tts_buffer), delete_after_play=True) + + else: + await ctx.send_followup(response_text) -- cgit v1.2.3