Update app.py

app.py

@@ -12,8 +12,16 @@ logging.basicConfig(level=logging.INFO)
 
 # Global variables
 cart = []  # Stores items as [item_name, price, quantity] in the cart
-menu_preferences = "all"  # Tracks the current menu preference
+menu_preferences = None  # Tracks the current menu preference
+section_preferences = None  # Tracks the current section preference
 default_menu_preferences = "all"  # To reset menu preferences
+default_sections = {
+    "biryani": ["veg biryani", "paneer biryani", "chicken biryani", "mutton biryani"],
+    "starters": ["samosa", "onion pakoda", "chilli gobi", "chicken manchurian", "veg manchurian"],
+    "curries": ["paneer butter", "chicken curry", "fish curry", "chilli chicken"],
+    "desserts": ["gulab jamun", "ice cream"],
+    "soft drinks": ["cola", "lemon soda"]
+}
 prices = {
     "samosa": 9,
     "onion pakoda": 10,
@@ -27,17 +35,20 @@ prices = {
     "veg manchurian": 12,
     "chilli chicken": 14,
     "paneer biryani": 13,
-    "chicken curry": 14
+    "chicken curry": 14,
+    "gulab jamun": 8,
+    "ice cream": 6,
+    "cola": 5,
+    "lemon soda": 6
 }
 menus = {
     "all": list(prices.keys()),
     "vegetarian": [
-        "samosa", "onion pakoda", "chilli gobi", "veg biryani", "paneer butter", "veg manchurian", "paneer biryani"
+        "samosa", "onion pakoda", "chilli gobi", "veg biryani", "paneer butter", "veg manchurian", "paneer biryani", "gulab jamun", "ice cream", "cola", "lemon soda"
     ],
     "non-vegetarian": [
-        "chicken biryani", "mutton biryani", "fish curry", "chicken manchurian", "chilli chicken", "chicken curry"
-    ],
-    "guilt-free": ["samosa", "onion pakoda"]
+        "chicken biryani", "mutton biryani", "fish curry", "chicken manchurian", "chilli chicken", "chicken curry", "cola", "lemon soda"
+    ]
 }
 
 @app.route("/")
@@ -46,9 +57,10 @@ def index():
 
 @app.route("/reset-cart", methods=["GET"])
 def reset_cart():
-    global cart, menu_preferences
+    global cart, menu_preferences, section_preferences
     cart = []
-    menu_preferences = default_menu_preferences  # Reset menu preferences
+    menu_preferences = None
+    section_preferences = None
     return "Cart reset successfully."
 
 @app.route("/process-audio", methods=["POST"])
@@ -80,8 +92,7 @@ def process_audio():
         logging.info(f"Raw recognized command: {raw_command}")
 
         # Preprocess the command
-        all_menu_items = menus["all"]
-        command = preprocess_command(raw_command, all_menu_items)
+        command = preprocess_command(raw_command)
 
         # Process the command and get a response
         response = process_command(command)
@@ -96,113 +107,90 @@ def process_audio():
 
     return jsonify({"response": response})
 
-def preprocess_command(command, menu_items):
+def preprocess_command(command):
     """
     Preprocess the user command to improve matching:
-    - Use fuzzy matching for menu items.
-    - Use phonetic matching as a fallback.
+    - Normalize user input.
     """
-    def phonetic_match(word, options):
-        word_phonetic = doublemetaphone(word)[0]
-        for option in options:
-            if doublemetaphone(option)[0] == word_phonetic:
-                return option
-        return None
-
-    # First, try fuzzy matching
-    closest_match = process.extractOne(command, menu_items, scorer=fuzz.token_set_ratio)
-    if closest_match and closest_match[1] > 60:
-        return closest_match[0]
-
-    # Fallback to phonetic matching
-    words = command.split()
-    for word in words:
-        match = phonetic_match(word, menu_items)
-        if match:
-            return match
-
-    return command
+    return command.strip().lower()
 
 def process_command(command):
     """
     Process the user command based on the current menu and cart.
     """
