Update app.py

app.py

@@ -213,10 +213,10 @@ if prompt := st.chat_input("How can I help you today?"):
         
         output = [result['result']]
 
-    for item in output:
-        full_response += item
-        message_placeholder.markdown(full_response + "▌")
-        message_placeholder.markdown(full_response)    
+    # for item in output:
+    #     full_response += item
+    #     message_placeholder.markdown(full_response + "▌")
+    #     message_placeholder.markdown(full_response)    
     #st.write(repr(result['source_documents'][0].metadata['page']))
     #st.write(repr(result['source_documents'][0]))
 
@@ -266,15 +266,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
 
     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
 
 
     #st.button(':speaker:', type='primary',on_click=generate_audio)
@@ -322,11 +324,7 @@ if prompt := st.chat_input("How can I help you today?"):
     #st.button("Display Image", on_click=generate_pdf)
     #st.button("Play Sound", on_click=generate_audio)
 
-    if st.sidebar.button("Display Image"):
-        generate_pdf()
 
-    if st.sidebar.button("Play Sound"):
-        generate_audio()
     
     # # Check if the image has been displayed and display it if it has not
     # if not st.session_state.image_displayed:
@@ -335,7 +333,22 @@ if prompt := st.chat_input("How can I help you today?"):
     # # Check if the sound has been played and play it if it has not
     # if not st.session_state.sound_played:
     #     generate_audio()
-        
+
+
+    for item in output:
+        full_response += item
+        message_placeholder.markdown(full_response + "▌")
+        message_placeholder.markdown(full_response)
+
+    if st.sidebar.button("Display Image"):
+        a=generate_pdf()
+        message_placeholder.image(a)
+
+
+    if st.sidebar.button("Play Sound"):
+        x=generate_audio()
+        message_placeholder.audio(x)
+    
     st.session_state.messages.append({"role": "assistant", "content": full_response})