Update app.py

app.py

@@ -38,19 +38,20 @@ def format_prompt(message, history, system_prompt=None):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-def generate_voice(text, voice_name, api_key):
+def generate_voice(text, voice_name, api_key, tts_password):
     set_api_key(elevenlabs_api_key) #set API key
-    try:
-        audio = generate(
-            voice=voice_name, 
-            model="eleven_multilingual_v2"
-        )
-        print("Generating voice...")
-        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)
+    if tts_password == ttsPassword:
+        try:
+            audio = generate(
+                voice=voice_name, 
+                model="eleven_multilingual_v2"
+            )
+            print("Generating voice...")
+            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)
 
 def generate(
     prompt, history, enable_tts, tts_password, temperature=0.2, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0
@@ -142,32 +143,19 @@ def generate(
         # Check if the checkbox is selected and the correct password is provided
 
 
-checkbox = gr.Checkbox(label="Enable TTS")
 passwordInput = gr.Textbox(label="TTS Password", type="password")
-passwordConfirm = gr.Button(value="Confirm")
 mychatbot = gr.Chatbot(avatar_images=["./user.png", "./stella.jpg"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True)
 
-with mychatbot.change:
-    enable_tts = checkbox
-
-    def ttsEnable():        
-        if checkbox == True and passwordInput == ttsPassword:
-            enable_tts = True
-
-        elif checkbox == True and passwordInput != ttsPassword:
-            gr.Warning("Incorrect Password!")
-            
-    passwordConfirm.click(fn=ttsEnable)
-    
-    if enable_tts == True:
-
 chat = gr.ChatInterface(
         fn=generate,
         chatbot=mychatbot,
         title="Stella ",
         retry_btn=None,
         undo_btn=None,
-        additional_inputs=[checkbox, passwordInput],
+        additional_inputs=[passwordInput],
     )
 
+with chat:
+    generate_voice(passwordInput);
+
 chat.launch(show_api=False)