-    global cart, menu_preferences
-    command = command.lower()
+    global cart, menu_preferences, section_preferences
 
     # Handle menu preferences
-    if "non-vegetarian" in command:
-        menu_preferences = "non-vegetarian"
-        return "You have chosen the Non-Vegetarian menu. To view the menu, say 'menu'."
-    elif "vegetarian" in command and "non-vegetarian" not in command:
-        menu_preferences = "vegetarian"
-        return "You have chosen the Vegetarian menu. To view the menu, say 'menu'."
-    elif "guilt-free" in command:
-        menu_preferences = "guilt-free"
-        return "You have chosen the Guilt-Free menu. To view the menu, say 'menu'."
-    elif "all" in command:
-        menu_preferences = "all"
-        return "You have chosen the complete menu. To view the menu, say 'menu'."
+    if menu_preferences is None:
+        if "vegetarian" in command:
+            menu_preferences = "vegetarian"
+            return "You have chosen the Vegetarian menu. Which section would you like? (biryani, starters, curries, desserts, soft drinks)"
+        elif "non-vegetarian" in command:
+            menu_preferences = "non-vegetarian"
+            return "You have chosen the Non-Vegetarian menu. Which section would you like? (biryani, starters, curries, desserts, soft drinks)"
+        elif "all" in command:
+            menu_preferences = "all"
+            return "You have chosen the complete menu. Which section would you like? (biryani, starters, curries, desserts, soft drinks)"
+        else:
+            return "Please specify your preference: Vegetarian, Non-Vegetarian, or All."
 
-    # Handle commands related to the current menu
-    menu = menus.get(menu_preferences, menus["all"])
+    # Handle section preferences
+    if section_preferences is None:
+        for section in default_sections.keys():
+            if section in command:
+                section_preferences = section
+                available_items = [item for item in default_sections[section] if item in menus[menu_preferences]]
+                return f"Here are the items in {section}: {', '.join(available_items)}. Say the item name to add to the cart."
+        return "Please specify a section: biryani, starters, curries, desserts, or soft drinks."
 
-    if "price of" in command:
-        item = command.replace("price of", "").strip()
-        closest_match = process.extractOne(item, prices.keys(), scorer=fuzz.token_set_ratio)
-        if closest_match and closest_match[1] > 60:
-            matched_item = closest_match[0]
-            return f"The price of {matched_item} is ${prices[matched_item]}."
-        return "Sorry, I couldn't find that item in the menu."
-
-    if "menu" in command:
-        return f"Here is your menu: {', '.join(menu)}. To select an item, say the item name."
+    # Handle item addition
+    available_items = [item for item in default_sections[section_preferences] if item in menus[menu_preferences]]
+    for item in available_items:
+        if item in command:
+            # Ask for quantity
+            quantity = extract_quantity(command)
+            if quantity:
+                for cart_item in cart:
+                    if cart_item[0] == item:
+                        cart_item[2] += quantity
+                        break
+                else:
+                    cart.append([item, prices[item], quantity])
+                cart_summary = ", ".join([f"{i[0]} x{i[2]} (₹{i[1] * i[2]})" for i in cart])
+                return f"Added {quantity} x {item} to your cart. Current cart: {cart_summary}. Would you like to choose another section or finalize your order?"
+            else:
+                return "Please specify a quantity between 1 and 10."
 
+    # Handle remove command
     if "remove" in command:
         for item in cart:
-            if item[0].lower() in command:
+            if item[0] in command:
                 cart.remove(item)
-                total = sum(i[1] * i[2] for i in cart)
                 cart_summary = ", ".join([f"{i[0]} x{i[2]} (₹{i[1] * i[2]})" for i in cart])
