Update app.py

app.py

@@ -1,63 +1,105 @@
+import gradio as gr
+import datetime
+from fuzzywuzzy import fuzz
 import pyttsx3
-import speech_recognition as sr
-from transformers import pipeline
-from gtts import gTTS
-import os
 
-def speak(text):
-    tts = gTTS(text)
-    tts.save("response.mp3")
-    os.system("mpg321 response.mp3")
-
-# Initialize Text-to-Speech engine
+# Initialize TTS engine
 engine = pyttsx3.init()
 
-# Set up the Hugging Face conversational pipeline
-conversational_pipeline = pipeline("conversational", model="microsoft/DialoGPT-medium")
+# Menu with Indian dishes and prices
+menu = {
+    "Starters": {
+        "Paneer Tikka": 150.0,
+        "Samosa": 20.0,
+        "Pakora": 50.0
+    },
+    "Main Course": {
+        "Butter Chicken": 350.0,
+        "Paneer Butter Masala": 300.0,
+        "Dal Makhani": 200.0,
+        "Veg Biryani": 180.0,
+        "Chicken Biryani": 220.0
+    },
+    "Breads": {
+        "Tandoori Roti": 20.0,
+        "Butter Naan": 30.0,
+        "Stuffed Paratha": 50.0
+    },
+    "Desserts": {
+        "Gulab Jamun": 40.0,
+        "Rasgulla": 35.0,
+        "Kheer": 60.0
+    },
+    "Drinks": {
+        "Masala Chai": 15.0,
+        "Lassi": 50.0,
+        "Buttermilk": 20.0
+    }
+}
 
-def speak(text):
-    """Function to convert text to speech"""
-    engine.say(text)
-    engine.runAndWait()
+# Function to wish the user based on the time of day
+def wish_user():
+    hour = int(datetime.datetime.now().hour)
+    if hour >= 0 and hour < 12:
+        return "Good Morning! How can I help you today?"
+    elif hour >= 12 and hour < 18:
+        return "Good Afternoon! What would you like to order?"
+    else:
+        return "Good Evening! Hope you're doing well!"
+
+# Function to recognize and process the command
+def recognize_command(command):
+    if not command:
+        return "Sorry, I couldn't process your request. Please try again."
+
+    for category, items in menu.items():
+        for item in items:
+            if fuzz.partial_ratio(command.lower(), item.lower()) > 80:
+                return f"Got it! {item} has been added to your order."
+
+    if "menu" in command.lower():
+        menu_text = "Here is the menu for the restaurant:\n"
+        for category, items in menu.items():
+            menu_text += f"\n{category}:\n"
+            for item, price in items.items():
+                menu_text += f"  - {item}: ₹{price}\n"
+        return menu_text
+
+    if "price" in command.lower():
+        return "Which item would you like to know the price for? (Type the item name)"
 
-def listen():
-    """Function to listen to audio and convert it to text using speech recognition"""
-    recognizer = sr.Recognizer()
-
-    with sr.Microphone() as source:
-        print("Listening...")
-        recognizer.adjust_for_ambient_noise(source)
-        audio = recognizer.listen(source)
-        
-        try:
-            print("Recognizing...")
-            text = recognizer.recognize_google(audio)
-            print(f"You said: {text}")
-            return text
-        except sr.UnknownValueError:
-            print("Sorry, I did not understand that.")
-            return None
-        except sr.RequestError:
-            print("Sorry, the service is down.")
-            return None
-
-def respond_to_query(query):
-    """Function to process query using Hugging Face's conversational pipeline"""
-    response = conversational_pipeline(query)
-    return response[0]['generated_text']
-
-def main():
-    """Main function to run the voice assistant"""
-    speak("Hello! How can I assist you today?")
-    
-    while True:
-        user_input = listen()
-        if user_input:
-            if 'exit' in user_input.lower():
-                speak("Goodbye!")
-                break
-            response = respond_to_query(user_input)
-            speak(response)
+    if "exit" in command.lower() or "bye" in command.lower():
+        return "Thank you for visiting! Have a great day!"
+
+    return "Sorry, I didn't understand that. Could you please repeat?"
+
+# Function to provide the price of an item
+def provide_price(item):
+    for category, items in menu.items():
+        if item in items:
+            price = items[item]
+            return f"The price for {item} is ₹{price}."
+    return f"Sorry, {item} is not available in the menu."
+
+# Gradio interface functions
+def assistant_response(command):
+    if "price for" in command.lower():
+        item = command.lower().replace("price for", "").strip()
+        return provide_price(item)
+    return recognize_command(command)
+
+# Gradio Interface
+interface = gr.Interface(
+    fn=assistant_response,
+    inputs=gr.Textbox(lines=2, placeholder="Type your request here..."),
+    outputs="text",
+    title="Restaurant Voice Assistant",
+    description="Interact with the restaurant assistant! Ask for the menu, prices, or place orders."
+)
 
 if __name__ == "__main__":
-    main()
+    wish_message = wish_user()
+    print(wish_message)
+    engine.say(wish_message)
+    engine.runAndWait()
+    interface.launch()