Revert "chore: disable deprecated streamlit features"

This reverts commit 9dff5cc.
alphanome-ai · Jun 9, 2024 · bbe0853 · bbe0853
1 parent a095800
commit bbe0853
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 13 deletions.
diff --git a/dev_utils/dashboard_app/app.py b/dev_utils/dashboard_app/app.py
@@ -53,7 +53,8 @@
 ### Persistence ###
 ###################
 
-url_params_queries = st.query_params.get_all("q")
+url_query_params = st.experimental_get_query_params()
+url_params_queries = url_query_params.get("q", [])
 
 if "select_reports__queries" not in st.session_state:
     default = ""
@@ -71,7 +72,7 @@
 
 if "view_parsed__filter_by_text" not in st.session_state:
     default = None
-    if url_filter_by_text := st.query_params.get_all(URL_PARAM_KEY_FILTER_BY_TEXT):
+    if url_filter_by_text := url_query_params.get(URL_PARAM_KEY_FILTER_BY_TEXT, []):
         default = url_filter_by_text[0]
     st.session_state["view_parsed__filter_by_text"] = default
 
@@ -102,9 +103,9 @@ def deserialize(cls, name: str):
 
 default_nav_bar_selection = NavbarItems.SELECT_REPORTS.value
 try:
-    if URL_PARAM_KEY in st.query_params:
+    if URL_PARAM_KEY in url_query_params:
         default_nav_bar_selection = NavbarItems.deserialize(
-            st.query_params[URL_PARAM_KEY],
+            url_query_params[URL_PARAM_KEY][0]
         ).value
 except Exception as e:
     st.toast(

diff --git a/dev_utils/dashboard_app/streamlit_utils.py b/dev_utils/dashboard_app/streamlit_utils.py
@@ -132,7 +132,7 @@ def get_style_modify_decoration():
 
 
 def get_style_change_top_page_margin():
-    return "div[class^='block-container'] { padding-top: 2.8rem; }".strip()
+    return "div[class^='block-container'] { padding-top: 2rem; }".strip()
 
 
 def st_keep(key, value_func=None) -> None:

diff --git a/dev_utils/dashboard_app/view_parsed/semantic_elements.py b/dev_utils/dashboard_app/view_parsed/semantic_elements.py
@@ -34,10 +34,7 @@ def render_view_parsed_semantic_elements(
     do_show_nested_composite_elements: bool,
     are_any_elements_filtered: bool,
 ):
-    url_params = st.query_params
-    url_params = (
-        {}
-    )  # TODO: Temporarily disabled until https://github.com/alphanome-ai/sec-parser/issues/85 is solved
+    url_params = st.experimental_get_query_params()
     new_url_params = []
     pagination_size = None
     do_use_pagination = len(elements) >= PAGINATION_DISABLE_THRESHOLD

diff --git a/dev_utils/dashboard_app/view_parsed/view_parsed.py b/dev_utils/dashboard_app/view_parsed/view_parsed.py
@@ -45,8 +45,7 @@ def render_view_parsed():
         return
     metadatas: list[FilingMetadata] = st.session_state.select_reports__report_metadatas
 
-    url_params = st.query_params
-    url_params = {} # TODO: Temporarily disabled until https://github.com/alphanome-ai/sec-parser/issues/85 is solved
+    url_params = st.experimental_get_query_params()
     new_url_params = []
 
     #################
@@ -103,8 +102,7 @@ def serialize(self):
         def deserialize(cls, name: str):
             return ViewParsedItems[name.upper()]
 
-    url_query_params = st.query_params
-    url_query_params = {} # TODO: Temporarily disabled until https://github.com/alphanome-ai/sec-parser/issues/85 is solved
+    url_query_params = st.experimental_get_query_params()
     default_nav_bar_selection = ViewParsedItems.SEMANTIC_ELEMENTS.value
     try:
         if URL_PARAM_KEY in url_query_params: