task completed failed update

chatbot_simulator.py

@@ -12,7 +12,7 @@ class ChatbotSimulation:
         self.user_state = user_state
         self.user_state['current_page'] = 'Home'  # Initialize current page
         self.user_state['last_page'] = 'Home'
-        self.user_state['task_completed'] = 0
+        self.user_state['task_completed'] = 'False'
         self.task = task
         self.app_name = app_name
         self.log_location = log_location
@@ -115,7 +115,7 @@ Sitemap: {self.sitemap}
 
 Instructions:
 1. If the 'current_page' has changed, update it to a page from the sitemap.
-2. If the task is finished, update 'task_completed' to 1. Otherwise, leave it as 0.
+2. If the task is finished, update 'task_completed' to True. Otherwise, leave it as False.
 3. If no updates are needed, return the user state exactly as provided, without modification.
 
 Important:
@@ -126,9 +126,9 @@ Example Output Format:
 {{
     'current_page': 'Home',
     'last_page': 'Home',
-    'task_completed': 0,
+    'task_completed': 'False',
 }}
-        """
+                """
 
         self.conversation.append({"role": "assistant", "content": update_prompt})
         updated_state = self._get_openai_response(self.conversation)
@@ -137,11 +137,10 @@ Example Output Format:
         updated_state = json_repair.loads(updated_state)
 
         try:
-            updated_state['task_completed'] = int(updated_state['task_completed'])
-            if updated_state['task_completed']:
+            if updated_state['task_completed'].lower() == 'true':
                 return f"Task completed! You took {self.prompt_count} steps."
         except:
-            updated_state['task_completed'] = 0
+            updated_state['task_completed'] = 'False'
 
         self.user_state = updated_state