Push

core/query_params.py

@@ -37,14 +37,23 @@ def go_to_tab(tabs: list[str]):
     params = st.experimental_get_query_params()
     if QueryParams.OPEN_TAB in params:
         try:
-            tab = int(params[QueryParams.OPEN_TAB][0])
-            if 0 <= tab and tab < len(tabs):
-                tab_id = f"tabs-bui3-tab-{tab}"
+            index = int(params[QueryParams.OPEN_TAB][0])
+            if 0 <= index and index < len(TABS):
+                tab = TABS[index]
                 # Click on the tab.
                 js = f"""
                     <script>
-                        const tab = window.parent.document.getElementById('{tab_id}');
-                        tab.click();
+                    function contains(selector, text) {{
+                      const document = window.parent.document;
+                      const elements = document.querySelectorAll(selector);
+                      return Array.from(elements).filter(function(element) {{
+                        return RegExp(text).test(element.innerText);
+                      }});
+                    }}
+                    const tab = contains('button', '{tab}');
+                    if (tab.length) {{
+                        tab[0].click();
+                    }}
                     </script>
                 """
                 st.components.v1.html(js)

views/files.py

@@ -69,7 +69,7 @@ def _render_resources_panel(files: list[Resource]) -> Resource | None:
     if not name:
         return None
     file = filename_to_file[name]
-    # set_tab(RESOURCES)
+    set_tab(RESOURCES)
     return file
 
 

views/load.py

@@ -21,7 +21,7 @@ def _on_file_upload(key):
     try:
         dataset = mlc.Dataset(newfile_name)
         st.session_state[Metadata] = Metadata.from_canonical(dataset.metadata)
-        # save_current_project()
+        save_current_project()
     except mlc.ValidationError as e:
         st.warning(e)
         st.toast(body="Invalid Croissant File!", icon="🔥")

views/splash.py

@@ -4,6 +4,7 @@ import requests
 import streamlit as st
 
 from core.constants import OAUTH_CLIENT_ID
+from core.past_projects import save_current_project
 from core.query_params import set_project
 from core.state import CurrentProject
 from core.state import Metadata
@@ -28,9 +29,7 @@ def render_splash():
 
             def create_new_croissant():
                 st.session_state[Metadata] = Metadata()
-                project = CurrentProject.create_new()
-                st.session_state[CurrentProject] = project
-                set_project(project)
+                save_current_project()
 
             st.button(
                 "Create",
@@ -45,9 +44,7 @@ def render_splash():
                     json = requests.get(url).json()
                     metadata = mlc.Metadata.from_json(mlc.Issues(), json, None)
                     st.session_state[Metadata] = Metadata.from_canonical(metadata)
-                    project = CurrentProject.create_new()
-                    st.session_state[CurrentProject] = project
-                    set_project(project)
+                    save_current_project()
                 except Exception as exception:
                     logging.error(exception)
                     st.error(

views/wizard.py

@@ -5,10 +5,7 @@ import streamlit_nested_layout  # Do not remove this allows nesting columns.
 
 from core.constants import TABS
 from core.past_projects import save_current_project
-from core.query_params import get_project_timestamp
 from core.query_params import go_to_tab
-from core.query_params import QueryParams
-from core.query_params import set_tab
 from core.state import Metadata
 import mlcroissant as mlc
 from views.files import render_files