Update app.py

app.py

@@ -20,31 +20,57 @@ async def generate_tts_response(text):
         await communicate.save(tmp_path)
     return tmp_path
 
-# Welcome Note Handler
-async def play_welcome_note():
-    welcome_text = "Hello, welcome to Biryani Hub. What do you want to order?"
-    return await generate_tts_response(welcome_text), welcome_text
+# 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 = "menu"  # This should be replaced with audio transcription logic
+    transcription = "veg samosa"  # Replace with transcription logic
 
     if "menu" in transcription.lower():
-        menu_data = load_menu()
         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 "veg samosa" in transcription.lower():
-        response_text = "Order confirmed for Veg Samosa. It has been added to your cart."
+    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 = "I'm sorry, I didn't understand that. Please try again."
+        response_text = add_to_cart(transcription)
         tts_path = await generate_tts_response(response_text)
         return tts_path, response_text
 
@@ -53,39 +79,24 @@ def app():
     with gr.Blocks() as demo:
         gr.Markdown("# Welcome to the Menu")
 
-        # Welcome note audio output
-        welcome_audio_output = gr.Audio(label="Welcome Note", autoplay=True)
-
-        # Customer audio input and response
+        # 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)
-        transcription_output = gr.Textbox(label="Transcription", placeholder="Detected text will appear here")
 
-        # Cart details
+        # 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")
 
-        # Play welcome note
-        def play_welcome():
-            tts_path, welcome_text = asyncio.run(play_welcome_note())
-            return tts_path, welcome_text
-
-        # Process customer input
+        # Process customer input logic
         async def handle_customer_audio(audio):
             tts_path, response_text = await process_customer_input(audio)
-            return tts_path, response_text
+            cart_summary, total_bill = get_cart_summary()
+            return tts_path, response_text, cart_summary, f"${total_bill}"
 
-        # Welcome button logic
-        gr.Button("Play Welcome Note").click(
-            play_welcome,
-            inputs=[],
-            outputs=[welcome_audio_output, transcription_output],
-        )
-
-        # Process customer input logic
         customer_audio_input.change(
             lambda audio: asyncio.run(handle_customer_audio(audio)),
             inputs=[customer_audio_input],
-            outputs=[customer_audio_output, transcription_output],
+            outputs=[customer_audio_output, gr.Textbox(), cart_output, total_bill_output],
         )
 
     return demo