Update app.py

app.py

@@ -207,10 +207,10 @@ if prompt := st.chat_input("How can I help you today?"):
     st.button('Ref', on_click=click_button_onr)
     
     if st.session_state.onr:
-        # The message and nested widget will remain on the page
-            generate_pdf()
-            st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
-            st.image(st.session_state['reference'])
+        st.write('ref')
+        generate_pdf()
+        st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
+        st.image(st.session_state['reference'])
     
     #onr = st.toggle('Reference',key='onr')
     #if choice == 'Reference':
@@ -231,6 +231,7 @@ if prompt := st.chat_input("How can I help you today?"):
     st.button('Audio', on_click=click_button_ontts)
     
     if st.session_state.ontts:
+        st.write('audio')
         # The message and nested widget will remain on the page
         with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
             tts = gTTS(result['result'], lang='en', tld = 'co.in')
@@ -239,7 +240,7 @@ if prompt := st.chat_input("How can I help you today?"):
         sound.export("/home/user/app/audio/audio.wav", format="wav")
         st.session_state['audio'] = '/home/user/app/audio/audio.wav'
         st.audio(st.session_state['audio'])
-
+        
         
     # ontts = st.toggle('Audio')
     # #if choice == 'TTS':