Update app.py

app.py

@@ -173,7 +173,7 @@ if prompt := st.chat_input("How can I help you today?"):
         #st.session_state.reference = True
 
     st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
-    st.session_state['audio'] = '/home/user/app/audio/beep-06.mp3'
+    st.session_state['audio'] = ''
 
     def generate_audio():
         with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
@@ -201,9 +201,16 @@ if prompt := st.chat_input("How can I help you today?"):
             st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
             st.image(st.session_state['reference'])
 
+        # if choice == 'TTS':
+        #     generate_audio()
+        #     st.session_state['audio'] = '/home/user/app/audio/audio.wav'
+        #     st.audio(st.session_state['audio'])
+
         if choice == 'TTS':
-            generate_audio()
-            st.session_state['audio'] = '/home/user/app/audio/audio.wav'
+            sound_file = BytesIO()
+            tts = gTTS(result['result'], lang='en', tld = 'co.in')
+            tts.write_to_fp(sound_file)
+            st.session_state['audio'] = sound_file
             st.audio(st.session_state['audio'])
 
     st.session_state.messages.append({"role": "assistant", "content": full_response})