Update app.py

app.py

@@ -24,6 +24,7 @@ from PIL import Image
 from langchain.vectorstores import FAISS
 import transformers
 from pydub import AudioSegment
+from streamlit_extras.stateful_button import button
 
 user_session_id = uuid.uuid4()
 
@@ -185,32 +186,32 @@ 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'] = ''
+    # st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
+    # st.session_state['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
+    # if "reference" not in st.session_state:
+    #     st.session_state.reference = False
+    # if "audio" not in st.session_state:
+    #     st.session_state.audio = False
     
-    #choice = st.radio(label ="References and TTS", options =["Reference" , 'TTS'])
+    # #choice = st.radio(label ="References and TTS", options =["Reference" , 'TTS'])
 
-    if 'onr' not in st.session_state:
-        st.session_state.onr = False
+    # if 'onr' not in st.session_state:
+    #     st.session_state.onr = False
     
-    def click_button_onr():
-        st.session_state.onr = True
-        if st.session_state.onr:
-            with st.sidebar:
-                st.write('ref')
-                generate_pdf()
-                st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
-                st.image(st.session_state['reference'])
+    # def click_button_onr():
+    #     st.session_state.onr = True
+    #     if st.session_state.onr:
+    #         with st.sidebar:
+    #             st.write('ref')
+    #             generate_pdf()
+    #             st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
+    #             st.image(st.session_state['reference'])
         
-    st.button('Ref', on_click=click_button_onr)
+    # st.button('Ref', on_click=click_button_onr)
     
 
     
@@ -224,24 +225,40 @@ if prompt := st.chat_input("How can I help you today?"):
     #        st.image(st.session_state['reference'])
 
 
-    if 'ontts' not in st.session_state:
-        st.session_state.ontts = False
+    # if 'ontts' not in st.session_state:
+    #     st.session_state.ontts = False
     
-    def click_button_ontts():
-        st.session_state.ontts = True
-        if st.session_state.ontts:
-            with st.sidebar:
-                st.write('audio')
-                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'])
+    # def click_button_ontts():
+    #     st.session_state.ontts = True
+    #     if st.session_state.ontts:
+    #         with st.sidebar:
+    #             st.write('audio')
+    #             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.button('Audio', on_click=click_button_ontts)
-    
+    # st.button('Audio', on_click=click_button_ontts)
+
+    if button('Reference and TTS'):
+        if button('Reference'):
+            with st.sidebar:
+                generate_pdf()
+                st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
+                st.image(st.session_state['reference'])
+        if button('TTS'):
+            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'])
+