Update app.py

app.py

@@ -23,8 +23,8 @@ def format_prompt(message, history):
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
-def text_to_speech(text):
-    tts = gTTS(text=text, lang='es')
+def text_to_speech(text, speed=1.5):
+    tts = gTTS(text=text, lang='es', speed=speed)
     tts.save('output.mp3')
     return 'output.mp3'
 
@@ -50,9 +50,9 @@ def generate(
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
     response = " ".join(response.token.text for response in stream)
 
-    st.markdown(f"<p style='text-align: justify;'>{response}</p>", unsafe_allow_html=True)
-    audio_file = text_to_speech(response)
-    st.audio(audio_file, format="audio/mp3", start_time=0)
+    st.write(response, style='white-space: pre-line;')
+    audio_file = text_to_speech(response, speed=1.2) 
+    st.audio(audio_file, format="audio/mp3", start_time=0, key='audio')
 
     return response