Update app.py

app.py

@@ -29,15 +29,25 @@ def login(phone):
     return "Customer not found. Please register."
 
 def display_menu():
-    return "\n".join([f"{item['id']}. {item['name']} - ${item['price']}" for item in menu_items])
+    if 'customer_id' not in session_data:
+        return "Please log in first to view the menu."
+    
+    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}"
 
 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."
+    
     cart = session_data.get('cart', [])
     cart.append({"item_id": item_id, "quantity": quantity})
     session_data['cart'] = cart
     return f"Item added to cart. Cart has {len(cart)} item(s)."
 
 def view_cart():
+    if 'customer_id' not in session_data:
+        return "Please log in first to view your cart."
+    
     cart = session_data.get('cart', [])
     cart_items = []
     total_price = 0
@@ -49,6 +59,9 @@ def view_cart():
     return "\n".join(cart_items) + f"\nTotal Price: ${total_price}"
 
 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', [])
     order_details = {
@@ -70,4 +83,3 @@ order_interface = gr.Interface(fn=proceed_to_order, inputs="text", outputs="text
 # Launch Gradio app
 demo = gr.TabbedInterface([login_interface, menu_interface, cart_interface, order_interface], ["Login", "Menu", "Cart", "Order"])
 demo.launch()
-