Update app.py

app.py

@@ -17,20 +17,23 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # Init TTS
 tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
 
-# Menu Data (You can modify this as per your actual menu)
+# Updated Menu Data (Including Drinks, Vegetarian, Non-Vegetarian)
 def load_menu():
     menu = {
-        "Breads": [
-            {"name": "Roti", "price": 1.50, "description": "Indian flatbread"},
-            {"name": "Naan", "price": 2.00, "description": "Soft leavened flatbread"}
+        "Drinks": [
+            {"name": "Coca-Cola", "price": 1.50, "description": "Refreshing carbonated soft drink"},
+            {"name": "Lemonade", "price": 1.20, "description": "Freshly made lemon drink with a sweet twist"},
+            {"name": "Mango Lassi", "price": 2.00, "description": "Smooth yogurt drink with mango flavor"}
         ],
-        "Curries": [
-            {"name": "Butter Chicken", "price": 7.99, "description": "Creamy and rich chicken curry"},
-            {"name": "Paneer Tikka Masala", "price": 6.99, "description": "Cottage cheese in a spiced gravy"}
+        "Vegetarian": [
+            {"name": "Paneer Butter Masala", "price": 6.99, "description": "Cottage cheese in a creamy, spiced gravy"},
+            {"name": "Chole Bhature", "price": 5.50, "description": "Spicy chickpeas served with deep-fried bread"},
+            {"name": "Aloo Gobi", "price": 5.00, "description": "Potatoes and cauliflower cooked with spices"}
         ],
-        "Biryanis": [
-            {"name": "Chicken Biryani", "price": 8.99, "description": "Fragrant rice with spiced chicken"},
-            {"name": "Vegetable Biryani", "price": 7.50, "description": "Fragrant rice with mixed vegetables"}
+        "Non-Vegetarian": [
+            {"name": "Butter Chicken", "price": 7.99, "description": "Grilled chicken cooked in a rich, creamy sauce"},
+            {"name": "Chicken Tikka", "price": 6.50, "description": "Marinated chicken chunks grilled to perfection"},
+            {"name": "Lamb Rogan Josh", "price": 8.99, "description": "Tender lamb cooked in aromatic spices and gravy"}
         ]
     }
     return menu
@@ -38,18 +41,18 @@ def load_menu():
 # Function to process commands and get menu details
 def process_command(command):
     menu = load_menu()
-    
+
     if 'menu' in command.lower():
         return "Here's our menu: \n" + "\n".join([f"{category}: {', '.join([item['name'] for item in items])}" for category, items in menu.items()])
     
-    elif 'breads' in command.lower():
-        return "Our breads: " + ", ".join([item['name'] for item in menu["Breads"]])
+    elif 'drinks' in command.lower():
+        return "Our drinks: " + ", ".join([item['name'] for item in menu["Drinks"]])
 
-    elif 'curries' in command.lower():
-        return "Our curries: " + ", ".join([item['name'] for item in menu["Curries"]])
+    elif 'vegetarian' in command.lower():
+        return "Our vegetarian dishes: " + ", ".join([item['name'] for item in menu["Vegetarian"]])
 
-    elif 'biryani' in command.lower():
-        return "Our biryanis: " + ", ".join([item['name'] for item in menu["Biryanis"]])
+    elif 'non-vegetarian' in command.lower():
+        return "Our non-vegetarian dishes: " + ", ".join([item['name'] for item in menu["Non-Vegetarian"]])
 
     return "I'm sorry, I didn't understand that command."
 
@@ -90,7 +93,7 @@ def interact_with_assistant(user_input, speaker_audio, language):
 iface = gr.Interface(fn=interact_with_assistant, 
                      inputs=[gr.Textbox(lines=2, placeholder="Enter the text...", label="Text"),
                              gr.Audio(type="filepath", label="Upload audio file"),
-                             gr.Radio([ 'english'], label="language"),
+                             gr.Radio(['ru', 'en', 'zh-cn', 'ja', 'de', 'fr', 'it', 'pt', 'pl', 'tr', 'ko', 'nl', 'cs', 'ar', 'es', 'hu'], label="Language"),
                             ],
                      outputs=gr.Audio(type="filepath", label="Generated audio file"), 
                      title="Voice Assistant - Menu and Voice Cloning")