Update app.py

app.py

@@ -4,98 +4,106 @@ import pandas as pd
 # Load menu from Excel file
 def load_menu(file_path="menu.xlsx"):
     """
-    Load menu data from an Excel file dynamically based on column headers.
+    Load menu data from an Excel file.
     """
     menu_df = pd.read_excel(file_path)
-    # Automatically detect column names
-    columns = menu_df.columns.str.strip()  # Trim column names to avoid issues with spaces
-    name_column = [col for col in columns if "name" in col.lower()][0]
-    price_column = [col for col in columns if "price" in col.lower()][0]
-    description_column = [col for col in columns if "description" in col.lower()][0]
-    image_column = [col for col in columns if "image" in col.lower()][0]
-
-    menu = []
-    for _, row in menu_df.iterrows():
-        menu.append({
-            "name": row[name_column],
-            "price": row[price_column],
-            "description": row[description_column],
-            "image": row[image_column]
-        })
+    menu = menu_df.to_dict(orient="records")
     return menu
 
-# Initialize the menu and cart
+# Initialize menu and cart
 menu_data = load_menu()
 cart = []
 
 def display_menu():
     """
-    Function to display the menu in a formatted style with an 'Add' button for each item.
+    Function to render the menu as HTML.
     """
     menu_html = ""
     for item in menu_data:
         menu_html += f"""
         <div style="display: flex; margin-bottom: 15px; align-items: center; border: 1px solid #ddd; padding: 10px; border-radius: 5px;">
-            <img src="{item['image']}" alt="{item['name']}" style="width: 100px; height: 100px; margin-right: 15px; border-radius: 5px;">
+            <img src="{item['Image']}" alt="{item['Name']}" style="width: 100px; height: 100px; margin-right: 15px; border-radius: 5px;">
             <div style="flex-grow: 1;">
-                <h4 style="margin: 0;">{item['name']} - ${item['price']}</h4>
-                <p style="margin: 5px 0; font-size: 12px;">{item['description']}</p>
+                <h4 style="margin: 0;">{item['Name']} - ${item['Price']}</h4>
+                <p style="margin: 5px 0; font-size: 12px;">{item['Description']}</p>
+                <label for="quantity-{item['Name']}" style="margin-right: 10px;">Quantity:</label>
+                <input type="number" id="quantity-{item['Name']}" value="1" min="1" style="width: 50px;">
             </div>
-            <button style="background-color: #28a745; color: white; border: none; padding: 8px 12px; border-radius: 5px; cursor: pointer;" onclick="add_to_cart('{item['name']}', 1)">Add</button>
+            <button onclick="add_to_cart('{item['Name']}', document.getElementById('quantity-{item['Name']}').value)" style="background-color: #28a745; color: white; border: none; padding: 8px 12px; border-radius: 5px; cursor: pointer;">Add</button>
         </div>
         """
     return menu_html
 
-def add_to_cart(food_item, quantity):
+def add_to_cart(item_name, quantity):
     """
-    Add an item to the cart with its price and quantity.
+    Add an item to the cart and return success message with cart count.
     """
-    for item in menu_data:
-        if item["name"] == food_item:
-            cart.append({
-                "item": food_item,
-                "quantity": quantity,
-                "price": item["price"] * quantity
-            })
+    quantity = int(quantity)
+    # Check if the item already exists in the cart
+    for item in cart:
+        if item["name"] == item_name:
+            item["quantity"] += quantity
             break
-    return f"{food_item} added to cart!"
+    else:
+        # Add new item to cart
+        for item in menu_data:
+            if item["Name"] == item_name:
+                cart.append({"name": item_name, "price": item["Price"], "quantity": quantity})
+                break
+    return f"{item_name} added to cart! Total items: {len(cart)}"
 
 def view_cart():
     """
-    View the cart items and calculate the total price.
+    Render the cart contents and calculate the total price.
     """
     if not cart:
         return "Your cart is empty."
     
-    cart_summary = "<h3>Cart Summary:</h3><ul>"
+    cart_html = "<h3>Cart Summary:</h3><ul>"
     total_price = 0
     for item in cart:
-        cart_summary += f"<li>{item['item']} (x{item['quantity']}) - ${item['price']}</li>"
-        total_price += item["price"]
-    cart_summary += f"</ul><h4>Total Price: ${total_price}</h4>"
-    return cart_summary
+        cart_html += f"<li>{item['name']} (x{item['quantity']}) - ${item['price'] * item['quantity']}</li>"
+        total_price += item["price"] * item["quantity"]
+    cart_html += f"</ul><h4>Total Price: ${total_price}</h4>"
+    return cart_html
 
-# Gradio Interface with Pages
+# Gradio Interface
 def menu_page():
     return gr.update(value=display_menu(), visible=True), gr.update(visible=False)
 
 def cart_page():
-    return gr.update(visible=False), gr.update(value=view_cart(), visible=True)
+    return gr.update(value=view_cart(), visible=True), gr.update(visible=False)
 
 with gr.Blocks() as demo:
     gr.Markdown("# Dynamic Food Menu")
     
     # Menu and Cart Views
-    menu_display = gr.HTML(visible=True)
+    menu_display = gr.HTML(value=display_menu(), visible=True)
     cart_display = gr.HTML(visible=False)
+    notification = gr.Textbox(label="Notification", value="", interactive=False)
     
     # Navigation Buttons
     with gr.Row():
-        menu_button = gr.Button("Menu")
-        cart_button = gr.Button("Cart")
+                menu_button = gr.Button("Menu")
+        cart_button = gr.Button("View Cart")
     
-    # Event Handling for Navigation
+    # Add item interaction
+    food_item_input = gr.Textbox(visible=False)  # Hidden input to capture item name
+    quantity_input = gr.Number(visible=False)   # Hidden input to capture quantity
+    add_button = gr.Button("Add to Cart", visible=False)  # Simulated button for functionality
+
+    # Notification when item is added
+    add_button.click(add_to_cart, inputs=[food_item_input, quantity_input], outputs=notification)
+    
+    # Navigation between Menu and Cart pages
     menu_button.click(menu_page, outputs=[menu_display, cart_display])
-    cart_button.click(cart_page, outputs=[menu_display, cart_display])
+    cart_button.click(cart_page, outputs=[cart_display, menu_display])
+    
+    # Render menu and cart
+    demo += menu_display
+    demo += cart_display
+    demo += notification
 
 demo.launch()
+
+