forgot to store task itself back to db

app.py

@@ -37,9 +37,10 @@ def find_smallest_incomplete_task(app_name):
     return smallest_incomplete_idx
 
 
-def write_task_data(app_name, idx, task_complete, task_completed_step):
+def write_task_data(app_name, idx, task, task_complete, task_completed_step):
     doc_ref = db.collection(app_name).document(f"{app_name}_{idx}")
     doc_ref.set({
+        "task": task,
         "task_completed": task_complete,
         "task_completed_steps": task_completed_step
     })
@@ -51,6 +52,7 @@ class AppSimulator:
         self.openai_api_key = openai_api_key
         self.app_name = None
         self.smallest_index = None
+        self.task = None
 
     def initialize_simulator(self, sitemap_url, progress=gr.Progress(track_tqdm=True)):
         """Initialize the simulator with retries and elapsed time tracking."""
@@ -80,6 +82,8 @@ class AppSimulator:
         solution = incomplete_task["steps"]
         user_data = incomplete_task["attributes"]["user_data"]
 
+        self.task = task
+
         self.simulation = ChatbotSimulation(
             app_name=app_name,
             site_map=sitemap,
@@ -122,7 +126,8 @@ def chat(user_input, history):
         task_completed_step = int(match.group(1))
         app_name = simulator_app.app_name
         idx = simulator_app.smallest_index
-        write_task_data(app_name, idx, task_complete, task_completed_step)
+        task = simulator_app.task
+        write_task_data(app_name, idx, task, task_complete, task_completed_step)
 
     return response
 
@@ -134,8 +139,9 @@ def give_up():
 
     app_name = simulator_app.app_name
     idx = simulator_app.smallest_index
+    task = simulator_app.task
 
-    write_task_data(app_name, idx, task_completed, task_completed_steps)
+    write_task_data(app_name, idx, task, task_completed, task_completed_steps)
 
     return "Task marked as abandoned (Give-Up action)."