better ai

1 files 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)