Update audio.py

audio.py

@@ -1,9 +1,24 @@
-from elevenlabs import generate, save
+from elevenlabs import voices, generate, set_api_key, UnauthenticatedRateLimitError
+import numpy as np
+import gradio as gr
 
-def generate_audio(input_text):
-	audio = generate(
-	  text=input_text,
-	  voice="Rachel",
-	  model="eleven_monolingual_v1"
-	)	
-	return audio
+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:
+        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)