buttons update

chatbot_simulator.py

@@ -53,6 +53,7 @@ class ChatbotSimulation:
         current_page = self.page_history[-1] if len(self.page_history) >= 1 else "Home"
         last_page = self.page_history[-2] if len(self.page_history) > 1 else "Home"
         page_info = self._get_page_details(current_page)
+        page_linkage = self.sitemap.get(current_page, {}).get('links_to')
 
         return get_system_prompt(app_name=self.app_name,
                                  system_data=self.system_data,
@@ -62,7 +63,8 @@ class ChatbotSimulation:
                                  last_page=last_page,
                                  actions=self.actions,
                                  user_state=self.user_state,
-                                 page_info=page_info
+                                 page_info=page_info,
+                                 page_linkage=page_linkage
                                  )
 
     @retry(