Update app.py

app.py

@@ -9,7 +9,7 @@ restaurants = {
         "Vegetable Pizza": 15.99,
         "Caesar Salad": 8.99,
         "Garlic Knots": 5.99,
-        "Soda (Coke, Diet Coke, Sprite)": 2.99,
+        "Diet Coke": 2.99,
         "Chocolate Chip Cookie": 2.99
     },
     "Burger King": {
@@ -30,22 +30,22 @@ restaurants = {
         "Baja Blast Freeze": 2.49,
         "Cinnamon Twists": 1.99
     },
-    "McDonald's": {
-        "Big Mac Meal": 6.49,
-        "10-Piece Chicken McNuggets": 4.29,
-        "Filet-O-Fish": 3.79,
-        "Quarter Pounder with Cheese": 5.19,
+    "Mr McDonald's": {
+        "Big Mac-Meal": 6.49,
+        "10-Piece Chicken-McNuggets": 4.29,
+        "Filet Fish": 3.79,
+        "Cheessy Quarte-Pounder": 5.19,
         "French Fries": 1.89,
         "Soft Drink": 1.00,
         "Apple Pie": 0.99
     },
-    "Chipotle": {
+    "Chipotle House": {
         "Burrito Bowl": 7.99,
         "Steak Quesadilla": 4.99,
         "Crispy Tacos": 3.99,
         "Barbacoa Salad": 8.99,
-        "Chips and Guac": 3.99,
-        "Soda": 2.29,
+        "Chips Guac": 3.99,
+        "Soft Drinks": 2.29,
         "Chocolate Brownie": 2.25
     }
 }
@@ -106,18 +106,18 @@ def main(audio):
             command = recognize_speech(audio)
             print(f"You said: {command}")
             txt_command ="extract the food related command keyword from the sentence :\n\n"+command
-            food_command = nlp(txt_command).strip()
+            food_command = (" ".join(nlp(txt_command).strip().split()[-1:])).lower()
 
             restaurant_name = ''
             txt_extract = "extract the restaurants name keyword from the sentence :\n\n"+command
-            restaurant_name = ((nlp(txt_extract)).strip()).title()
+            restaurant_name = " ".join(((nlp(txt_extract)).strip()).title().split()[-2:]))
             found_rest = False
             if(restaurant_name in restaurants.keys()):
               found_rest = True
 
             item_name = ''
             txt_extract = "extract the food item  from the sentence :\n\n"+command
-            item_name = ((nlp(txt_extract)).strip()).title()
+            item_name = " ".join( ((nlp(txt_extract)).strip()).title().strip(".,;:").split()[-2:])
             found_item = False
             if(restaurant_name in restaurants.keys()):
               found_item = True 
@@ -162,6 +162,7 @@ def main(audio):
 
 
             elif food_command in ['menu' , 'menus' , 'catalogue' , 'items' , 'something']:
+             if found_rest:
               for restaurant in restaurants:
                 if restaurant in command:
                   menu_items = get_menu_items(restaurant)
@@ -169,6 +170,8 @@ def main(audio):
                   break
                 else:
                   response= "Sorry, I didn't catch the restaurant name. Could you please repeat that?"
+             else:
+                 response = chatbot(command)
             elif identify_food_command(command) == 'food':
               response=chatbot("Respond a person properly who has come to your restaurant asking food")
             else: