Update app.py

app.py

@@ -198,25 +198,60 @@ if prompt := st.chat_input("How can I help you today?"):
     
     #choice = st.radio(label ="References and TTS", options =["Reference" , 'TTS'])
 
-    onr = st.toggle('Reference')
-    #if choice == 'Reference':
-    if onr:
-        with st.sidebar:
+    if 'onr' not in st.session_state:
+        st.session_state.onr = False
+    
+    def click_button_onr():
+        st.session_state.onr = True
+        
+    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'])
+    
+    #onr = st.toggle('Reference',key='onr')
+    #if choice == 'Reference':
+    #if onr:
+    #    st.session_state['onr'] = True
+    #    with st.sidebar:
+    #        generate_pdf()
+    #        st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
+    #        st.image(st.session_state['reference'])
+
+
+    if 'ontts' not in st.session_state:
+        st.session_state.ontts = False
+    
+    def click_button_ontts():
+        st.session_state.ontts = True
+        
+    st.button('Audio', on_click=click_button_ontts)
+    
+    if st.session_state.ontts:
+        # 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')
+            tts.write_to_fp(sound_file)
+        sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
+        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':
-    if ontts:
-        with st.sidebar:
-            with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
-                tts = gTTS(result['result'], lang='en', tld = 'co.in')
-                tts.write_to_fp(sound_file)
-            sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
-            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':
+    # if ontts:
+    #     with st.sidebar:
+    #         with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
+    #             tts = gTTS(result['result'], lang='en', tld = 'co.in')
+    #             tts.write_to_fp(sound_file)
+    #         sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
+    #         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'])
         
     st.session_state.messages.append({"role": "assistant", "content": full_response})