Update APPS

app.py

@@ -14,24 +14,35 @@ st.title("Code:blue[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:
+    # Venus Data
+    venus_ds = load_dataset("Elfsong/leetcode_data", split='train')
+    for problem in venus_ds:
         problem_id = problem["title"]
         problem['type'] = "leetcode"
         problem_dict[problem_id] = problem
+
+    # APPS Data
+    apps_ds = load_dataset("Elfsong/APPS_Python", split='test')
+    for problem in apps_ds:
+        problem_id = f'apps_{problem["problem_id"]}'
+        problem['type'] = "apps"
+        problem_dict[problem_id] = problem
     
     problem_count = len(problem_dict)
 
 
 if "problem" in st.query_params:
     problem_id = str(st.query_params["problem"])
+    problem_type = problem['type']
     problem_instance = problem_dict[problem_id]
 
     st.header(problem_id)
     
     with st.expander("Problem Description"):
-        st.markdown(problem_instance["question_content"])
+        if problem_type == "leetcode":
+            st.markdown(problem_instance["question_content"])
+        elif problem_type == "apps":
+            st.markdown(problem_instance["problem_content"])
 
     with st.expander("Test Cases"):
         test_cases = json.loads(problem_instance["test_cases"])
@@ -51,10 +62,13 @@ if "problem" in st.query_params:
         )
 
     with st.expander("Test Case Generator"):
-        test_case_generator = problem_instance["test_case_generator"]
-        prompt = "# For now, we only disclose the top 20 lines of the test case generator.\n# the full version will be released after the paper review process.\n"
-        test_case_generator = "\n".join(test_case_generator.split("\n")[:20])
-        st.code(prompt+test_case_generator)
+        if problem_type == "leetcode":
+            test_case_generator = problem_instance["test_case_generator"]
+            prompt = "# For now, we only disclose the top 20 lines of the test case generator.\n# the full version will be released after the paper review process.\n"
+            test_case_generator = "\n".join(test_case_generator.split("\n")[:20])
+            st.code(prompt+test_case_generator)
+        else:
+            st.code("Stay tuned!")
 
     
 else: