Update app.py

app.py

@@ -24,7 +24,6 @@ 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()
 
@@ -169,8 +168,6 @@ if prompt := st.chat_input("How can I help you today?"):
             pix.save(output_image, "png")
             pdf_document.close()
         pdf_page_to_image('/home/user/app/pdf2image/output.pdf', page_number, '/home/user/app/pdf2image/output.png')
-        pdfPath = '/home/user/app/pdf2image/output.png'
-        return pdfPath
         #image = Image.open('/home/user/app/pdf2image/output.png')
         #message_placeholder.image(image)
         #st.session_state.reference = True
@@ -183,23 +180,38 @@ if prompt := st.chat_input("How can I help you today?"):
             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")
-        audioPath = "/home/user/app/audio/audio.wav"
-        return audioPath
+
+    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 + "▌")
         message_placeholder.markdown(full_response)
 
-    # if button('Audio', key='audio'):
-    #     ap = generate_audio()
-    #     st.sidebar.audio(ap)
 
-    if button('Ref', key='ref'):
-        pp = generate_pdf()
-        st.sidebar.image(pp)
+    if "reference" not in st.session_state:
+        st.session_state.reference = False
+    if "audio" not in st.session_state:
+        st.session_state.audio = False
+
+
+    with st.sidebar:
+        choice = st.radio("References",["Reference"], index=None,)
 
+        if choice == 'Reference':
+            generate_pdf()
+            st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
+            st.image(st.session_state['reference'])
 
+        # if choice == 'TTS':
+        #     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})