Update

app.py

@@ -15,14 +15,27 @@ with st.spinner("Loading data...", show_time=True):
 
 if "problem" in st.query_params:
     problem_id = int(st.query_params["problem"])
-    st.write(f"Problem Description - [{problem_id}]")
+    st.write(f"Problem [{problem_id}]")
     problem_instance = problem_dict[problem_id]
+    
+    with st.expander("Problem Description"):
+        st.markdown(problem_instance["question_content"])
 
-    st.markdown(problem_instance["question_content"])
-
-    st.code(problem_instance["test_case_generator"])
-
-    st.write(problem_instance["test_cases"])
+    with st.expander("Test Cases"):
+        df = pd.DataFrame(
+            {
+                "input": [problem['input'] for problem in problem_instance["test_cases"]],
+                "output": [problem['output'] for problem in problem_instance["test_cases"]],
+            }
+        )
+        st.dataframe(
+            df,
+            column_config={
+                "input": st.column_config.TextColumn("Problem ID"),
+                "output": st.column_config.TextColumn("Dynamic Point"),
+            },
+            column_order=("input", "output"),
+        )
     
 else:
     with st.spinner("Loading Framework...", show_time=True):