Create autoplay feature

Use ChatGPT to get song recommendations based on current song queue.

1 files changed, 95 insertions, 0 deletions

diff --git a/code/cogs/autoplay.py b/code/cogs/autoplay.py
new file mode 100644
index 0000000..cc5de91
--- /dev/null
+++ b/code/cogs/autoplay.py
@@ -0,0 +1,95 @@
+import discord
+import datetime
+from discord import app_commands
+from discord.ext import commands
+from cogs.music import Music
+from typing import Literal
+from ai_recommendations import add_song_recommendations
+
+from global_variables import BOT_COLOR
+
+
+class Autoplay(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @app_commands.command()
+    @app_commands.check(Music.create_player)
+    @app_commands.describe(toggle="Turn autoplay ON or OFF")
+    async def autoplay(
+        self, interaction: discord.Interaction, toggle: Literal["ON", "OFF"]
+    ):
+        "Keep the music playing forever with music suggestions from OpenAI"
+        if toggle == "OFF":
+            self.bot.autoplay.remove(interaction.guild.id)
+
+            embed = discord.Embed(
+                title="Autoplay Off",
+                description="Autoplay has been turned off. I will no longer automatically add new songs to the queue based on AI recommendations.",
+                color=BOT_COLOR,
+            )
+            return await interaction.response.send_message(embed=embed)
+
+        # Otherwise, toggle must be "ON", so enable autoplaying
+
+        if interaction.guild.id in self.bot.autoplay:
+            embed = discord.Embed(
+                title="Autoplay Already Enabled",
+                description="Autoplay is already enabled. If you would like to turn it off, choose the `OFF` option in the `/autoplay` command.",
+                color=BOT_COLOR,
+            )
+            return await interaction.response.send_message(embed=embed, ephemeral=True)
+
+        player = self.bot.lavalink.player_manager.get(interaction.guild.id)
+
+        if len(player.queue) < 5:
+            embed = discord.Embed(
+                title="Not Enough Context",
+                description="You must have at least 5 songs in the queue so that I can get a good understanding of what music I should continue to play. Add some more music to the queue, then try again.",
+                color=BOT_COLOR,
+            )
+            embed.set_footer(
+                text=datetime.datetime.now(datetime.timezone.utc).strftime(
+                    "%Y-%m-%d %H:%M:%S"
+                )
+                + " UTC"
+            )
+            return await interaction.response.send_message(embed=embed, ephemeral=True)
+
+        inputs = {}
+        for song in player.queue[:10]:
+            inputs[song.title] = song.author
+
+        embed = discord.Embed(
+            title="Getting AI Recommendations",
+            description="Attempting to generate recommendations based on the current songs in your queue. Just a moment...",
+            color=BOT_COLOR,
+        )
+        await interaction.response.send_message(embed=embed)
+
+        if await add_song_recommendations(self.bot.user, player, 5, inputs):
+            self.bot.autoplay.append(interaction.guild.id)
+            embed = discord.Embed(
+                title=":infinity: Autoplay Enabled :infinity:",
+                description=f"I have added a few similar songs to the queue and will continue to do so once the queue gets low again. Now just sit back and enjoy the music!\n\nEnabled by: {interaction.user.mention}",
+                color=BOT_COLOR,
+            )
+            await interaction.edit_original_response(embed=embed)
+
+        else:
+            embed = discord.Embed(
+                title="Autoplay Error",
+                description="Autoplay is an experimental feature, meaning sometimes it doesn't work as expected. I had an error when attempting to get similar songs for you, please try running the command again. If the issue persists, fill out a bug report with the </bug:1224840889906499626> command.",
+                color=BOT_COLOR,
+            )
+            embed.set_footer(
+                text=datetime.datetime.now(datetime.timezone.utc).strftime(
+                    "%Y-%m-%d %H:%M:%S"
+                )
+                + " UTC"
+            )
+            await interaction.edit_original_response(embed=embed)
+
+
+async def setup(bot):
+    await bot.add_cog(Autoplay(bot))