Update app.py

app.py

@@ -167,11 +167,12 @@ 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')
-        message_placeholder.image(image)
-        st.session_state.reference = True
-
+        #image = Image.open('/home/user/app/pdf2image/output.png')
+        #message_placeholder.image(image)
+        #st.session_state.reference = True
 
+    st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
+    
     def generate_audio():
         sound_file = BytesIO()
         tts = gTTS(result['result'], lang='en')
@@ -186,18 +187,18 @@ if prompt := st.chat_input("How can I help you today?"):
         message_placeholder.markdown(full_response)
 
 
-    # if "reference" not in st.session_state:
-    #     st.session_state.reference = 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
 
-    # with st.sidebar:
-    #     choice = st.radio("References and TTS",
-    #         ["Reference", "Text-to-Speech"])
+    with st.sidebar:
+        choice = st.radio("References and TTS",["Reference"])
         
-    #     if choice == 'Reference':
-    #         st.session_state.reference = True
-    #         generate_pdf()       
+        if choice == 'Reference':
+            generate_pdf()
+            st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
+            st.image(st.session_state['reference'])
 
     #     if choice == ':speaker:':
     #         generate_audio()
@@ -205,33 +206,4 @@ if prompt := st.chat_input("How can I help you today?"):
 
 
 
-    from streamlit.report_thread import get_report_ctx
-    from streamlit.server.server import Server
-
-    image_url = "/home/user/app/pdf2image/output.png"  # Replace with your image URL
-    button_label = "Toggle Image"
-    
-    # Check if the SessionState object exists, otherwise create a new one
-    ctx = get_report_ctx()
-    session_id = ctx.session_id
-    if session_id not in Server.get_current()._session_info:
-        Server.get_current()._session_info[session_id] = {}
-    
-    session_state = Server.get_current()._session_info[session_id]
-    
-    # Create a button to toggle the image visibility
-    toggle_button = st.button(button_label, key=session_id)
-    
-    # Check if the toggle button is pressed
-    if toggle_button:
-        # Store the current state of the button in the session state
-        session_state[session_id] = not session_state.get(session_id, False)
-    
-    # Check if the image should be displayed based on the session state
-    if session_state.get(session_id, False):
-        # Display the image
-        st.image(image_url, caption='Image Caption')
-            
-    st.session_state.messages.append({"role": "assistant", "content": full_response})
-