music-bot / app.py
not-lain's picture
Update app.py
adfd0c7 verified
raw
history blame
4.17 kB
import discord
from discord.ext import commands
from huggingface_hub import hf_hub_download
import gradio as gr
from dotenv import load_dotenv
import os
import threading
import asyncio
# Load environment variables
load_dotenv()
# Download song
if os.path.exists('assets') is False:
os.makedirs('assets', exist_ok=True)
hf_hub_download("not-lain/assets", "sample.mp3", repo_type="dataset",local_dir="assets")
# Bot configuration
intents = discord.Intents.default()
intents.message_content = True
bot = commands.Bot(command_prefix='!', intents=intents)
# Music bot class
class MusicBot:
def __init__(self):
self.is_playing = False
self.voice_client = None
async def join_voice(self, ctx):
if ctx.author.voice:
channel = ctx.author.voice.channel
if self.voice_client is None:
self.voice_client = await channel.connect()
else:
await self.voice_client.move_to(channel)
else:
await ctx.send("You need to be in a voice channel!")
async def play_next(self, ctx):
if not self.is_playing:
self.is_playing = True
try:
audio_source = discord.FFmpegPCMAudio("assets/sample.mp3")
def after_playing(e):
self.is_playing = False
# test loop by default
if e:
print(f"Playback error: {e}")
asyncio.run_coroutine_threadsafe(self.play_next(ctx), bot.loop)
self.voice_client.play(audio_source, after=after_playing)
except Exception as e:
print(f"Error playing file: {e}")
await ctx.send("Error playing the song.")
self.is_playing = False
music_bot = MusicBot()
@bot.event
async def on_ready():
print(f'Bot is ready! Logged in as {bot.user}')
print("Syncing commands...")
try:
await bot.tree.sync(guild=None) # Set to None for global sync
print("Successfully synced commands globally!")
except discord.app_commands.errors.CommandSyncFailure as e:
print(f"Failed to sync commands: {e}")
except Exception as e:
print(f"An error occurred while syncing commands: {e}")
@bot.tree.command(name="play", description="Play the sample music")
async def play(interaction: discord.Interaction):
await interaction.response.defer()
ctx = await commands.Context.from_interaction(interaction)
await music_bot.join_voice(ctx)
if not music_bot.is_playing:
await music_bot.play_next(ctx)
await interaction.followup.send('Playing sample music!')
else:
await interaction.followup.send('Already playing!')
# Replace the existing skip command with this version
@bot.tree.command(name="skip", description="Skip the current song")
async def skip(interaction: discord.Interaction):
if music_bot.voice_client:
music_bot.voice_client.stop()
await interaction.response.send_message('Skipped current song!')
else:
await interaction.response.send_message('No song is currently playing!')
# Replace the existing leave command with this version
@bot.tree.command(name="leave", description="Disconnect bot from voice channel")
async def leave(interaction: discord.Interaction):
if music_bot.voice_client:
await music_bot.voice_client.disconnect()
music_bot.voice_client = None
music_bot.queue = []
music_bot.is_playing = False
await interaction.response.send_message('Bot disconnected!')
else:
await interaction.response.send_message('Bot is not in a voice channel!')
def run_discord_bot():
bot.run(os.getenv('DISCORD_TOKEN'))
# Create the Gradio interface
with gr.Blocks() as iface:
gr.Markdown("# Discord Music Bot Control Panel")
gr.Markdown("Bot is running in background")
if __name__ == "__main__":
# Start the Discord bot in a separate thread
bot_thread = threading.Thread(target=run_discord_bot, daemon=True)
bot_thread.start()
# Run Gradio interface in the main thread
iface.launch(debug=True)