Update menu.py

menu.py

@@ -128,7 +128,10 @@ def menu():
         # Frequency analysis for add-ons and instructions
         addon_counts = {}
         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')
 
@@ -139,33 +142,27 @@ def menu():
                     addons = item_parts[1].strip().replace('Add-Ons:', '').split(', ')
                     instructions = item_parts[2].strip().replace('Instructions:', '').split(', ')
 
-                    # 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
-
+                    # Track the frequency of spice levels
+                    for addon in addons:
+                        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 the cleaned add-ons and instructions to check what we're fetching
-        print(f"Add-ons fetched (cleaned): {addon_counts}")
-        print(f"Instructions fetched: {instruction_counts}")
-
-        # Get the most frequent add-ons and instructions
-        most_common_addons = sorted(addon_counts, key=addon_counts.get, reverse=True)[:3]  # Top 3 most frequent addons
-        most_common_instructions = sorted(instruction_counts, key=instruction_counts.get, reverse=True)[:3]  # Top 3 most frequent instructions
+        # 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(f"Most common add-ons (cleaned): {most_common_addons}")
-        print(f"Most common instructions: {most_common_instructions}")
+        # 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:
+                most_common_addons.append(most_ordered_spice)
+            most_common_addons = most_common_addons[:3]
 
         categories = ["All", "Veg", "Non veg"]
 
@@ -190,7 +187,6 @@ def menu():
         most_common_instructions=most_common_instructions  # Pass most common instructions
     )
 
-
 @menu_blueprint.route('/api/addons', methods=['GET'])
 def get_addons():
     item_name = request.args.get('item_name')