audio.py

from elevenlabs import voices, generate, set_api_key, UnauthenticatedRateLimitError
import numpy as np
import gradio as gr
import os

def pad_buffer(audio):
    # Pad buffer to multiple of 2 bytes
    buffer_size = len(audio)
    element_size = np.dtype(np.int16).itemsize
    if buffer_size % element_size != 0:
        audio = audio + b'\0' * (element_size - (buffer_size % element_size))
    return audio 

def generate_audio(text):
    try:
        eleven_api_key = os.getenv('ELEVEN_API_KEY')
        set_api_key(eleven_api_key)

        audio = generate(
            text[:250], # Limit to 250 characters
            voice="Arnold", 
            model="eleven_multilingual_v1"
        )
        return (44100, np.frombuffer(pad_buffer(audio), dtype=np.int16))
    except UnauthenticatedRateLimitError as e:
        raise gr.Error("Thanks for trying out ElevenLabs TTS! You've reached the free tier limit. Please provide an API key to continue.") 
    except Exception as e:
        raise gr.Error(e)