Update app.py

app.py

@@ -12,79 +12,91 @@ def load_menu():
     except Exception as e:
         raise ValueError(f"Error loading menu file: {e}")
 
-# Text-to-speech response
-def generate_tts_response(text):
+# Generate TTS response
+async def generate_tts_response(text):
     communicate = edge_tts.Communicate(text)
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
         tmp_path = tmp_file.name
-        asyncio.run(communicate.save(tmp_path))
+        await communicate.save(tmp_path)
     return tmp_path
 
-# Process user input for orders and menu
-cart = {}
-def process_input(user_input):
+# Global cart to store orders
+cart = []
+menu_data = load_menu()
+
+# Add item to cart
+def add_to_cart(item_name):
     global cart
-    menu_data = load_menu()
-    response_text = ""
-
-    if "menu" in user_input.lower():
-        response_text = "Here are the menu items: "
-        for _, item in menu_data.iterrows():
-            response_text += f"{item['Dish Name']} - ${item['Price ($)']}. "
-
-    elif any(item['Dish Name'].lower() in user_input.lower() for _, item in menu_data.iterrows()):
-        for _, item in menu_data.iterrows():
-            if item['Dish Name'].lower() in user_input.lower():
-                dish_name = item['Dish Name']
-                price = item['Price ($)']
-                if dish_name in cart:
-                    cart[dish_name]['quantity'] += 1
-                else:
-                    cart[dish_name] = {"price": price, "quantity": 1}
-                response_text = f"Order confirmed for {dish_name}. It has been added to your cart."
-                break
-
-    elif "remove" in user_input.lower():
-        for dish_name in list(cart.keys()):
-            if dish_name.lower() in user_input.lower():
-                del cart[dish_name]
-                response_text = f"{dish_name} has been removed from your cart."
-                break
-        else:
-            response_text = "The item you want to remove is not in the cart."
-
-    elif "cart" in user_input.lower():
-        if cart:
-            response_text = "Your cart contains: "
-            total = 0
-            for dish_name, details in cart.items():
-                response_text += f"{dish_name} (x{details['quantity']}) - ${details['price'] * details['quantity']}. "
-                total += details['price'] * details['quantity']
-            response_text += f"Total: ${total}"
-        else:
-            response_text = "Your cart is empty."
+    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."
 
-    else:
-        response_text = "I'm sorry, I didn't understand that. Please try again."
+# 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
 
-    tts_path = generate_tts_response(response_text)
-    return tts_path, response_text
+    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
+
+    else:
+        response_text = add_to_cart(transcription)
+        tts_path = await generate_tts_response(response_text)
+        return tts_path, response_text
 
 # Build Gradio app
 def app():
     with gr.Blocks() as demo:
         gr.Markdown("# Welcome to the Biryani Hub")
 
-        # Input and Output components
-        user_input = gr.Textbox(label="Your Request", placeholder="Ask about the menu, place an order, or manage your cart.")
-        audio_output = gr.Audio(label="Assistant Response", autoplay=True)
-        response_text = gr.Textbox(label="Assistant Response (Text)", interactive=False)
+        # 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)
+
+        # Cart and total bill outputs
+        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}"
 
-        # Process input button
-        user_input.submit(
-            process_input, 
-            inputs=[user_input], 
-            outputs=[audio_output, response_text]
+        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],
         )
 
     return demo