fix(modal): temporary fix for incompatible third-party modals

pyproject.toml

@@ -54,7 +54,6 @@ dependencies = [
     "streamlit-option-menu==0.3.6",
     "streamlit-authenticator==0.2.3",
     "streamlit-javascript==0.1.5",
-    "streamlit-modal==0.1.0",
     "progress==1.6",
     "beautifulsoup4==4.12.3",
     "trino==0.327.0",

testgen/ui/components/widgets/modal.py

@@ -2,10 +2,34 @@
 
 import streamlit
 import streamlit.components.v1 as components
-from streamlit_modal import Modal as BaseModal
 
+# This is a custom version of the "streamlit-modal" third-party library
+# The original library is not compatible with streamlit 1.30+
+# https://github.com/teamtv/streamlit_modal/issues/19
+
+# This is temporary until we replace our modals with the new native st.dialog feature
+# https://docs.streamlit.io/develop/api-reference/execution-flow/st.dialog
+
+class Modal:
+
+    def __init__(self, title, key, padding=20, max_width=None):
+        self.title = title
+        self.padding = padding
+        self.max_width = max_width
+        self.key = key
+
+    def is_open(self):
+        return streamlit.session_state.get(f"{self.key}-opened", False)
+
+    def open(self):
+        streamlit.session_state[f"{self.key}-opened"] = True
+        streamlit.rerun()
+
+    def close(self, rerun=True):
+        streamlit.session_state[f"{self.key}-opened"] = False
+        if rerun:
+            streamlit.rerun()
 
-class Modal(BaseModal):
     @contextmanager
     def container(self):
         streamlit.markdown(self._modal_styles(), unsafe_allow_html=True)