Update app.py

app.py

@@ -3,17 +3,17 @@ import json
 from datasets import load_dataset
 import sqlparse
 
-st.set_page_config(page_title="BIRD SQL inspection", layout="wide")
+st.set_page_config(page_title="Proofsteps from Proof-Pile-V2", layout="wide")
 
 st.markdown("<h1 style='text-align: center; color: #00BFFF;'>BIRD SQL inspection 🔍</h1>", unsafe_allow_html=True)
 
 st.markdown("""
-Here you can inspect BIRD SQL data with schemas from [BIRD-Bench](https://bird-bench.github.io/).
+Here you can inspect proofsteps from [Proof-Pile-V2](https://huggingface.co/datasets/EleutherAI/proof-pile-2).
 """)
 
 @st.cache()
 def load_data():
-    ds = load_dataset("xu3kev/BIRD-SQL-data", split="train")
+    ds = load_dataset("xu3kev/proof-pile-2-proofsteps", split="lean_proofsteps")
     return ds
 
 
@@ -21,12 +21,15 @@ samples = load_data()
 st.sidebar.header('Sample Selection')
 index_example = st.sidebar.number_input(f"Choose a sample from the existing {len(samples)} notebooks:", min_value=0, max_value=max(0, len(samples)-1), value=0, step=1)
 
-db_id = samples[index_example]["db_id"]
-st.markdown(f'<h2 style="color:blue;">{index_example} Question:</h2>', unsafe_allow_html=True)
-st.code(samples[index_example]["question"])
-sql_str = samples[index_example]["SQL"]
-sql_str_pretty = sqlparse.format(sql_str, reindent=True, keyword_case='upper')
-st.markdown(f'<h2 style="color:blue;">SQL:</h2>', unsafe_allow_html=True)
-st.code(sql_str_pretty)
-st.markdown(f'<h2 style="color:blue;">Database schema:</h2>', unsafe_allow_html=True)
-st.code(samples[index_example]["schema"])
+#db_id = samples[index_example]["db_id"]
+#st.markdown(f'<h2 style="color:blue;">{index_example} Question:</h2>', unsafe_allow_html=True)
+#st.code(samples[index_example]["question"])
+#sql_str = samples[index_example]["SQL"]
+#sql_str_pretty = sqlparse.format(sql_str, reindent=True, keyword_case='upper')
+#st.markdown(f'<h2 style="color:blue;">SQL:</h2>', unsafe_allow_html=True)
+#st.code(sql_str_pretty)
+st.markdown(f'<h2 style="color:blue;">Content:</h2>', unsafe_allow_html=True)
+st.code(samples[index_example]["text"])
+st.markdown(f'<h2 style="color:blue;">Metadata:</h2>', unsafe_allow_html=True)
+st.code(samples[index_example]["meta"])
+