Update menu.py

menu.py

@@ -130,8 +130,6 @@ def menu():
         instruction_counts = {}
         spice_levels = ['Mild', 'Medium', 'Spicy', 'Extra Spicy']
 
-        # Iterate through past orders and calculate frequency of add-ons and spice levels
-        spice_counts = {}
         for order in past_orders:
             order_details = order.get('Order_Details__c', '').split('\n')
 
@@ -142,39 +140,44 @@ def menu():
                     addons = item_parts[1].strip().replace('Add-Ons:', '').split(', ')
                     instructions = item_parts[2].strip().replace('Instructions:', '').split(', ')
 
-                    # Track the frequency of spice levels
-                    for addon in addons:
-                        print(f"Addon: {addon.strip()}") 
-                        for spice in spice_levels:
-                            if spice.lower() in addon.lower():
-                                spice_counts[spice] = spice_counts.get(spice, 0) + 1
-
-                    # Track instructions
-                    for instruction in instructions:
-                        if instruction:
-                            instruction_counts[instruction] = instruction_counts.get(instruction, 0) + 1
-        print("\nAddon counts:")
+                    # Clean the add-on names by removing any price information
+                    cleaned_addons = [re.sub(r"\s?\(\$\d+(\.\d{2})?\)", "", addon) for addon in addons]
+                    # Split add-ons by semicolon and process separately
+                    for addon in cleaned_addons:
+                        # If the add-on contains a semicolon, split it into multiple add-ons
+                        individual_addons = addon.split(';')  # Split add-ons separated by semicolons
+                        for individual_addon in individual_addons:
+                            individual_addon = individual_addon.strip()  # Remove extra spaces
+                            if individual_addon:
+                                addon_counts[individual_addon] = addon_counts.get(individual_addon, 0) + 1
+
+        # Print the cleaned add-ons and their counts in the requested format
+        print(f"Add-ons fetched (cleaned): {addon_counts}")
+        print("Add-on Counts in the format [Add-onName[Count]]:")
         for addon, count in addon_counts.items():
-            print(f"{addon}: {count}")
-
-        # Get the most common add-ons, including spice levels
-        most_common_addons = sorted(addon_counts, key=addon_counts.get, reverse=True)[:3]
-        most_common_instructions = sorted(instruction_counts, key=instruction_counts.get, reverse=True)[:3]
-         # Print to check the final most common add-ons and instructions
-        print("\nTop 3 Most Common Add-Ons:")
-        for addon in most_common_addons:
-            print(f"{addon}")
-
-        # Include most ordered spice level if it's not already in the top 3
-        if not any(spice in most_common_addons for spice in spice_levels):
-            most_ordered_spice = max(spice_counts, key=spice_counts.get, default=None)
-            if most_ordered_spice:
-                print(f"\nMost Ordered Spice: {most_ordered_spice}")
-                most_common_addons.append(most_ordered_spice)
-            most_common_addons = most_common_addons[:3]
-        print("\nFinal Most Common Add-Ons (Including Spice Level):")
-        for addon in most_common_addons:
-            print(f"{addon}")
+            print(f"{addon}[{count}]")
+
+        # Get the most frequent add-ons
+        most_common_addons = sorted(addon_counts, key=addon_counts.get, reverse=True)[:3]  # Top 3 most frequent addons
+
+        # Print the most common add-ons in the required format
+        print(f"Most common add-ons (cleaned): {most_common_addons}")
+
+        # Now include any spice level if it's not already in the top 3
+        most_common_spice = None
+        for spice in spice_levels:
+            if spice in addon_counts:
+                most_common_spice = spice
+                break
+
+        if most_common_spice and most_common_spice not in most_common_addons:
+            most_common_addons.append(most_common_spice)
+
+        # Ensure the list does not exceed 3
+        most_common_addons = most_common_addons[:3]
+
+        print(f"Final Most Common Add-Ons (Including Spice Level): {most_common_addons}")
+
         categories = ["All", "Veg", "Non veg"]
 
     except Exception as e: