Update app.py

app.py

@@ -116,10 +116,7 @@ st.session_state["llm_model"] = model_name_or_path
 
 if "messages" not in st.session_state:
     st.session_state.messages = []
-if "image_displayed" not in st.session_state:
-    st.session_state.image_displayed = False
-if "sound_played" not in st.session_state:
-    st.session_state.sound_played = False
+
 
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
@@ -171,11 +168,8 @@ 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')
-        # image = Image.open('/home/user/app/pdf2image/output.png')
-        # st.image(image)
-        # st.session_state.image_displayed = True
-        path = '/home/user/app/pdf2image/output.png'
-        return path
+        image = Image.open('/home/user/app/pdf2image/output.png')
+        st.image(image)
 
     def generate_audio():
         sound_file = BytesIO()
@@ -190,25 +184,11 @@ if prompt := st.chat_input("How can I help you today?"):
         message_placeholder.markdown(full_response + "▌")
         message_placeholder.markdown(full_response)
 
-    def display_image(image_path):
-        """Displays an image at the given path."""
-        st.image(image_path)
-
-
-    # Create a button to display the image.
-    if st.button("Display Image"):
-        # Get the image path from the session state dictionary.
-        image_path = st.session_state.get("image_path")
+    st.button('Reference',on_click=generate_pdf)
+    st.button(':speaker:',on_click=generate_audio)
 
-        # If the image path is not set, set it to the default image path.
-        if image_path is None:
-            image_path = generate_pdf()
 
-        # Display the image.
-        display_image(image_path)
 
-    # Store the image path in the session state dictionary.
-    st.session_state["image_path"] = image_path
 
     st.session_state.messages.append({"role": "assistant", "content": full_response})