Update app.py

app.py

@@ -25,7 +25,7 @@ def generate_voice_response(text):
     return temp_file.name
 
 def calculate_total(cart):
-    return sum(menu_items[item] for item in cart)
+    return len(cart)  # Only count items, not prices
 
 def restaurant_voice_assistant(audio, state_json):
     global cart
@@ -46,46 +46,53 @@ def restaurant_voice_assistant(audio, state_json):
     if not state.get("menu_shown", False):
         # Show menu dynamically
         response = "Welcome to our restaurant! Here is our menu:\n"
-        for item, price in menu_items.items():
-            response += f"{item}: ${price:.2f}\n"
-        response += "\nPlease tell me the item you would like to add to your cart."
+        for item in menu_items.keys():
+            response += f"{item}\n"
+        response += "\nPlease tell me the item you would like to add to your cart or ask for the price of an item."
         state["menu_shown"] = True
     elif not input_text.strip():
         # Wait for valid input without responding if no input is provided
         return "", None, json.dumps(state)
-    else:
+    elif "price of" in input_text:
+        # Handle price queries
+        matched_items = [item for item in menu_items if item.lower() in input_text]
+        if len(matched_items) == 1:
+            item = matched_items[0]
+            response = f"The price of {item} is ${menu_items[item]:.2f}."
+        elif len(matched_items) > 1:
+            response = f"I detected multiple items in your input: {', '.join(matched_items)}. Please ask for the price of one item at a time."
+        else:
+            response = "I couldn't find that item on the menu. Please ask for an item available in the menu."
+    elif any(item.lower() in input_text for item in menu_items):
         # Match the input text with menu items
         matched_items = [item for item in menu_items if item.lower() in input_text and item not in state.get("current_items", [])]
-
         if len(matched_items) == 1:
             item = matched_items[0]
             cart.append(item)
             state.setdefault("current_items", []).append(item)  # Track items added in the current cycle
-            total = calculate_total(cart)
             response = f"{item} has been added to your cart. Your current cart includes:\n"
             for cart_item in cart:
-                response += f"- {cart_item}: ${menu_items[cart_item]:.2f}\n"
-            response += f"\nTotal: ${total:.2f}. Would you like to add anything else?"
+                response += f"- {cart_item}\n"
+            response += "\nWould you like to add anything else?"
         elif len(matched_items) > 1:
             response = f"I detected multiple items in your input: {', '.join(matched_items)}. Please mention one item at a time."
-        elif "menu" in input_text:
-            response = "Here is our menu again:\n"
-            for item, price in menu_items.items():
-                response += f"{item}: ${price:.2f}\n"
-            response += "\nWhat would you like to add to your cart?"
-        elif "final order" in input_text or "submit order" in input_text:
-            if cart:
-                total = calculate_total(cart)
-                response = "Your final order includes:\n"
-                for item in cart:
-                    response += f"- {item}: ${menu_items[item]:.2f}\n"
-                response += f"\nTotal: ${total:.2f}.\nThank you for ordering!"
-                cart = []  # Clear cart after finalizing order
-                state["current_items"] = []  # Clear current cycle tracking
-            else:
-                response = "Your cart is empty. Would you like to order something?"
+    elif "menu" in input_text:
+        response = "Here is our menu again:\n"
+        for item in menu_items.keys():
+            response += f"{item}\n"
+        response += "\nWhat would you like to add to your cart or ask about?"
+    elif "final order" in input_text or "submit order" in input_text:
+        if cart:
+            response = "Your final order includes:\n"
+            for item in cart:
+                response += f"- {item}\n"
+            response += f"\nThank you for ordering!"
+            cart = []  # Clear cart after finalizing order
+            state["current_items"] = []  # Clear current cycle tracking
         else:
-            response = "I didn’t quite catch that. Please tell me what you’d like to order."
+            response = "Your cart is empty. Would you like to order something?"
+    else:
+        response = "I didn’t quite catch that. Please tell me what you’d like to order or ask about."
 
     voice_path = generate_voice_response(response)
     return response, voice_path, json.dumps(state)