Update app.py

app.py

@@ -172,14 +172,14 @@ 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'] = ''
+
     
     def generate_audio():
         sound_file = BytesIO()
         tts = gTTS(result['result'], lang='en')
         tts.write_to_fp(sound_file)
-        message_placeholder.audio(sound_file)
-        st.session_state.audio = True
-
+        return sound_file
 
     for item in output:
         full_response += item
@@ -193,17 +193,18 @@ if prompt := st.chat_input("How can I help you today?"):
     #     st.session_state.audio = False
 
     with st.sidebar:
-        choice = st.radio("References and TTS",["Reference"])
+        choice = st.radio("References and TTS",["Reference" , 'TTS'])
         
         if choice == 'Reference':
             generate_pdf()
             st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
             st.image(st.session_state['reference'])
 
-    #     if choice == ':speaker:':
-    #         generate_audio()
-    #         st.session_state.audio = True
-
+        if choice == 'TTS':
+            x=generate_audio()
+            st.session_state['audio'] = x
+            st.image(st.session_state['audio'])
 
+    st.session_state.messages.append({"role": "assistant", "content": full_response})