force json format in updated user state

chatbot_simulator.py

@@ -136,11 +136,18 @@ Example Output Format:
         # Parse and update the user state
         updated_state = json_repair.loads(updated_state)
 
-        if isinstance(updated_state, dict):
-            pass
-        else:
-            print(updated_state)
-            print(type(updated_state))
+        while not isinstance(updated_state, dict):
+            transform_prompt = f""""
+            Transform {updated_state} to a properly formate JSON file.
+            Example Output Format:
+            {{
+                'current_page': 'Home',
+                'last_page': 'Home',
+                'task_completed': 'False',
+            }}
+            """
+            updated_state = self._get_openai_response({'role': 'system', 'content': transform_prompt})
+            updated_state = json_repair.loads(updated_state)
 
         try:
             if updated_state['task_completed'].lower() == 'true':