Update

app.py

@@ -11,6 +11,9 @@ if "problem" in st.query_params:
     with st.spinner("Loading data...", show_time=True):
         ds = load_dataset("Elfsong/leetcode_data", split='train')
 
+    if st.button("Back"):
+        del st.session_state['problem']
+
 else:
     with st.spinner("Loading data...", show_time=True):
         ds = load_dataset("Elfsong/leetcode_data", split='train')