Replace data load page with dbt data

src/pages/98_Load_Data copy.py

@@ -1,33 +0,0 @@
-import streamlit as st
-
-from config import DEFAULT_ICON
-from shared_page import common_page_config
-
-from dbt_data_client import run_dbt_build, get_current_tables, execute_db_command_df
-
-
-def get_page():
-    page_title = "Data Loader - dbt"
-    st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
-    common_page_config()
-    st.title(page_title)
-
-    if st.button("Refresh Data"):
-        run_dbt_build()
-
-    current_tables_list = get_current_tables()
-
-    if selected_table := st.selectbox("Describe a table:", current_tables_list, index=0):
-        describe_df = execute_db_command_df(f"DESCRIBE {selected_table}")
-        st.dataframe(
-            describe_df,
-            hide_index=True,
-            use_container_width=True,
-        )
-
-        if st.checkbox("Explore data"):
-            st.dataframe(execute_db_command_df(f"SELECT * FROM {selected_table} LIMIT 50"))
-
-
-if __name__ == "__main__":
-    get_page()

src/pages/98_Load_Data.py

@@ -1,26 +1,24 @@
-import duckdb
 import streamlit as st
 
 from config import DEFAULT_ICON
 from shared_page import common_page_config
 
-from queries.nflverse.github_data import load_assets, get_current_tables
+from dbt_data_client import run_dbt_build, get_current_tables, execute_db_command_df
 
 
 def get_page():
-    page_title = "Data Loader"
+    page_title = "Data Loader - dbt"
     st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
     common_page_config()
     st.title(page_title)
 
-    current_tables_list = get_current_tables()
-
     if st.button("Refresh Data"):
-        load_assets()
-        st.rerun()
+        run_dbt_build()
+
+    current_tables_list = get_current_tables()
 
     if selected_table := st.selectbox("Describe a table:", current_tables_list, index=0):
-        describe_df = duckdb.sql(f"DESCRIBE {selected_table}").df()
+        describe_df = execute_db_command_df(f"DESCRIBE {selected_table}")
         st.dataframe(
             describe_df,
             hide_index=True,
@@ -28,7 +26,7 @@ def get_page():
         )
 
         if st.checkbox("Explore data"):
-            st.dataframe(duckdb.sql(f"SELECT * FROM {selected_table} LIMIT 50").df())
+            st.dataframe(execute_db_command_df(f"SELECT * FROM {selected_table} LIMIT 50"))
 
 
 if __name__ == "__main__":