record result update

app.py

@@ -8,7 +8,7 @@ import os
 import re
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
-openai_api_key = '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'#os.getenv("OPENAI_API_KEY")
+openai_api_key = os.getenv("OPENAI_API_KEY")
 
 task_completed = 0
 task_completed_steps = None
@@ -48,8 +48,8 @@ class AppSimulator:
         human_result_url = "jjz5463/simulator_human_result"
         synthetic_tasks = load_dataset(human_result_url, app_name, split='train')
 
-        incomplete_tasks = synthetic_tasks.filter(is_task_incomplete)
-        give_up_tasks = synthetic_tasks.filter(is_task_given_up)
+        incomplete_tasks = synthetic_tasks.filter(is_task_incomplete, with_indices=True,)
+        give_up_tasks = synthetic_tasks.filter(is_task_given_up, with_indices=True,)
         if len(incomplete_tasks) > 0:
             incomplete_task = incomplete_tasks[0]
             task = incomplete_task["tasks"]
@@ -110,7 +110,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)
+        incomplete_tasks = synthetic_tasks.filter(is_task_incomplete, with_indices=True,)
         if len(incomplete_tasks) > 0:
             incomplete_task_index = incomplete_tasks['__index_level_0__'][0]
             synthetic_tasks = synthetic_tasks.map(
@@ -142,7 +142,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)
+    incomplete_tasks = synthetic_tasks.filter(is_task_incomplete, with_indices=True,)
     if len(incomplete_tasks) > 0:
         incomplete_task_index = incomplete_tasks['__index_level_0__'][0]