Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import uuid
 
-# Sample data for menu items (You can replace these with your actual menu items)
+# Sample menu items (Add your actual items here)
 menu_items = [
     {"id": 1, "name": "Burger", "price": 5.99},
     {"id": 2, "name": "Pizza", "price": 8.99},
@@ -10,17 +10,17 @@ menu_items = [
     {"id": 5, "name": "Soda", "price": 1.99}
 ]
 
-# Simulated database (In real applications, use a real database)
-customers = {}  # Stores customers by phone number
+# Simulated database for customers and session management
+customers = {}
 session_data = {}
 
-# Function for user registration (signup)
+# Function for user signup
 def signup(name, phone):
     if phone in customers:
         return f"Phone number {phone} is already registered. Please log in."
     
-    # Create a unique customer ID
-    customer_id = f"CUST{str(uuid.uuid4().int)[:6]}"  # Shortened unique customer ID
+    # Generate unique customer ID
+    customer_id = f"CUST{str(uuid.uuid4().int)[:6]}"
     customers[phone] = {"name": name, "phone": phone, "customer_id": customer_id}
     
     return f"Signup successful! Welcome {name}. Your customer ID is {customer_id}."
@@ -35,23 +35,23 @@ def login(phone):
     else:
         return "Phone number not found. Please register first."
 
-# Display menu and allow adding items to cart
-def menu(item_name, quantity):
+# Function to add items to the cart
+def add_to_cart(item_name):
     if 'customer_id' not in session_data:
-        return "Please log in first to view the menu and add items to the cart."
+        return "Please log in first to add items to the 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
+    # Add item to cart
     cart = session_data.get('cart', [])
-    cart.append({"item_id": selected_item['id'], "name": selected_item['name'], "quantity": quantity, "price": selected_item['price']})
+    cart.append({"item_id": selected_item['id'], "name": selected_item['name'], "quantity": 1, "price": selected_item['price']})
     session_data['cart'] = cart
-    return f"Added {quantity} x {selected_item['name']} to the cart."
+    return f"Added 1 x {selected_item['name']} to the cart."
 
-# View cart contents
+# Function to display cart
 def view_cart():
     if 'customer_id' not in session_data:
         return "Please log in first to view your cart."
@@ -63,11 +63,11 @@ def view_cart():
     cart_summary = []
     total_price = 0
     for item in cart:
-        cart_summary.append(f"{item['name']} x {item['quantity']} - ${item['price'] * item['quantity']:.2f}")
-        total_price += item['price'] * item['quantity']
+        cart_summary.append(f"{item['name']} x {item['quantity']} - ${item['price']:.2f}")
+        total_price += item['price']
     return "\n".join(cart_summary) + f"\n\nTotal Price: ${total_price:.2f}"
 
-# Proceed to order function
+# Function to proceed to order
 def proceed_to_order(table_number):
     if 'customer_id' not in session_data:
         return "Please log in first to proceed with your order."
@@ -85,7 +85,15 @@ def proceed_to_order(table_number):
         "items": cart
     }
     session_data['cart'] = []  # Clear cart after order
-    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])
+    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']:.2f}" for item in cart])
+
+# Dynamic menu interface with add buttons
+def generate_menu():
+    menu_elements = []
+    for item in menu_items:
+        button = gr.Button(f"Add {item['name']} (${item['price']:.2f})")
+        menu_elements.append((button, lambda name=item['name']: add_to_cart(name)))
+    return menu_elements
 
 # Gradio Interface for Signup
 signup_interface = gr.Interface(fn=signup, inputs=["text", "text"], outputs="text", title="Signup")
@@ -93,16 +101,15 @@ signup_interface = gr.Interface(fn=signup, inputs=["text", "text"], outputs="tex
 # 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=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 Menu with Add Buttons
+menu_buttons = generate_menu()
+menu_interface = gr.Blocks()
+
+with menu_interface:
+    with gr.Row():
+        gr.Markdown("# Menu")
+    for button, function in menu_buttons:
+        button.click(fn=function, outputs="text")
 
 # Gradio Interface for Cart
 cart_interface = gr.Interface(fn=view_cart, inputs=None, outputs="text", title="Cart")