Update app.py

app.py

@@ -12,36 +12,15 @@ def load_menu():
     except Exception as e:
         raise ValueError(f"Error loading menu file: {e}")
 
-# Process user audio input and respond
-def process_audio(audio_file, transcription):
+# Generate menu details
+def generate_menu_details():
     menu_data = load_menu()
-    cart = []
-    response_text = ""
-
-    if "menu" in transcription.lower():
-        menu_details = "\n".join(
-            [f"{item['Dish Name']} - ${item['Price ($)']}" for _, item in menu_data.iterrows()]
-        )
-        response_text = f"Here is our menu:\n{menu_details}"
-    elif any(dish.lower() in transcription.lower() for dish in menu_data['Dish Name']):
-        for _, item in menu_data.iterrows():
-            if item['Dish Name'].lower() in transcription.lower():
-                cart.append(f"{item['Dish Name']} - ${item['Price ($)']}")
-                response_text = f"Order confirmed for {item['Dish Name']}. It has been added to your cart."
-                break
-    elif "remove" in transcription.lower():
-        if cart:
-            removed_item = cart.pop()
-            response_text = f"{removed_item} has been removed from your cart."
-        else:
-            response_text = "Your cart is empty."
-    else:
-        response_text = "I'm sorry, I didn't understand that. Please try again."
-
-    total_bill = sum(float(item.split("$")[1]) for item in cart)
-    tts_path = generate_tts_response(response_text)
-
-    return tts_path, response_text, "\n".join(cart), f"Total Bill: ${total_bill}"
+    menu_details = ""
+    for _, item in menu_data.iterrows():
+        menu_details += f"{item['Dish Name']} - ${item['Price ($)']}\n"
+        menu_details += f"{item['Description']}\n"
+        menu_details += f"[Energy: {item['Energy (kcal)']} kcal, Protein: {item['Protein (g)']}g, Carbohydrates: {item['Carbohydrates (g)']}g, Fiber: {item['Fiber (g)']}g, Fat: {item['Fat (g)']}g, Sugar: {item['Sugar (g)']}g]\n\n"
+    return menu_details
 
 # Text-to-speech response
 def generate_tts_response(text):
@@ -51,27 +30,62 @@ def generate_tts_response(text):
         asyncio.run(communicate.save(tmp_path))
     return tmp_path
 
+# Process user input and respond
+def process_user_input(preference, cart):
+    menu_data = load_menu()
+    response_text = ""
+
+    if "menu" in preference.lower():
+        response_text = "Here are the menu details:\n" + generate_menu_details()
+    else:
+        matched_item = menu_data[menu_data['Dish Name'].str.contains(preference, case=False, na=False)]
+        if not matched_item.empty:
+            dish_name = matched_item.iloc[0]['Dish Name']
+            price = matched_item.iloc[0]['Price ($)']
+            cart.append((dish_name, price))
+            response_text = f"Order confirmed for {dish_name}. It has been added to your cart."
+        elif "remove" in preference.lower():
+            item_to_remove = preference.lower().replace("remove", "").strip()
+            cart = [item for item in cart if item[0].lower() != item_to_remove]
+            response_text = f"Removed {item_to_remove} from your cart."
+        else:
+            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
+
+# Calculate total bill
+def calculate_total(cart):
+    total = sum(item[1] for item in cart)
+    return f"Total: ${total:.2f}"
+
 # Build Gradio app
 def app():
+    cart = []
+
+    def handle_interaction(preference):
+        tts_path, response_text, updated_cart = process_user_input(preference, cart)
+        cart_display = "\n".join([f"{item[0]} - ${item[1]}" for item in updated_cart])
+        total_bill = calculate_total(updated_cart)
+        return tts_path, response_text, cart_display, total_bill
+
     with gr.Blocks() as demo:
         gr.Markdown("# Welcome to the Menu")
 
-        # Audio input and output
-        audio_input = gr.Audio(type="filepath", label="Speak your preference or order")
-        audio_output = gr.Audio(label="Assistant Response", autoplay=True)
-
-        # Textbox for transcription
-        transcription = gr.Textbox(label="Transcription", placeholder="Detected text will appear here")
+        # Input and outputs
+        with gr.Row():
+            preference_input = gr.Textbox(label="Your Request", placeholder="Ask for menu or order items")
+            audio_output = gr.Audio(label="Assistant Response", autoplay=True)
 
-        # Cart output
-        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")
+        transcription_output = gr.Textbox(label="Response Text")
+        cart_output = gr.Textbox(label="Cart")
+        total_output = gr.Textbox(label="Total Bill")
 
-        # Audio processing
-        audio_input.change(
-            process_audio,
-            inputs=[audio_input, transcription],
-            outputs=[audio_output, transcription, cart_output, total_bill_output]
+        # Button for interaction
+        preference_input.submit(
+            handle_interaction,
+            inputs=preference_input,
+            outputs=[audio_output, transcription_output, cart_output, total_output],
         )
 
     return demo