Update app.py

app.py

@@ -8,7 +8,7 @@ Created on Thu Sep 21 22:17:43 2023
 import streamlit as st
 import sentencepiece
 from gtts import gTTS
-from io import BytesIO
+import base64
 from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
 
 model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/T5_MineAI_Prototype").to("cpu")
@@ -27,6 +27,8 @@ if 'messages' not in st.session_state:
 
 st.subheader("Mine AI")
 
+audio_stream=r"sample_audio.mp3"
+
 for message in st.session_state.messages:
     with st.chat_message(message['role']):
         st.markdown(message['content'])
@@ -38,9 +40,13 @@ if st.session_state.messages==[]:
         with st.chat_message("assistant"):
             st.markdown(gr)
 
-        tts=gTTS(gr)
-        tts.write_to_fp(sound_file)
-        st.audio(sound_file)
+        tts=gTTS(prompt)
+        tts.save(audio_stream)
+        with open(audio_stream, 'rb') as file:
+            audio_data=file.read()
+        audio_base64 = base64.b64encode(audio_data).decode('utf-8')
+        audio_tag = f'<audio autoplay="true" src="data:audio/mp3;base64,{audio_base64}">'
+        st.markdown(audio_tag, unsafe_allow_html=True)
     
         st.session_state.messages.append({
             'role':'assistant',
@@ -62,9 +68,13 @@ if prompt:=st.chat_input("Enter your query"):
     with st.chat_message("assistant"):
         st.markdown(response)
 
-    tts=gTTS(response)
-    tts.write_to_fp(sound_file)
-    st.audio(sound_file)
+    tts=gTTS(prompt)
+    tts.save(audio_stream)
+    with open(audio_stream, 'rb') as file:
+        audio_data=file.read()
+    audio_base64 = base64.b64encode(audio_data).decode('utf-8')
+    audio_tag = f'<audio autoplay="true" src="data:audio/mp3;base64,{audio_base64}">'
+    st.markdown(audio_tag, unsafe_allow_html=True)
         
     st.session_state.messages.append({
         'role':'assistant',