Update app.py

app.py

@@ -9,12 +9,29 @@ st.markdown("<h1 style='text-align: center; color: #00BFFF;'>Kaggle Notebooks in
 st.markdown("Here you can inspect Kaggle notebooks that were converted to python scripts and deduplicated.")
 @st.cache()
 def load_data():
-    ds = load_dataset("loubnabnl/kaggle_scripts_dedup_subset", split="train")
+    ds = load_dataset("loubnabnl/subset_kaggle_scripts", split="train")
     return ds
 
+def show_extra_info(e):
+    kv = json.loads(e["kversion"])[0]
+    try:
+        data_v = json.loads(e["dataset_versions"])[0]
+    except:
+        datav = ""
+    nb_title = kv["Title"]
+    nb_votes = kv["TotalVotes"]
+    if data_v:
+        data_title = data_v["Title"]
+        import numpy as np
+        decription = data_v["Description"] if str(data_v["Description"]) != 'nan' else "<empty_description>"
+        data_text = f"It uses dataset: '{data_title}', described as: {decription}"
+    else:
+        data_text = ""
+    
+    text = f"The title of the notebook is: '{nb_title}' and it has {nb_votes} upvotes. {data_text}"
+    return text
+
 samples = load_data()
-#col1, _ = st.columns([2, 4])
-#with col1:
 index_example = st.number_input(f"Index of the chosen sample from the existing {len(samples)}", min_value=0, max_value=len(samples)-1, value=0, step=1)
-
+st.markdown(show_extra_info(samples[index_example]))
 st.code(samples[index_example]["script"])