add youtube support

.gitignore

@@ -1,3 +1,5 @@
 .env
 assets/
-.cache
+.cache
+test*
+cookies.txt

app.py

@@ -1,31 +1,38 @@
 import discord
 from discord.ext import commands
-from huggingface_hub import hf_hub_download
-import gradio as gr 
+import gradio as gr
 from dotenv import load_dotenv
 import os
 import threading
 import asyncio
-
+import yt_dlp
 
 # 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")
+# Check if cookies.txt exists
+if not os.path.exists('cookies.txt'):
+    txt = os.getenv('COOKIES')
+    with open('cookies.txt', 'w') as f:
+        f.write(txt)
 
 # 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
+        self.queue = []
+        self.ydl_opts = {
+            'format': 'bestaudio/best',
+            'cookiefile': 'cookies.txt',
+            'quiet': True,
+            'no_warnings': True,
+            'extract_flat': 'in_playlist'
+        }
 
     async def join_voice(self, ctx):
         if ctx.author.voice:
@@ -34,25 +41,44 @@ class MusicBot:
                 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!")
+            return True
+        return False
 
     async def play_next(self, ctx):
-        if not self.is_playing:
+        if self.queue and not self.is_playing:
             self.is_playing = True
+            url = self.queue.pop(0)
+            
             try:
-                audio_source = discord.FFmpegPCMAudio("assets/sample.mp3")
-                def after_playing(e):
+                with yt_dlp.YoutubeDL(self.ydl_opts) as ydl:
+                    info = ydl.extract_info(url, download=False)
+                    audio_url = info['url']
+                    title = info.get('title', 'Unknown title')
+                
+                # Create FFmpeg audio source
+                ffmpeg_options = {
+                    'before_options': '-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5',
+                    'options': '-vn -bufsize 64k'
+                }
+                audio_source = discord.FFmpegPCMAudio(audio_url, **ffmpeg_options)
+                
+                def after_playing(error):
                     self.is_playing = False
-                    # test loop by default
-                    if e:
-                        print(f"Playback error: {e}")
+                    if error:
+                        print(f"Playback error: {error}")
                     asyncio.run_coroutine_threadsafe(self.play_next(ctx), bot.loop)
-                self.voice_client.play(audio_source, after=after_playing)
+
+                # Apply volume transformation and play
+                transformed_source = discord.PCMVolumeTransformer(audio_source, volume=0.5)
+                self.voice_client.play(transformed_source, after=after_playing)
+                
+                await ctx.send(f"Now playing: {title}")
+                
             except Exception as e:
-                print(f"Error playing file: {e}")
-                await ctx.send("Error playing the song.")
+                print(f"Error playing URL {url}: {e}")
                 self.is_playing = False
+                await ctx.send(f"Error playing the song: {str(e)}")
+                await self.play_next(ctx)
 
 music_bot = MusicBot()
 
@@ -68,19 +94,28 @@ async def on_ready():
     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):
+@bot.tree.command(name="play", description="Play a YouTube URL")
+async def play(interaction: discord.Interaction, url: str):
     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!')
+    if not await music_bot.join_voice(ctx):
+        await interaction.followup.send("You need to be in a voice channel!")
+        return
+
+    try:
+        with yt_dlp.YoutubeDL(music_bot.ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=False)
+            title = info.get('title', 'Unknown title')
+        
+        music_bot.queue.append(url)
+        await interaction.followup.send(f'Added to queue: {title}')
+        
+        if not music_bot.is_playing:
+            await music_bot.play_next(ctx)
+    except Exception as e:
+        await interaction.followup.send(f'Error: {str(e)}')
 
-# 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:
@@ -89,7 +124,6 @@ async def skip(interaction: discord.Interaction):
     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:

requirements.txt

@@ -1,7 +1,7 @@
 discord.py
-youtube_dl
 PyNaCl
 ffmpeg-python
-yt-dlp
 python-dotenv
-spotipy
+spotipy
+gradio
+yt-dlp