Update app.py

app.py

@@ -35,23 +35,21 @@ def login(phone):
     else:
         return "Phone number not found. Please register first."
 
-# Display menu after login
-def display_menu():
+# Display menu and allow adding items to cart
+def menu(item_name, quantity):
     if 'customer_id' not in session_data:
-        return "Please log in first to view the menu."
+        return "Please log in first to view the menu and add items to the cart."
     
-    menu = "\n".join([f"{item['id']}. {item['name']} - ${item['price']}" for item in menu_items])
-    return f"Welcome, {session_data['customer_name']}! Here is the menu:\n\n{menu}"
-
-# Add item to cart
-def add_to_cart(item_id, quantity):
-    if 'customer_id' not in session_data:
-        return "Please log in first to add items to your cart."
+    # Find the selected item in the menu
+    selected_item = next((item for item in menu_items if item["name"] == item_name), None)
+    if not selected_item:
+        return f"Item '{item_name}' not found in the menu."
     
+    # Add the item to the cart
     cart = session_data.get('cart', [])
-    cart.append({"item_id": item_id, "quantity": quantity})
+    cart.append({"item_id": selected_item['id'], "name": selected_item['name'], "quantity": quantity, "price": selected_item['price']})
     session_data['cart'] = cart
-    return f"Item added to cart. Cart has {len(cart)} item(s)."
+    return f"Added {quantity} x {selected_item['name']} to the cart."
 
 # View cart contents
 def view_cart():
@@ -59,22 +57,26 @@ def view_cart():
         return "Please log in first to view your cart."
     
     cart = session_data.get('cart', [])
-    cart_items = []
+    if not cart:
+        return "Your cart is empty."
+    
+    cart_summary = []
     total_price = 0
     for item in cart:
-        item_details = next((menu_item for menu_item in menu_items if menu_item['id'] == item['item_id']), None)
-        if item_details:
-            cart_items.append(f"{item_details['name']} x {item['quantity']} - ${item_details['price'] * item['quantity']}")
-            total_price += item_details['price'] * item['quantity']
-    return "\n".join(cart_items) + f"\nTotal Price: ${total_price}"
+        cart_summary.append(f"{item['name']} x {item['quantity']} - ${item['price'] * item['quantity']:.2f}")
+        total_price += item['price'] * item['quantity']
+    return "\n".join(cart_summary) + f"\n\nTotal Price: ${total_price:.2f}"
 
 # Proceed to order function
 def proceed_to_order(table_number):
     if 'customer_id' not in session_data:
         return "Please log in first to proceed with your order."
     
-    order_id = str(uuid.uuid4())  # Generate unique order ID
     cart = session_data.get('cart', [])
+    if not cart:
+        return "Your cart is empty. Add items to your cart before proceeding."
+    
+    order_id = str(uuid.uuid4())  # Generate unique order ID
     order_details = {
         "order_id": order_id,
         "customer_id": session_data['customer_id'],
@@ -83,23 +85,39 @@ def proceed_to_order(table_number):
         "items": cart
     }
     session_data['cart'] = []  # Clear cart after order
-    return f"Order ID: {order_id}\nTable Number: {table_number}\nOrder Details: {order_details}"
+    return f"Order placed successfully!\n\nOrder ID: {order_id}\nTable Number: {table_number}\n\nOrder Details:\n" + "\n".join([f"{item['name']} x {item['quantity']} - ${item['price'] * item['quantity']:.2f}" for item in cart])
 
-# Gradio Interface for Signup (Registration)
+# Gradio Interface for Signup
 signup_interface = gr.Interface(fn=signup, inputs=["text", "text"], outputs="text", title="Signup")
 
 # Gradio Interface for Login
 login_interface = gr.Interface(fn=login, inputs="text", outputs="text", title="Login")
 
 # Gradio Interface for Menu
-menu_interface = gr.Interface(fn=display_menu, inputs=None, outputs="text", title="Menu")
+menu_interface = gr.Interface(
+    fn=menu,
+    inputs=[
+        gr.Dropdown(choices=[item["name"] for item in menu_items], label="Select Item"),
+        gr.Number(label="Quantity", value=1, precision=0)
+    ],
+    outputs="text",
+    title="Menu"
+)
 
 # Gradio Interface for Cart
 cart_interface = gr.Interface(fn=view_cart, inputs=None, outputs="text", title="Cart")
 
 # Gradio Interface for Proceed to Order
-order_interface = gr.Interface(fn=proceed_to_order, inputs="text", outputs="text", title="Proceed to Order")
+order_interface = gr.Interface(
+    fn=proceed_to_order,
+    inputs=gr.Textbox(label="Table Number"),
+    outputs="text",
+    title="Proceed to Order"
+)
 
 # Launch Gradio app with multiple tabs
-demo = gr.TabbedInterface([signup_interface, login_interface, menu_interface, cart_interface, order_interface], ["Signup", "Login", "Menu", "Cart", "Order"])
+demo = gr.TabbedInterface(
+    [signup_interface, login_interface, menu_interface, cart_interface, order_interface],
+    ["Signup", "Login", "Menu", "Cart", "Order"]
+)
 demo.launch()