Update

app.py

@@ -5,15 +5,19 @@ from datasets import load_dataset
 
 st.title("Code Arena")
 
+with st.spinner("Loading data...", show_time=True):
+    problem_dict = dict()
+    ds = load_dataset("Elfsong/leetcode_data", split='train')
+    for problem in ds:
+        problem_id = problem["problem_id"]
+        problem_dict[problem_id] = problem
+
+
 if "problem" in st.query_params:
     problem_id = st.query_params["problem"]
     st.write(f"Problem Description - [{problem_id}]")
-    with st.spinner("Loading data...", show_time=True):
-        ds = load_dataset("Elfsong/leetcode_data", split='train')
-
-    if st.button("Back"):
-        del st.query_params['problem']
-
+    st.write(problem_dict[problem_id])
+    
 else:
     with st.spinner("Loading data...", show_time=True):
         ds = load_dataset("Elfsong/leetcode_data", split='train')