96 lines
4.0 KiB
Python
96 lines
4.0 KiB
Python
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))
|