Update app.py

app.py

@@ -268,17 +268,17 @@ 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')
-        st.sidebar.image(image)
+        #st.sidebar.image(image)
         st.session_state.image_displayed = True
-        #return image
+        return image
 
     def generate_audio():
         sound_file = BytesIO()
         tts = gTTS(result['result'], lang='en')
         tts.write_to_fp(sound_file)
-        st.sidebar.audio(sound_file)
+        #st.sidebar.audio(sound_file)
         st.session_state.sound_played = True
-        #return sound_file
+        return sound_file
 
 
     #st.button(':speaker:', type='primary',on_click=generate_audio)
@@ -342,14 +342,14 @@ if prompt := st.chat_input("How can I help you today?"):
         message_placeholder.markdown(full_response + "▌")
         message_placeholder.markdown(full_response)
 
-    if st.sidebar.button("Display Image"):
+    if st.button("Display Image"):
         a=generate_pdf()
-        message_placeholder.image(a)
+        st.sidebar.image(a)
 
 
-    if st.sidebar.button("Play Sound"):
+    if st.button("Play Sound"):
         x=generate_audio()
-        message_placeholder.audio(x)
+        st.sidebar.audio(x)
     
     st.session_state.messages.append({"role": "assistant", "content": full_response})