Update

app.py

@@ -13,19 +13,21 @@ st.title("Code Arena")
 
 with st.spinner("Loading data...", show_time=True):
     problem_dict = dict()
+
+    # Leetcode Data
     ds = load_dataset("Elfsong/leetcode_data", split='train')
     for problem in ds:
-        problem_id = problem["problem_id"]
+        problem_id = problem["title"]
+        problem['type'] = "leetcode"
         problem_dict[problem_id] = problem
+
+    
     problem_count = len(problem_dict)
 
 
 if "problem" in st.query_params:
     problem_id = int(st.query_params["problem"])
     problem_instance = problem_dict[problem_id]
-
-    with st.container(border=True):
-        st.write(f"Problem [{problem_id}]")
     
     with st.expander("Problem Description"):
         st.markdown(problem_instance["question_content"])
@@ -63,6 +65,7 @@ else:
                 {
                     "problem_id": [int(problem['problem_id']) for problem in ds],
                     "difficulty": [str(problem['difficulty']) for problem in ds],
+                    "type": [str(problem['type']) for problem in ds],
                     "problem_link": ["https://huggingface.co/spaces/Elfsong/CodeArena/?problem=" + str(problem['problem_id']) for problem in ds],
                     "acceptance_rate": [[random.randint(0, 100) for _ in range(20)] for problem in ds],
                 }
@@ -72,6 +75,7 @@ else:
             column_config={
                 "problem_id": st.column_config.NumberColumn("Problem ID", width='small'),
                 "difficulty": st.column_config.TextColumn("Difficulty", width='small'),
+                "type": st.column_config.TextColumn("Type", width='small'),
                 "acceptance_rate": st.column_config.LineChartColumn("Acceptance Rate", y_min=0, y_max=100),
                 "problem_link": st.column_config.LinkColumn("Link", display_text="Open", width='small'),
             },