Update app.py

app.py

@@ -1,50 +1,51 @@
 import gradio as gr
 import pandas as pd
 
-# Load menu from Excel file
+# Load menu data from Excel file
 def load_menu(file_path="menu.xlsx"):
     """
     Load menu data from an Excel file dynamically based on column headers.
     """
     menu_df = pd.read_excel(file_path)
+    # Normalize column names (strip spaces and make lowercase)
+    menu_df.columns = menu_df.columns.str.strip().str.lower()
     
-    # Automatically normalize column names
-    columns = menu_df.columns.str.strip()  # Trim leading/trailing 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]
-
+    # Check for required columns
+    required_columns = {"name", "price", "description", "image"}
+    if not required_columns.issubset(set(menu_df.columns)):
+        raise ValueError(f"Excel file is missing one or more required columns: {required_columns}")
+    
+    # Map data to required structure
     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]
+            "name": row["name"],
+            "price": row["price"],
+            "description": row["description"],
+            "image": row["image"]
         })
     return menu
 
 # Initialize menu and cart
-menu_data = load_menu()
+menu_data = load_menu("menu.xlsx")
 cart = []
 
 def display_menu():
     """
-    Function to render the menu as HTML.
+    Render the menu dynamically 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>
-                <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;">
+                <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 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>
+            <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
@@ -62,8 +63,8 @@ def add_to_cart(item_name, quantity):
     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})
+            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)}"
 
@@ -120,6 +121,3 @@ with gr.Blocks() as demo:
     demo += notification
 
 demo.launch()
-
-
-