update app

app.py

@@ -431,6 +431,8 @@ def get_example_input():
     problem = problems[task_id]
     prefix, suffix = problem['prompt'], problem['suffix']
     test_case = problem['test']
+    test_case = test_case.replace('def check(candidate):', 'def run_test():')
+    test_case = test_case.replace('candidate', test_case['entry_point'])
     return prefix, '', suffix, test_case, task_id, ''
 
 def check_result(task_id, completion):
@@ -458,6 +460,14 @@ def create_chatbot_demo():
             "[[Blog](https://hkunlp.github.io/blog/2025/dreamon/)]"
         )
 
+        with gr.Row():
+            sample_btn = gr.Button("Example Prompt")
+
+        with gr.Row():
+            generate_btn = gr.Button("Generate", variant="primary")
+            clear_btn = gr.Button("Clear")
+            check_btn = gr.Button("Run test case")
+
         with gr.Row():
             with gr.Column():
                 # Prefix input
@@ -502,13 +512,7 @@ def create_chatbot_demo():
                     lines=2
                 )
         
-        with gr.Row():
-            sample_btn = gr.Button("Example Prompt")
 
-        with gr.Row():
-            generate_btn = gr.Button("Generate", variant="primary")
-            clear_btn = gr.Button("Clear")
-            check_btn = gr.Button("Run test case")
 
         # Generation Settings
         with gr.Accordion("Generation Settings"):