Update streamlitapp.py

streamlitapp.py

@@ -7,13 +7,16 @@ import streamlit as st
 import streamlit.components.v1 as components
 from datasets import load_dataset
 
-proj_dir = Path(__file__).parent
+# If you duplicate this space, change this to match your space:
 servername = 'comet-team-kangas-demo.hf.space'
-src = f"https://{servername}/kangas/?datagrid=.%2Fdatagrids%2Fcoco-500.datagrid"
 st.set_page_config(layout="wide")
+proj_dir = Path(__file__).parent
+
+if 'iframe_source' not in st.session_state:
+    st.session_state['iframe_source'] = f"https://{servername}/kangas/?datagrid=.%2Fdatagrids%2Fcoco-500.datagrid"
 
-st.markdown("# HuggingFace + Streamlit + Kangas")
-st.markdown("1. Select a HuggingFace dataset, or select one below")
+st.markdown("# HuggingFace Spaces + Streamlit + Kangas")
+st.markdown("1. Select a HuggingFace dataset and Download, or select one below in Kangas")
 
 def kangas_fn(dataset_repo):
     global src
@@ -27,10 +30,10 @@ def kangas_fn(dataset_repo):
     with st.spinner("4. Saving Kangas..."):
         if not os.path.exists(dg_full_path):
             dg.save(dg_full_path)
-    with st.spinner("4. Loading Kangas..."):
-        src = f"https://{servername}/kangas/?datagrid=.%2Fdatagrids%2F" + dg_file_name
+    with st.spinner("5. Loading Kangas..."):
+        st.session_state['iframe_source'] = f"https://{servername}/kangas/?datagrid=.%2Fdatagrids%2F" + dg_file_name
         
 hf_dataset = st.text_input("HuggingFace Dataset", value='beans')
 st.button("Download", on_click=partial(kangas_fn, hf_dataset))
-st.markdown("""Click the dropdown in Kangas select a dataset""")
-components.iframe(src, None, 1000, scrolling=True)
+st.markdown("""Click the dropdown in Kangas to select a previously-downloaded dataset""")
+components.iframe(st.session_state['iframe_source'], None, 1000, scrolling=True)