update

app.py

@@ -20,7 +20,6 @@ with st.spinner("Loading data...", show_time=True):
         problem_id = problem["title"]
         problem['type'] = "leetcode"
         problem_dict[problem_id] = problem
-
     
     problem_count = len(problem_dict)
 
@@ -63,11 +62,11 @@ else:
         with st.spinner("Loading Framework...", show_time=True):
             df = pd.DataFrame(
                 {
-                    "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],
+                    "problem_id": [int(problem['problem_id']) for problem in problem_dict.values()],
+                    "difficulty": [str(problem['difficulty']) for problem in problem_dict.values()],
+                    "type": [str(problem['type']) for problem in problem_dict.values()],
+                    "problem_link": ["https://huggingface.co/spaces/Elfsong/CodeArena/?problem=" + str(problem['problem_id']) for problem in problem_dict.values()],
+                    "acceptance_rate": [[random.randint(0, 100) for _ in range(20)] for problem in problem_dict.values()],
                 }
             )
         st.dataframe(