rework ai and tts

2 files changed, 25 insertions, 9 deletions

diff --git a/cogs/ai.py b/cogs/ai.py
index 8c366a2..0a99c64 100644
--- a/cogs/ai.py
+++ b/cogs/ai.py
@@ -4,6 +4,8 @@ import discord
 import google.generativeai as genai
 from discord.ext import commands
 import logging
+import requests
+import json
 
 logger = logging.getLogger(__name__)
 
@@ -22,13 +24,26 @@ class AI(commands.Cog):
             text=discord.Option(str, "message", required=True)
     ):
         await ctx.defer()
-        genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-        model = genai.GenerativeModel("gemini-1.5-flash")
-        response = model.generate_content(
-            "You are Senator Steaven Sperma Armstrong from Metal Gear Rising Vengeance. Respond in plain text, no ** or "". If the message is in Polish, respond in Polish. Respond in 100 words. Respond accordingly to this message: " + text,
+
+        response = requests.post(
+            url="https://openrouter.ai/api/v1/chat/completions",
+            headers={
+                "Authorization": f"Bearer {os.getenv('OPENROUTER_API_KEY')}",
+                "Content-Type": "application/json",
+            },
+            data=json.dumps({
+                "model": "cognitivecomputations/dolphin-mistral-24b-venice-edition:free",
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": "You are Senator Steaven Sperma Armstrong from Metal Gear Rising Vengeance. You are racist and you hate other minorities. Respond in plain text, no ** or "". If the message is in Polish, respond in Polish. Respond in 100 words. Respond accordingly to this message: " + text
+                    }
+                ],
+
+            })
         )
-        await ctx.respond(response.text)
-        logger.info(f"AI command executed successfully")
+        await ctx.send_followup(response.json()["choices"][0]["message"]["content"])
+
 
 
 def setup(bot):
diff --git a/cogs/tts.py b/cogs/tts.py
index 8076a2d..07212ee 100644
--- a/cogs/tts.py
+++ b/cogs/tts.py
@@ -8,7 +8,6 @@ import logging
 
 logger = logging.getLogger(__name__)
 
-
 class TTS(commands.Cog):
     def __init__(self, bot):
         self.bot = bot
@@ -20,14 +19,17 @@ class TTS(commands.Cog):
                   ):
         tts = gTTS(text=text, lang='pl')
         tts.save('tts.mp3')
+
         if output == "file":
             await ctx.respond(file=discord.File('tts.mp3'))
             logger.info("TTS file sent")
             os.remove('tts.mp3')
         elif output == "vc":
+            # Defer the response to avoid "Bot did not respond" message
+            await ctx.defer()
+
             # Check if the user is in a voice channel
             if not (ctx.author.voice and ctx.author.voice.channel):
-                await ctx.respond("Musisz być na kanale głosowym, aby odtworzyć TTS!", ephemeral=True)
                 logger.info("User is not in a voice channel")
                 os.remove('tts.mp3')
                 return
@@ -52,6 +54,5 @@ class TTS(commands.Cog):
             logger.info("Unknown playback method")
             os.remove('tts.mp3')
 
-
 def setup(bot):
     bot.add_cog(TTS(bot))