1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
import discord
import datetime
from discord import app_commands
from discord.ext import commands
from cogs.music import Music
from typing import Literal
from utils.ai_recommendations import add_song_recommendations
from utils.config 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:1228216490386391052> 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.openai, self.bot.user, player, 5, inputs
):
self.bot.autoplay.append(interaction.guild.id)
embed = discord.Embed(
title=":infinity: Autoplay Enabled :infinity:",
description=(
"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:"
f" {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))
|
