Update app.py

app.py

@@ -186,22 +186,51 @@ 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 "audio" not in st.session_state:
-        st.session_state.audio = False
-
-    with st.sidebar:
-        choice = st.radio("References and TTS",
-            ["Reference", "Text-to-Speech"])
+    # 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"])
         
-        if choice == 'Reference':
-            st.session_state.reference = True
-            generate_pdf()       
-
-        if choice == ':speaker:':
-            generate_audio()
-            st.session_state.audio = True
+    #     if choice == 'Reference':
+    #         st.session_state.reference = True
+    #         generate_pdf()       
+
+    #     if choice == ':speaker:':
+    #         generate_audio()
+    #         st.session_state.audio = True
+
+
+
+    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})