Update app.py

app.py

@@ -106,28 +106,22 @@ def launch_bot():
                 # st.write("Mistral:" ) #Needs finishing
               #  else:
                 st.write(response) 
-                
+
+                audio_result = st.button("Convert to Audio 🔊") 
+                if audio_result:
+                   # st.session_state.messages.append({"role": "user", "content": "Convert to Audio 🔊"})
+                       # with st.chat_message("user"):
+                           # st.write("Convert to Audio 🔊")
+                    with st.popover("Open Audio"):
+                        sound_file = BytesIO()
+                        tts = gTTS(response, lang='en')
+                        tts.write_to_fp(sound_file)
+                            
+                        st.audio(sound_file)
+                        
                 message = {"role": "assistant", "content": response}
                 st.session_state.messages.append(message)
 
-    # If assistant has most recently reaponded create audio of response 
-    # if st.session_state.messages[-1]["role"] == "assistant":
-        audio_result = st.button("Convert to Audio 🔊") 
-        if audio_result:
-            st.session_state.messages.append({"role": "user", "content": "Convert to Audio 🔊"})
-            with st.chat_message("user"):
-                st.write("Convert to Audio 🔊")
-            with st.popover("Open Audio"):
-                message2 = {"role": "assistant", "content": "Converting text to speech. This may take a few minutes."}
-                # Create temporary file for audio
-                with tempfile.NamedTemporaryFile(suffix=".mp3") as temp_file:
-                    tts = gTTS('Add text-to-speech to your app', lang='en')
-                    tts.write_to_fp(temp_file)
-                    temp_file.seek(0)  # rewind the file pointer
-    
-                    # Play the audio file
-                    st.audio(temp_file)
-
 
                 
 if __name__ == "__main__":