Update

app.py

@@ -62,6 +62,7 @@ else:
             df = pd.DataFrame(
                 {
                     "problem_id": [int(problem['problem_id']) for problem in ds],
+                    "difficulty": [str(problem['difficulty']) 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],
                 }
@@ -70,6 +71,7 @@ else:
             df,
             column_config={
                 "problem_id": st.column_config.NumberColumn("Problem ID"),
+                "difficulty": st.column_config.TextColumn("Difficulty"),
                 "acceptance_rate": st.column_config.LineChartColumn("Acceptance Rate", y_min=0, y_max=100),
                 "problem_link": st.column_config.LinkColumn("Link", display_text="Open"),
             },