Update app.py

app.py

@@ -1,106 +1,7 @@
+import asyncio
 import gradio as gr
 import pandas as pd
-import edge_tts
-import asynciimport gradio as gr
-from gtts import gTTS
-import os
-import tempfile
-import json
-import speech_recognition as sr
-
-# Store cart in a temporary storage
-cart = []
-
-# Define the menu items dynamically
-menu_items = {
-    "Pizza": 10.99,
-    "Burger": 8.49,
-    "Pasta": 12.99,
-    "Salad": 7.99,
-    "Soda": 2.49
-}
-
-def generate_voice_response(text):
-    tts = gTTS(text)
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
-    temp_file.close()
-    tts.save(temp_file.name)
-    return temp_file.name
-
-def calculate_total(cart):
-    return sum(menu_items[item] for item in cart)
-
-def restaurant_voice_assistant(audio, state_json):
-    global cart
-    state = json.loads(state_json) if state_json else {}
-    response = ""
-    voice_path = None
-
-    # Convert audio input to text
-    if audio:
-        recognizer = sr.Recognizer()
-        with sr.AudioFile(audio) as source:
-            try:
-                input_text = recognizer.recognize_google(recognizer.record(source))
-            except sr.UnknownValueError:
-                input_text = ""
-    else:
-        input_text = ""
-
-    if not state.get("menu_shown", False):
-        # Show menu dynamically
-        response = "Welcome to our restaurant! Here is our menu:\n"
-        for item, price in menu_items.items():
-            response += f"{item}: ${price:.2f}\n"
-        response += "\nPlease tell me the item you would like to add to your cart."
-        state["menu_shown"] = True
-    elif any(item.lower() in input_text.lower() for item in menu_items):
-        # Check if input matches a menu item
-        for item in menu_items:
-            if item.lower() in input_text.lower():
-                cart.append(item)
-                total = calculate_total(cart)
-                response = f"{item} has been added to your cart. Your current cart includes:\n"
-                for cart_item in cart:
-                    response += f"- {cart_item}: ${menu_items[cart_item]:.2f}\n"
-                response += f"\nTotal: ${total:.2f}. Would you like to add anything else?"
-                break
-    elif "menu" in input_text.lower():
-        response = "Here is our menu again:\n"
-        for item, price in menu_items.items():
-            response += f"{item}: ${price:.2f}\n"
-        response += "\nWhat would you like to add to your cart?"
-    elif "final order" in input_text.lower() or "submit order" in input_text.lower():
-        if cart:
-            total = calculate_total(cart)
-            response = "Your final order includes:\n"
-            for item in cart:
-                response += f"- {item}: ${menu_items[item]:.2f}\n"
-            response += f"\nTotal: ${total:.2f}.\nThank you for ordering!"
-            cart = []  # Clear cart after finalizing order
-        else:
-            response = "Your cart is empty. Would you like to order something?"
-    else:
-        response = "I didn’t quite catch that. Please tell me what you’d like to order."
-
-    voice_path = generate_voice_response(response)
-    return response, voice_path, json.dumps(state)
-
-with gr.Blocks() as demo:
-    state = gr.State(value=json.dumps({}))
-
-    with gr.Row():
-        user_audio = gr.Audio(type="filepath", label="Your Voice Input")
-        output_text = gr.Textbox(label="Response Text")
-
-    with gr.Row():
-        voice_output = gr.Audio(label="Response Audio", autoplay=True)
-
-    # Automatically process audio when recording stops
-    user_audio.change(restaurant_voice_assistant, inputs=[user_audio, state], outputs=[output_text, voice_output, state])
-
-demo.launch()
-o
+from edge_tts import Communicate
 import tempfile
 
 # Load menu data
@@ -111,91 +12,74 @@ def load_menu():
     except Exception as e:
         raise ValueError(f"Error loading menu file: {e}")
 
+# Handle menu inquiries
+def handle_menu_request():
+    menu_data = load_menu()
+    menu_details = ""
+    for _, item in menu_data.iterrows():
+        menu_details += f"{item['Dish Name']} - ${item['Price ($)']}\n"
+    return menu_details
+
 # Generate TTS response
-async def generate_tts_response(text):
-    communicate = edge_tts.Communicate(text)
+def generate_tts_response(text):
+    communicate = Communicate(text)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
         tmp_path = tmp_file.name
-        await communicate.save(tmp_path)
+        asyncio.run(communicate.save(tmp_path))
     return tmp_path
 
