1 files changed, 47 insertions, 0 deletions

diff --git a/cogs/tts.py b/cogs/tts.py
new file mode 100644
index 0000000..41dc366
--- /dev/null
+++ b/cogs/tts.py
@@ -0,0 +1,47 @@
+import discord
+from discord.ext import commands
+from gtts import gTTS
+import os
+import datetime
+
+class TTS(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.slash_command(name="tts", description="Generuje TTS dla danego tekstu")
+    async def tts(self, ctx,
+        text: str = discord.Option(str, "Tekst do przeczytania", required=True),
+        output: str = discord.Option(str, "Wybierz sposób odtwarzania", choices=["file", "vc"], required=True)
+    ):
+        tts = gTTS(text=text, lang='pl')
+        tts.save('tts.mp3')
+        if output == "file":
+            await ctx.respond(file=discord.File('tts.mp3'))
+            os.remove('tts.mp3')
+        elif output == "vc":
+            # Check if 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)
+                os.remove('tts.mp3')
+                return
+            channel = ctx.author.voice.channel
+            # Connect to voice channel if not already connected
+            voice_client = discord.utils.get(self.bot.voice_clients, guild=channel.guild)
+            if not voice_client or not voice_client.is_connected():
+                voice_client = await channel.connect()
+            elif voice_client.channel != channel:
+                await voice_client.move_to(channel)
+            # Play the TTS audio
+            if not voice_client.is_playing():
+                audio_source = discord.FFmpegPCMAudio('tts.mp3', options='-filter:a "volume=6"')
+                voice_client.play(audio_source)
+                # Wait for playback to finish
+                while voice_client.is_playing():
+                    await discord.utils.sleep_until(datetime.datetime.now() + datetime.timedelta(seconds=1))
+            os.remove('tts.mp3')
+        else:
+            await ctx.respond("Nieznany sposób odtwarzania.", ephemeral=True)
+            os.remove('tts.mp3')
+
+def setup(bot):
+    bot.add_cog(TTS(bot))
\ No newline at end of file