Update order_assistant.py

order_assistant.py

@@ -1,21 +1,18 @@
-import gradio as gr
+import json
+import random
 import speech_recognition as sr
 import edge_tts
 import asyncio
-import random
 import tempfile
 
-# Example food menu
-food_menu = {
-    "Starters": ["Spring Rolls", "Samosa", "Garlic Bread"],
-    "Main Course": ["Biryani", "Pizza", "Pasta", "Butter Chicken"],
-    "Desserts": ["Ice Cream", "Gulab Jamun", "Chocolate Cake"]
-}
+# Load the food menu from menu.json
+with open("menu.json") as f:
+    food_menu = json.load(f)
 
-# Function to simulate food suggestions based on user input
-def get_food_suggestion(food_type):
+# Function to get food suggestions based on user input
+def get_food_suggestion(food_type, filter_type):
     if food_type in food_menu:
-        return random.choice(food_menu[food_type])
+        return random.choice(food_menu[food_type].get(filter_type, []))
     return "No suggestion available"
 
 # Speech-to-Text function
@@ -40,14 +37,14 @@ async def text_to_speech(text):
     return tmp_path
 
 # AI Food Ordering Assistant Function
-async def food_order_assistant():
+async def food_order_assistant(text_input, food_type):
     # Greeting
     greeting_text = "Welcome to the restaurant! Please tell me your food preference."
     audio_path = await text_to_speech(greeting_text)
 
     # Wait for user to input food type
-    user_food_type = recognize_speech_from_mic()
-    suggestion = get_food_suggestion(user_food_type)
+    filter_type = text_input.lower()  # Assuming the input is the filter type (vegan, halal, etc.)
+    suggestion = get_food_suggestion(food_type, filter_type)
 
     suggestion_text = f"I suggest you try {suggestion}."
     suggestion_audio = await text_to_speech(suggestion_text)