AIVoice / order_assistant.py
DSatishchandra's picture
Update order_assistant.py
b0c4baa verified
raw
history blame
2.31 kB
import json
import random
import speech_recognition as sr
import edge_tts
import asyncio
import tempfile
# Load the food menu from menu.json
with open("menu.json") as f:
food_menu = json.load(f)
# 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].get(filter_type, []))
return "No suggestion available"
# Speech-to-Text function
def recognize_speech_from_mic():
recognizer = sr.Recognizer()
with sr.Microphone() as source:
print("Say something!")
audio = recognizer.listen(source)
try:
return recognizer.recognize_google(audio)
except sr.UnknownValueError:
return "Sorry I didn't catch that"
except sr.RequestError:
return "Sorry, I'm having trouble reaching the service"
# Text-to-Speech function
async def text_to_speech(text):
communicate = edge_tts.Communicate(text, "en-US-AriaNeural", rate="0%", pitch="0Hz")
with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
tmp_path = tmp_file.name
await communicate.save(tmp_path)
return tmp_path
# AI Food Ordering Assistant Function
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
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)
# Wait for confirmation
confirmation = recognize_speech_from_mic()
if "yes" in confirmation.lower():
confirmation_text = f"Your order for {suggestion} has been confirmed."
confirmation_audio = await text_to_speech(confirmation_text)
return confirmation_audio, f"Confirmed order: {suggestion}"
else:
cancellation_text = "Okay, let's try again."
cancellation_audio = await text_to_speech(cancellation_text)
return cancellation_audio, "Order not confirmed"