Update app.py

app.py

@@ -23,6 +23,7 @@ import fitz
 from PIL import Image
 from langchain.vectorstores import FAISS
 import transformers
+from pydub import AudioSegment
 
 user_session_id = uuid.uuid4()
 
@@ -178,6 +179,8 @@ if prompt := st.chat_input("How can I help you today?"):
         with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
             tts = gTTS(result['result'], lang='en', tld='co.in')
             tts.write_to_fp(sound_file)
+        sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
+        sound.export("/home/user/app/audio/audio.wav", format="wav")
 
     for item in output:
         full_response += item
@@ -200,8 +203,8 @@ if prompt := st.chat_input("How can I help you today?"):
 
         if choice == 'TTS':
             generate_audio()
-            st.session_state['audio'] = '/home/user/app/audio/audio.mp3'
-            st.audio(st.session_state['audio'], format='audio/mp3')
+            st.session_state['audio'] = '/home/user/app/audio/audio.wav'
+            st.audio(st.session_state['audio'])
 
     st.session_state.messages.append({"role": "assistant", "content": full_response})