Update app.py

app.py

@@ -169,13 +169,15 @@ if prompt := st.chat_input("How can I help you today?"):
         pdf_page_to_image('/home/user/app/pdf2image/output.pdf', page_number, '/home/user/app/pdf2image/output.png')
         image = Image.open('/home/user/app/pdf2image/output.png')
         message_placeholder.image(image)
+        st.session_state.reference = True
+
 
     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.sound_played = True
+        st.session_state.audio = True
 
 
     for item in output:
@@ -183,11 +185,22 @@ if prompt := st.chat_input("How can I help you today?"):
         message_placeholder.markdown(full_response + "▌")
         message_placeholder.markdown(full_response)
 
-    st.button('Reference',on_click=generate_pdf)
-    st.button(':speaker:',on_click=generate_audio)
 
+    if "reference" not in st.session_state:
+        st.session_state.reference = False
+    if "audio" not in st.session_state:
+        st.session_state.audio = False
 
+    x = st.button('Reference',key='reference')
+    if x or st.session_state.reference:
+        generate_pdf()
+        st.session_state.reference = True
+        
 
+    y  = st.button(':speaker:',key='audio')
+    if y or st.session_state.audio:
+        generate_audio()
+        st.session_state.audio = True
 
     st.session_state.messages.append({"role": "assistant", "content": full_response})