update prompt

chatbot_simulator.py

@@ -95,7 +95,7 @@ Rules:
 
         # Update user state using GPT's response
         update_prompt = f"""
-        Update the user state based on the input: '{user_input}'.
+        Update the user state based on user input: '{user_input}' and task: {self.task}
 
         Current user state (JSON format): {self.user_state}
 
@@ -113,17 +113,6 @@ Rules:
         - Ensure 'current_page' and 'task_completed' are keys in the returned dictionary.
         - Return **only the JSON object** without additional output or wrapping.
         - **AVOID OUTPUT A LIST**, must be JSON!
-        
-        The format for each page should be:
-        
-        {{
-            "page_name": page_name
-            "features": {{}},
-            "user_data": {{
-                "Data Entry 1": "Details of Data Entry 1", 
-                "Data Entry 2": "Details of Data Entry 2"
-            }}
-        }}
         """
 
         self.conversation.append({"role": "assistant", "content": update_prompt})
@@ -134,20 +123,9 @@ Rules:
 
         if isinstance(updated_state, list):
             reformat_prompt = f'''
-            Given the {updated_state}, reformat it into a proper JSON
-            with only 3 keys: page_name, features, user_data
-            Follow the format:
-             {{
-            "page_name": page_name
-            "features": {{}},
-            "user_data": {{
-                "Data Entry 1": "Details of Data Entry 1", 
-                "Data Entry 2": "Details of Data Entry 2"
-                }}
-            }}
+            Given the {updated_state}, reformat it into a proper JSON.
             '''
-            self.conversation.append({"role": "assistant", "content": reformat_prompt})
-            reformat_state = self._get_openai_response(self.conversation)
+            reformat_state = self._get_openai_response([{"role": "assistant", "content": reformat_prompt}])
             updated_state = json_repair.loads(reformat_state)
 
         if updated_state['task_completed']: