Update my_pages/information_loss.py

my_pages/information_loss.py

@@ -1,6 +1,7 @@
 import pandas as pd
 import streamlit as st
 from utils import go_to
+from streamlit_modal import Modal
 
 ALL_FEATURES = [
     ("Liquid Assets", "Liquid Assets", None), 
@@ -77,7 +78,7 @@ def render():
                         st.session_state.show_message = (feature[2])
                     st.rerun()
 
-    # Centered message
+    modal = Modal("My Modal")
     if "show_message" in st.session_state:
         del st.session_state.show_message
 
@@ -85,10 +86,13 @@ def render():
         #     f"<div style='text-align:center; color:#c0392b; font-weight:600; margin:14px 0;'>{st.session_state.show_message}</div>",
         #     unsafe_allow_html=True,
         # )
-        with st.modal("Notice"):
-            st.write("This is your message inside a modal pop-up.")
+        modal.open()
+    
+    if modal.is_open():
+        with modal.container():
+            st.write("Here's the message!")
             if st.button("OK"):
-                st.rerun()
+                modal.close()
 
     # st.markdown("---")