click(number) instead of button name

chatbot_simulator.py

@@ -180,7 +180,7 @@ class ChatbotSimulation:
         return greeting + gpt_instruction
 
     def _extract_buttons(self):
-        """Extract buttons and their action types from the latest conversation if role is 'assistant'."""
+        """Extract button numbers and their action types from the latest conversation if role is 'assistant'."""
         # Get the last message
         last_message = self.conversation[-1]
 
@@ -191,7 +191,7 @@ class ChatbotSimulation:
         # Extract the content of the last message
         message_content = last_message.get("content", "")
 
-        # Make the split case-insensitive by searching for the phrase with re.IGNORECASE
+        # Split the message content to isolate the button section
         options_split = re.split(r"you have the following options:", message_content, flags=re.IGNORECASE)
 
         # If the split doesn't produce at least two parts, return an empty dictionary
@@ -200,11 +200,11 @@ class ChatbotSimulation:
 
         # Extract button definitions from the second part of the split content
         button_section = options_split[1]
-        pattern = r"\d+\.\s+(.*?):\s+([a-zA-Z_]+)"
+        pattern = r"(\d+)\.\s+(.*?):\s+([a-zA-Z_]+)"  # Capture the number, button name, and action type
         buttons = re.findall(pattern, button_section)
 
-        # Construct the dictionary with button names as keys and action types as values
-        return {name.strip().lower(): action_type.strip().lower() for name, action_type in buttons}
+        # Construct the dictionary with button numbers as keys and action types as values
+        return {number: action_type.strip().lower() for number, _, action_type in buttons}
 
     def _is_valid_input(self, user_input):
         """Validate user input format."""
@@ -214,27 +214,27 @@ class ChatbotSimulation:
             return [True, "Enter Anything is empty"]
 
         # Validate input format
-        pattern = r"^(?P<action_type>\w+)\((?P<button_name>[^,]+)(?:,\s*(?P<query>.+))?\)$"
+        pattern = r"^(?P<action_type>\w+)\((?P<button_number>[^,]+)(?:,\s*(?P<query>.+))?\)$"
         match = re.match(pattern, user_input)
 
         if not match:
-            return [False, "Your input doesn't match the format: action_type(button name), OR if type, use type(button name, query)"]
+            return [False, "Your input doesn't match the format: action_type(button number), OR if type, use type(button number, query)"]
 
         # Extract parsed components
         action_type = match.group("action_type").lower()
-        button_name = match.group("button_name").strip().lower()
+        button_name = match.group("button_number").strip().lower()
         query = match.group("query")  # Optional query for `type`
 
-        # Validate button name and action type
+        # Validate button number and action type
         if button_name not in valid_buttons:
             return [False,
-                    "Invalid Button name! Recall: Each button is in the format: `number. button name: action_type`"]  # Button name must match exactly (case insensitive)
+                    "Invalid Button number! Recall: Each button is in the format: `number. button name: action_type`"]  # Button number must match exactly (case insensitive)
         if action_type != valid_buttons[button_name]:
             return [False,
                     "Invalid action type! Recall: Each button is in the format: `number. button name: action_type`"]  # Action type must match the button's specified type
         if action_type == "type" and query is None:
             return [False,
-                    "Missing Query for action type 'type'! Recall: use the format: `type(button name, query)`"]  # `type` action requires a query
+                    "Missing Query for action type 'type'! Recall: use the format: `type(button number, query)`"]  # `type` action requires a query
         if action_type != "type" and query is not None:
             return [False,
                     "Non-`type` action_type cannot take query!"]  # Non-`type` actions must not have a query