record result update

app.py

@@ -14,11 +14,19 @@ task_completed = 0
 task_completed_steps = None
 
 
-def is_task_incomplete(example, idx):
+def is_task_incomplete(example):
     return example["task_completed"] is None and example["task_completed_steps"] is None
 
 
-def is_task_given_up(example, idx):
+def is_task_given_up(example):
+    return example["task_completed"] == 0 and example["task_completed_steps"] == 0
+
+
+def is_task_incomplete_idx(example, idx):
+    return example["task_completed"] is None and example["task_completed_steps"] is None
+
+
+def is_task_given_up_idx(example, idx):
     return example["task_completed"] == 0 and example["task_completed_steps"] == 0
 
 
@@ -110,7 +118,7 @@ def chat(user_input, history):
         human_result_url = "jjz5463/simulator_human_result"
         app_name = simulator_app.app_name
         synthetic_tasks = load_dataset(human_result_url, app_name, split='train')
-        incomplete_tasks = synthetic_tasks.filter(is_task_incomplete, with_indices=True,)
+        incomplete_tasks = synthetic_tasks.filter(is_task_incomplete_idx, with_indices=True,)
         if len(incomplete_tasks) > 0:
             incomplete_task_index = incomplete_tasks['__index_level_0__'][0]
             synthetic_tasks = synthetic_tasks.map(
@@ -142,7 +150,7 @@ def give_up():
     synthetic_tasks = load_dataset(human_result_url, app_name, split='train')
 
     # Find the first incomplete task
-    incomplete_tasks = synthetic_tasks.filter(is_task_incomplete, with_indices=True,)
+    incomplete_tasks = synthetic_tasks.filter(is_task_incomplete_idx, with_indices=True,)
     if len(incomplete_tasks) > 0:
         incomplete_task_index = incomplete_tasks['__index_level_0__'][0]