2 files changed, 60 insertions, 1 deletions

diff --git a/cogs/ai.py b/cogs/ai.py
index 31e384d..984e5f7 100644
--- a/cogs/ai.py
+++ b/cogs/ai.py
@@ -1,7 +1,6 @@
 import os
 
 import discord
-import google.generativeai as genai
 from discord.ext import commands
 import logging
 import requests
diff --git a/cogs/voice.py b/cogs/voice.py
new file mode 100644
index 0000000..600a7eb
--- /dev/null
+++ b/cogs/voice.py
@@ -0,0 +1,60 @@
+import discord
+from discord.ext import commands
+import logging
+import asyncio
+
+logger = logging.getLogger(__name__)
+
+class Voice(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    async def ensure_voice_client(self, ctx):
+        """Ensure we have a valid voice client connection."""
+        if not ctx.author.voice or not ctx.author.voice.channel:
+            await ctx.send("You need to be in a voice channel first!")
+            return None
+            
+        voice_client = discord.utils.get(self.bot.voice_clients, guild=ctx.guild)
+        
+        # If no voice client exists, create one
+        if not voice_client or not voice_client.is_connected():
+            try:
+                voice_client = await ctx.author.voice.channel.connect(timeout=30.0, reconnect=True)
+                logger.info(f"Connected to voice channel: {ctx.author.voice.channel}")
+            except asyncio.TimeoutError:
+                await ctx.send("Connection to voice channel timed out.")
+            except discord.ClientException as e:
+                logger.error(f"Failed to connect to voice channel: {e}")
+                await ctx.send("Failed to connect to voice channel. Please try again.", ephemeral=True)
+        # If we're in a different channel, move to the user's channel
+        elif voice_client.channel != ctx.author.voice.channel:
+            try:
+                await voice_client.move_to(ctx.author.voice.channel)
+                logger.info(f"Moved to voice channel: {ctx.author.voice.channel}")
+            except asyncio.TimeoutError:
+                await ctx.send("Connection to voice channel timed out.")
+            except discord.ClientException as e:
+                logger.error(f"Failed to move to voice channel: {e}")
+                await ctx.send("Failed to move to your voice channel. Please try again.", ephemeral=True)
+        
+        # Verify the connection is healthy
+        if not voice_client.is_connected():
+            try:
+                await voice_client.connect(reconnect=True, timeout=10.0)
+            except Exception as e:
+                logger.error(f"Failed to reconnect to voice: {e}")
+                await ctx.send("Failed to establish voice connection. Please try again.")
+        return None
+
+    async def connect(self, ctx):
+        """Connect to the user's voice channel with improved connection handling."""
+        voice_client = await self.ensure_voice_client(ctx)
+        if not voice_client:
+            return None
+            
+        return voice_client
+
+
+def setup(bot):
+    bot.add_cog(Voice(bot))
\ No newline at end of file