2. update streamlit

streamlitapp.py

@@ -1,11 +1,11 @@
 from functools import partial
 from pathlib import Path
+import os
 
 import kangas as kg
 import streamlit as st
 import streamlit.components.v1 as components
 from datasets import load_dataset
-import socket
 
 proj_dir = Path(__file__).parent
 
@@ -17,16 +17,18 @@ st.set_page_config(layout="wide")
 st.markdown("1. Select dataset of your choice")
 
 def kangas_fn(dataset_repo):
-    repo_wo_slash = dataset_repo.replace('/', '__') + '.datagrid'
-    dg_file_name = repo_wo_slash + '.datagrid'
-    with st.spinner("Loading Dataset..."):
+    dg_file_name = dataset_repo.replace('/', '__') + '.datagrid'
+    dg_full_path = str(proj_dir / 'datagrids' / dg_file_name)
+    with st.spinner("2. Loading Dataset..."):
         dataset = load_dataset(dataset_repo, split="train")
-    with st.spinner("Creating Kangas..."):
-        dg = kg.DataGrid(dataset)
-    with st.spinner("Saving Kangas..."):
-        dg.save(str(proj_dir / 'datagrids' / dg_file_name))
-
+    with st.spinner("3. Creating Kangas..."):
+        if not os.path.exists(dg_full_path):
+            dg = kg.DataGrid(dataset)
+            dg.save(dg_full_path)
+    with st.spinner("4. Loading Kangas..."):
+        src = f"https://{servername}/kangas/?datagrid=.%2Fdatagrids%2F" + dg_file_name
+        
 hf_dataset = st.text_input("HuggingFace Dataset", value='beans')
 st.button("Download and Run", on_click=partial(kangas_fn, hf_dataset))
 st.markdown("""Click the dropdown in Kangas to see pre-loaded datasets""")
-st.components.v1.iframe(src, None, 1000, scrolling=True)
+components.iframe(src, None, 1000, scrolling=True)