-# Global cart to store orders
-cart = []
-menu_data = load_menu()
-
-# Add item to cart
-def add_to_cart(item_name):
-    global cart
-    item = menu_data[menu_data["Dish Name"].str.lower() == item_name.lower()]
-    if not item.empty:
-        cart.append({"Dish Name": item.iloc[0]["Dish Name"], "Price": item.iloc[0]["Price ($)"]})
-        return f"{item_name} has been added to your cart."
-    return f"{item_name} is not available on the menu."
-
-# Remove item from cart
-def remove_from_cart(item_name):
-    global cart
-    for i, item in enumerate(cart):
-        if item["Dish Name"].lower() == item_name.lower():
-            cart.pop(i)
-            return f"{item_name} has been removed from your cart."
-    return f"{item_name} is not in your cart."
-
-# Get cart summary and total bill
-def get_cart_summary():
-    global cart
-    if not cart:
-        return "Your cart is empty.", 0
-    summary = ", ".join(f"{item['Dish Name']} (${item['Price']})" for item in cart)
-    total = sum(item["Price"] for item in cart)
-    return summary, total
-
 # Process customer input
-async def process_customer_input(audio_file):
-    # Simulated transcription (Replace with actual transcription logic)
-    transcription = "veg samosa"  # Replace with transcription logic
-
-    if "menu" in transcription.lower():
-        menu_details = "Here is our menu: " + ", ".join(
-            f"{item['Dish Name']} - ${item['Price ($)']}" for _, item in menu_data.iterrows()
-        )
-        tts_path = await generate_tts_response(menu_details)
-        return tts_path, menu_details
-
-    elif "remove" in transcription.lower():
-        item_name = transcription.replace("remove", "").strip()
-        response_text = remove_from_cart(item_name)
-        tts_path = await generate_tts_response(response_text)
-        return tts_path, response_text
-
+def process_customer_input(input_text, cart):
+    response_text = ""
+    menu_data = load_menu()
+    if "menu" in input_text.lower():
+        response_text = "Here is the menu:\n" + handle_menu_request()
+    elif any(dish.lower() in input_text.lower() for dish in menu_data['Dish Name']):
+        dish_name = next(dish for dish in menu_data['Dish Name'] if dish.lower() in input_text.lower())
+        cart.append(dish_name)
+        response_text = f"{dish_name} has been added to your cart."
+    elif "remove" in input_text.lower():
+        for dish in cart:
+            if dish.lower() in input_text.lower():
+                cart.remove(dish)
+                response_text = f"{dish} has been removed from your cart."
+                break
+        else:
+            response_text = "Item not found in cart to remove."
+    elif "cart" in input_text.lower():
+        if cart:
+            response_text = "Your cart contains:\n" + "\n".join(cart)
+        else:
+            response_text = "Your cart is empty."
     else:
-        response_text = add_to_cart(transcription)
-        tts_path = await generate_tts_response(response_text)
-        return tts_path, response_text
+        response_text = "I'm sorry, I didn't understand that. Please try again."
+
+    tts_path = generate_tts_response(response_text)
+    return tts_path, response_text, cart
 
-# Build Gradio app
+# Gradio app
 def app():
-    with gr.Blocks() as demo:
-        gr.Markdown("# Welcome to the Biryani Hub")
+    cart = []
 
-        # Audio input and output
-        customer_audio_input = gr.Audio(label="Click to speak your order", type="filepath")
-        customer_audio_output = gr.Audio(label="Assistant Response", autoplay=True)
+    with gr.Blocks() as demo:
+        gr.Markdown("# Welcome to the Menu")
 
-        # Cart and total bill outputs
+        # Input and output components
+        audio_input = gr.Audio(source="microphone", type="filepath", label="Speak your preference or order")
+        audio_output = gr.Audio(label="Assistant Response", autoplay=True)
+        transcription = gr.Textbox(label="Transcription", placeholder="Detected text will appear here")
         cart_output = gr.Textbox(label="Cart", placeholder="Your cart details will appear here")
-        total_bill_output = gr.Textbox(label="Total Bill", placeholder="Your total bill will appear here")
 
-        # Process customer input logic
-        async def handle_customer_audio(audio):
-            tts_path, response_text = await process_customer_input(audio)
-            cart_summary, total_bill = get_cart_summary()
-            return tts_path, response_text, cart_summary, f"${total_bill}"
+        # Define interaction
+        def handle_interaction(audio_file):
+            input_text = ""  # Placeholder for transcription logic
+            tts_path, response_text, updated_cart = process_customer_input(input_text, cart)
+            return tts_path, response_text, "\n".join(updated_cart)
 
-        customer_audio_input.change(
-            lambda audio: asyncio.run(handle_customer_audio(audio)),
-            inputs=[customer_audio_input],
-            outputs=[customer_audio_output, gr.Textbox(), cart_output, total_bill_output],
+        audio_input.change(
+            handle_interaction,
+            inputs=[audio_input],
+            outputs=[audio_output, transcription, cart_output]
         )
 
     return demo