Update app.py

app.py

@@ -172,14 +172,12 @@ 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'] = ''
+    st.session_state['audio'] = '/home/user/app/audio/audio.mp3'
 
     
     def generate_audio():
-        sound_file = BytesIO()
         tts = gTTS(result['result'], lang='en')
-        tts.write_to_fp(sound_file)
-        return sound_file
+        tts.save('/home/user/app/audio/audio.mp3')
 
     for item in output:
         full_response += item
@@ -201,9 +199,9 @@ if prompt := st.chat_input("How can I help you today?"):
             st.image(st.session_state['reference'])
 
         if choice == 'TTS':
-            x=generate_audio()
-            st.session_state['audio'] = x
-            st.image(st.session_state['audio'])
+            generate_audio()
+            st.session_state['audio'] = '/home/user/app/audio/audio.mp3'
+            st.audio(st.session_state['audio'])
 
     st.session_state.messages.append({"role": "assistant", "content": full_response})