Update app.py

app.py

@@ -172,8 +172,7 @@ if prompt := st.chat_input("How can I help you today?"):
         #message_placeholder.image(image)
         #st.session_state.reference = True
 
-    st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
-    st.session_state['audio'] = ''
+
 
     def generate_audio():
         with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
@@ -182,6 +181,9 @@ if prompt := st.chat_input("How can I help you today?"):
         sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
         sound.export("/home/user/app/audio/audio.wav", format="wav")
 
+    st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
+    st.session_state['audio'] = ''
+    
     for item in output:
         full_response += item
         message_placeholder.markdown(full_response + "▌")
@@ -211,7 +213,7 @@ if prompt := st.chat_input("How can I help you today?"):
             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.audio(sound_file)
 
     st.session_state.messages.append({"role": "assistant", "content": full_response})