Update 2_hf_repo_info.py

2_hf_repo_info.py

@@ -2,15 +2,23 @@ import streamlit as st
 from huggingface_hub import list_models
 from datetime import date, timedelta, datetime, timezone
 import pandas as pd
+import asyncio
 
 # os.environ["KAGGLE_USERNAME"] = "hahunavth"
 # os.environ["KAGGLE_KEY"] = ""
 from kaggle.api.kaggle_api_extended import KaggleApi
 
-api = KaggleApi()
-api.authenticate()
-
 st.set_page_config(layout="wide")
+
+
+@st.cache_resource
+def init_kaggle():
+    api = KaggleApi()
+    api.authenticate()
+    return api
+
+api = init_kaggle()
+
 # @st.cache_resource
 # def get_model_list():
 #     return [model for model in list(list_models(author="hahunavth", filter="emofs2"))]
@@ -47,12 +55,6 @@ for model in models:
 
     step = model.id.replace('hahunavth/emofs2-exp', '').split('_')[0]
 
-    kaggle_slug = f"ess-vlsp2023-train-{step}"
-    try:
-        kernel_status = api.kernel_status(user_name="hahunavth", kernel_slug="ess-vlsp2023-train-42")["status"]
-    except:
-        kernel_status = "not found"
-
     _models.append(
         {
             "id": model.id,
@@ -60,19 +62,47 @@ for model in models:
             "ckpt_list": last_ckpt,
             "huggingface": f'https://huggingface.co/{model.id}/tree/main' if model.id else None,
             "notebook": f"https://www.kaggle.com/code/hahunavth/ess-vlsp2023-train-{step}" if model.id else None,
-            "kaggle": f"/kaggle/repo/vlsp2023-ess/config/exp{step}",
-            "status": kernel_status,
+            "status": None,
+            "config": f"/kaggle/repo/vlsp2023-ess/config/exp{step}",
+            # "command": "st.balloons",
+            # "is_widget": True
         }
     )
 
-
 df = pd.DataFrame(_models)
 
-st.dataframe(
-    df, 
+@st.cache_data
+def get_kernel_status(step):
+    kaggle_slug = f"ess-vlsp2023-train-{step}"
+    try:
+        kernel_status = api.kernel_status(user_name="hahunavth", kernel_slug=kaggle_slug)["status"]
+    except:
+        kernel_status = "not found"
+    return kernel_status
+
+async def set_df_kernel_status(df, n=12):
+    for i in range(min(len(df), n)):
+        step = df.iloc[i]["id"].replace('hahunavth/emofs2-exp', '').split('_')[0]
+        status = get_kernel_status(step)
+        df.at[i, "status"] = status
+        print(i, status)
+
+asyncio.run(set_df_kernel_status(df))
+
+edit_df = st.data_editor(
+    df,
     column_config={
         "huggingface": st.column_config.LinkColumn(),
         "notebook": st.column_config.LinkColumn(),
+        # "a": st.button("Click me"),
     }, 
-    use_container_width=True
 )
+
+# st.dataframe(
+#     edit_df, 
+#     column_config={
+#         "huggingface": st.column_config.LinkColumn(),
+#         "notebook": st.column_config.LinkColumn(),
+#     }, 
+#     use_container_width=True
+# )