Upload app.py

app.py

@@ -37,20 +37,12 @@ def format_credits_message(credits_info):
     return f"Credits Remaining: {credits_info['character_count']}"
 
 # Discord bot setup
-class VoiceBot(discord.Client):
-    def __init__(self):
-        super().__init__(intents=discord.Intents.default())
-        self.tree = app_commands.CommandTree(self)
-        
-    async def setup_hook(self):
-        await self.tree.sync()
-
-client = VoiceBot()
-
-# Get your Discord server ID (replace with your actual server ID)
-GUILD_ID = discord.Object(id=os.getenv('DISCORD_GUILD_ID'))
+intents = discord.Intents.default()
+intents.message_content = True
+client = discord.Client(intents=intents)
+tree = app_commands.CommandTree(client)
 
-@client.tree.command(name="list", description="List all available voices", guild=GUILD_ID)
+@tree.command(name="list", description="List all available voices")
 async def voice_list(interaction: discord.Interaction):
     await interaction.response.defer()
     available_voices = get_available_voices()
@@ -75,7 +67,7 @@ async def voice_autocomplete(
         for voice in voices if current.lower() in voice.lower()
     ][:25]
 
-@client.tree.command(name="create", description="Create a voice message", guild=GUILD_ID)
+@tree.command(name="create", description="Create a voice message")
 @app_commands.describe(
     text="Text to convert to speech",
     voice_name="Select a voice to use",
@@ -121,8 +113,9 @@ async def voice_create(
         )
         
         # Save audio to temporary file
-        with open("temp.mp3", "wb") as f:
-            f.write(audio)
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as temp_file:
+            temp_file.write(audio)
+            temp_path = temp_file.name
         
         # Get updated credits
         credits_info = get_remaining_credits()
@@ -135,11 +128,11 @@ async def voice_create(
         )
         await interaction.followup.send(
             embed=embed,
-            file=discord.File("temp.mp3")
+            file=discord.File(temp_path)
         )
         
         # Clean up
-        os.remove("temp.mp3")
+        os.unlink(temp_path)
         
     except Exception as e:
         logger.error(f"Error generating audio: {str(e)}", exc_info=True)
@@ -149,256 +142,11 @@ async def voice_create(
 async def on_ready():
     logger.info(f"Bot is ready and logged in as {client.user}")
     try:
-        synced = await client.tree.sync()
+        synced = await tree.sync()
         logger.info(f"Synced {len(synced)} command(s)")
     except Exception as e:
         logger.error(f"Failed to sync commands: {e}")
 
-# Start Discord bot in a separate thread
-def start_discord_bot():
-    """Start the Discord bot"""
-    DISCORD_TOKEN = os.getenv("DISCORD_BOT_TOKEN")
-    if not DISCORD_TOKEN:
-        logger.error("DISCORD_BOT_TOKEN not found!")
-        return
-    
-    logger.info("Starting Discord bot...")
-    try:
-        client.run(DISCORD_TOKEN)
-    except Exception as e:
-        logger.error(f"Failed to start Discord bot: {str(e)}")
-import os
-import gradio as gr
-import threading
-import discord
-from discord import app_commands
-from typing import List
-from elevenlabs import set_api_key, voices, generate, Voice, VoiceSettings, User
-import tempfile
-import speech_recognition as sr
-from pydub import AudioSegment
-import logging
-
-# Set up logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# Set your ElevenLabs API key
-ELEVENLABS_API_KEY = os.getenv("ELEVENLABS_API_KEY")
-set_api_key(ELEVENLABS_API_KEY)
-
-# Create the voice command group
-voice_group = app_commands.Group(name="voice", description="Voice generation commands")
-
-# Discord bot setup
-intents = discord.Intents.default()
-intents.message_content = True
-client = discord.Client(intents=intents)
-tree = app_commands.CommandTree(client)
-tree.add_command(voice_group)
-
-def get_available_voices():
-    """Fetch only custom voices from ElevenLabs account"""
-    all_voices = voices()
-    return {voice.name: voice.voice_id for voice in all_voices if not voice.category == "premade"}
-
-def get_remaining_credits():
-    """Get remaining character credits from ElevenLabs"""
-    user = User.from_api()
-    subscription = user.subscription
-    return {
-        "character_count": subscription.character_count,
-        "character_limit": subscription.character_limit
-    }
-
-def format_credits_message(credits_info):
-    """Format credits information into a readable message"""
-    return f"Credits Remaining: {credits_info['character_count']}"
-
-# Discord bot commands
-# Command group for voice commands
-voice_group = app_commands.Group(name="voice", description="Voice generation commands")
-
-@voice_group.command(name="list", description="List all available voices")
-async def voice_list(interaction: discord.Interaction):
-    await interaction.response.defer()
-    available_voices = get_available_voices()
-    voice_list = "\n".join([f"• {name}" for name in available_voices.keys()])
-    credits_info = get_remaining_credits()
-    credits_msg = format_credits_message(credits_info)
-    
-    embed = discord.Embed(
-        title="Available Voices",
-        description=f"{voice_list}\n\n{credits_msg}",
-        color=0x2B2D31
-    )
-    await interaction.followup.send(embed=embed)
-
-@voice_group.command(name="create", description="Create a voice message")
-@app_commands.describe(
-    text="Text to convert to speech",
-    voice_name="Select a voice to use",
-    stability="Voice stability (0-1)",
-    clarity="Voice clarity (0-1)",
-    style="Speaking style (0-1)"
-)
-async def voice_create(
-    interaction: discord.Interaction,
-    text: str,
-    voice_name: str,
-    stability: float = 0.5,
-    clarity: float = 0.75,
-    style: float = 0.5
-):
-    await interaction.response.defer()
-    
-    available_voices = get_available_voices()
-    if voice_name not in available_voices:
-        embed = discord.Embed(
-            title="Voice Not Found",
-            description=f"The voice '{voice_name}' was not found. Use `/voice list` to see available voices.",
-            color=0x2B2D31
-        )
-        await interaction.followup.send(embed=embed)
-        return
-        
-    try:
-        voice_settings = VoiceSettings(
-            stability=stability,
-            similarity_boost=clarity,
-            style=style,
-            use_speaker_boost=True
-        )
-        
-        audio = generate(
-            text=text,
-            voice=Voice(
-                voice_id=available_voices[voice_name],
-                settings=voice_settings
-            )
-        )
-        
-        # Save audio to temporary file
-        with open("temp.mp3", "wb") as f:
-            f.write(audio)
-        
-        # Get updated credits
-        credits_info = get_remaining_credits()
-        credits_msg = format_credits_message(credits_info)
-        
-        embed = discord.Embed(
-            title="Voice Generated",
-            description=f"Voice: {voice_name}\n{credits_msg}",
-            color=0x2B2D31
-        )
-        await interaction.followup.send(
-            embed=embed,
-            file=discord.File("temp.mp3")
-        )
-        
-        # Clean up
-        os.remove("temp.mp3")
-        
-    except Exception as e:
-        logger.error(f"Error generating audio: {str(e)}", exc_info=True)
-        await interaction.followup.send(f"Error generating audio: {str(e)}")
-
-# Add autocomplete for voice names
-@voice_create.autocomplete('voice_name')
-async def voice_name_autocomplete(
-    interaction: discord.Interaction,
-    current: str,
-) -> List[app_commands.Choice[str]]:
-    voices = list(get_available_voices().keys())
-    return [
-        app_commands.Choice(name=voice, value=voice)
-        for voice in voices if current.lower() in voice.lower()
-    ][:25]
-    logger.info(f"Received voice command: action={action}, voice_name={voice_name}")
-    await interaction.response.defer()
-
-    if action.lower() == "list":
-        available_voices = get_available_voices()
-        voice_list = "\n".join([f"• {name}" for name in available_voices.keys()])
-        credits_info = get_remaining_credits()
-        credits_msg = f"\n\nCredits remaining: {credits_info['character_count']} / {credits_info['character_limit']}"
-        
-        embed = discord.Embed(
-            title="Available Voices",
-            description=f"{voice_list}\n{credits_msg}",
-            color=0x2B2D31  # Discord dark theme color
-        )
-        await interaction.followup.send(embed=embed)
-        
-    elif action.lower() == "create":
-        if not all([text, voice_name]):
-            embed = discord.Embed(
-                title="Missing Information",
-                description="Please provide both text and voice name. Use `/voice list` to see available voices.",
-                color=0x2B2D31
-            )
-            await interaction.followup.send(embed=embed)
-            return
-            
-        available_voices = get_available_voices()
-        if voice_name not in available_voices:
-            embed = discord.Embed(
-                title="Voice Not Found",
-                description=f"The voice '{voice_name}' was not found. Use `/voice list` to see available voices.",
-                color=0x2B2D31
-            )
-            await interaction.followup.send(embed=embed)
-            return
-            
-        try:
-            voice_settings = VoiceSettings(
-                stability=stability,
-                similarity_boost=clarity,
-                style=style,
-                use_speaker_boost=True
-            )
-            
-            audio = generate(
-                text=text,
-                voice=Voice(
-                    voice_id=available_voices[voice_name],
-                    settings=voice_settings
-                )
-            )
-            
-            # Save audio to temporary file
-            with open("temp.mp3", "wb") as f:
-                f.write(audio)
-            
-            # Get updated credits
-            credits_info = get_remaining_credits()
-            credits_msg = f"Credits remaining: {credits_info['character_count']} / {credits_info['character_limit']}"
-            
-            embed = discord.Embed(
-                title="Voice Generated",
-                description=f"Voice: {voice_name}\n{credits_msg}",
-                color=0x2B2D31
-            )
-            await interaction.followup.send(
-                embed=embed,
-                file=discord.File("temp.mp3")
-            )
-            
-            # Clean up
-            os.remove("temp.mp3")
-            
-        except Exception as e:
-            logger.error(f"Error generating audio: {str(e)}", exc_info=True)
-            await interaction.followup.send(f"Error generating audio: {str(e)}")
-    
-    else:
-        await interaction.followup.send("Invalid action. Use 'list' or 'create'.")
-
-@client.event
-async def on_ready():
-    logger.info(f"Bot is ready and logged in as {client.user}")
-    await tree.sync()
-
 # Gradio interface functions
 def text_to_speech(text, voice_name, stability, clarity, style):
     """Convert text to speech using selected voice and parameters"""
@@ -445,6 +193,9 @@ def speech_to_text(audio_file):
             return "Could not understand audio"
         except sr.RequestError:
             return "Error in speech recognition service"
+        finally:
+            # Clean up temporary wav file
+            os.unlink(wav_path)
 
 def speech_to_speech(audio_file, voice_name, stability, clarity, style):
     """Convert speech to speech by first converting to text, then to speech"""