Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 3,945 Bytes
60d2967 1bc04be 60d2967 1bc04be 60d2967 1bc04be 60d2967 1bc04be 60d2967 036628e 77815e1 036628e 77815e1 036628e 60d2967 1bc04be 77815e1 60d2967 1bc04be 60d2967 1bc04be 60d2967 036628e 60d2967 036628e 60d2967 036628e 60d2967 036628e 60d2967 |
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 |
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
# 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
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) |