-                return f"Removed {item[0]} from your cart. Updated cart: {cart_summary}. Total: ₹{total}."
+                return f"Removed {item[0]} from your cart. Current cart: {cart_summary}."
         return "The item you are trying to remove is not in your cart."
 
-    if "clear cart" in command or "empty cart" in command:
-        if cart:
-            cart.clear()
-            return "Your cart has been cleared."
-        return "Your cart is already empty."
-
-    if any(item in command for item in menu):
-        closest_match = process.extractOne(command, menu, scorer=fuzz.token_set_ratio)
-        if closest_match and closest_match[1] > 60:
-            matched_item = closest_match[0]
-            for cart_item in cart:
-                if cart_item[0] == matched_item:
-                    cart_item[2] += 1
-                    break
-            else:
-                cart.append([matched_item, prices[matched_item], 1])
-            cart_summary = ", ".join([f"{i[0]} x{i[2]} (₹{i[1] * i[2]})" for i in cart])
-            total = sum(i[1] * i[2] for i in cart)
-            return f"{matched_item.capitalize()} added to your cart. Current cart: {cart_summary}. Total: ₹{total}."
-        return "Sorry, I couldn't recognize the item. Could you try again."
-
+    # Handle final order
     if "final order" in command:
         if cart:
             total = sum(i[1] * i[2] for i in cart)
-            response = f"Your final order is: {', '.join(f'{i[0]} x{i[2]} (₹{i[1] * i[2]})' for i in cart)}. Total bill: ₹{total}. Thank you for ordering! To exit this conversation, say 'goodbye'."
-            cart.clear()
-            return response
+            order_summary = ", ".join([f"{i[0]} x{i[2]} (₹{i[1] * i[2]})" for i in cart])
+            return f"Your final order is: {order_summary}. Total bill: ₹{total}. Thank you for ordering!"
         return "Your cart is empty. Please add items to your cart first."
 
-    if "goodbye" in command:
-        cart.clear()
-        menu_preferences = default_menu_preferences
-        return "Goodbye! Thank you for using AI Dining Assistant."
-
     return "Sorry, I couldn't understand that. Please try again."
 
+def extract_quantity(command):
+    """
+    Extract quantity from the user command.
+    """
+    for word in command.split():
+        if word.isdigit():
+            quantity = int(word)
+            if 1 <= quantity <= 10:
+                return quantity
+    return None
+
 html_code = """
 <!DOCTYPE html>
 <html lang="en">
@@ -271,7 +259,7 @@ html_code = """
             }
         });
         function startConversation() {
-            const utterance = new SpeechSynthesisUtterance('Please choose your preference: All, Vegetarian, Non-Vegetarian, or Guilt-Free.');
+            const utterance = new SpeechSynthesisUtterance('Please choose your preference: All, Vegetarian, or Non-Vegetarian.');
             speechSynthesis.speak(utterance);
             utterance.onend = () => {
                 status.textContent = 'Listening...';
@@ -298,20 +286,10 @@ html_code = """
                         speechSynthesis.speak(utterance);
                         utterance.onend = () => {
                             console.log("Speech synthesis completed.");
-                            if (data.response.includes("Goodbye")) {
-                                status.textContent = 'Conversation ended. Press the mic button to start again.';
-                                isConversationActive = false;
-                                fetch('/reset-cart'); // Reset the cart dynamically on end
-                            } else if (data.response.includes("Your order is complete")) {
-                                status.textContent = 'Order complete. Thank you for using AI Dining Assistant.';
-                                isConversationActive = false;
-                                fetch('/reset-cart'); // Reset the cart after final order
-                            } else {
-                                status.textContent = 'Listening...';
-                                setTimeout(() => {
-                                    startListening();
-                                }, 100);
-                            }
+                            status.textContent = 'Listening...';
+                            setTimeout(() => {
+                                startListening();
+                            }, 100);
                         };
                         utterance.onerror = (e) => {
                             console.error("Speech synthesis error:", e.error);