Update app.py

app.py

@@ -122,12 +122,21 @@ def generate(
         # Play the audio file
         elevenlabs.play(audio)
 
+    # Return the audio file as a numpy array
+        return audio
+    else:
+        # Return None if the TTS is not enabled or the password is incorrect
+        return None
+
 mychatbot = gr.Chatbot(
     avatar_images=["./user.png", "./stella.jpg"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True)
 
 checkbox = gr.Checkbox(label="Enable TTS") # remove the fn parameter
 passwordInput = gr.Textbox(label="TTS Password", type="password")
 
+# Define the output component
+audio_output = gr.Audio(label="Audio") # Add the audio component as an output
+
 # Create a Gradio interface with the TTS checkbox and password input
 mychatbot = gr.Chatbot(
     avatar_images=["./user.png", "./stella.jpg"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True)
@@ -138,7 +147,8 @@ demo = gr.ChatInterface(
     title="Stella 🌸",
     retry_btn=None,
     undo_btn=None,
-    additional_inputs=[checkbox, passwordInput]
+    additional_inputs=[checkbox, passwordInput],
+    outputs=audio_output # Add the output
 )