Update app.py

app.py

@@ -2,6 +2,12 @@ from huggingface_hub import InferenceClient
 import gradio as gr
 from datetime import datetime
 import pytz
+import os
+
+elevenlabsKey = os.environ.get("11_READ_TOKEN", None)
+voice = os.environ.get("VOICE_ID", None)
+model = "eleven_multilingual_v2"
+ttsPassword = os.environ.get("TTS_PASSWORD", None)
 
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
@@ -106,10 +112,30 @@ def generate(
         yield output
     return output
 
+    
+# Check if the checkbox is selected and the correct password is provided
+if use_checkbox and password == ttsPassword:
+    # Generate the audio file
+    audio = elevenlabs.generate(output, voice, model, api_key=elevenlabsKey)
+        
+    # Play the audio file
+    elevenlabs.play(audio)
+    
+# Your existing code
+
+# Adding a checkbox and password input to the Gradio interface
+checkbox = gr.Checkbox(component=generate, label="Use Text-to-Speech", default=False)
+password_input = gr.Textbox("Password", default=None, password=True, placeholder="Enter password")
+
 mychatbot = gr.Chatbot(
     avatar_images=["./user.png", "./stella.jpg"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True)
 
 demo = gr.ChatInterface(fn=generate, 
+                        inputs=[
+                            gr.Textbox("User Input", default=None),
+                            gr.Conditional(password_input, condition=lambda use_checkbox: use_checkbox),
+                            checkbox
+                        ],
                         chatbot=mychatbot,
                         title="Stella 🌸",
                         retry_btn=None,