import gradio as gr
import pandas as pd
# Load menu data from Excel file
def load_menu(file_path="menu.xlsx"):
menu_df = pd.read_excel(file_path)
menu_df.columns = menu_df.columns.str.strip().str.lower()
column_mapping = {
"name": next((col for col in menu_df.columns if "name" in col), None),
"price": next((col for col in menu_df.columns if "price" in col), None),
"description": next((col for col in menu_df.columns if "description" in col), None),
"image": next((col for col in menu_df.columns if "image" in col), None),
}
missing_columns = [key for key, col in column_mapping.items() if col is None]
if missing_columns:
raise ValueError(f"Excel file is missing one or more required columns: {missing_columns}")
menu = []
for _, row in menu_df.iterrows():
menu.append({
"name": row[column_mapping["name"]],
"price": row[column_mapping["price"]],
"description": row[column_mapping["description"]],
"image": row[column_mapping["image"]],
})
return menu
# Initialize menu and cart
menu_data = load_menu("menu.xlsx")
cart = {}
# Function to render the menu
def display_menu():
"""
Generate the menu as HTML with functional Add buttons tied to Gradio's backend.
"""
menu_html = ""
for item in menu_data:
if item["name"] in cart:
# Quantity selector if the item is already in the cart
menu_html += f"""
{item['name']} - ${item['price']}
{item['description']}
{cart[item['name']]['quantity']}
"""
else:
# Add button if the item is not in the cart
menu_html += f"""
{item['name']} - ${item['price']}
{item['description']}
"""
return menu_html
# Function to add items to the cart
# Add to Cart Logic
def add_to_cart(item_name):
"""
Add an item to the cart and return the updated menu and cart summary.
"""
if item_name in cart:
cart[item_name]["quantity"] += 1
else:
for item in menu_data:
if item["name"] == item_name:
cart[item_name] = {
"name": item["name"],
"price": item["price"],
"quantity": 1
}
break
return display_menu(), update_cart_summary()
# Function to update the cart summary
def update_cart_summary():
num_items = sum(item["quantity"] for item in cart.values())
total_price = sum(item["price"] * item["quantity"] for item in cart.values())
return f"{num_items} items added", f"Total: ${total_price:.2f}"
# Function to render the cart page
def display_cart():
if not cart:
return "Your cart is empty."
cart_html = "
Your Cart:
"
total_price = 0
for item in cart.values():
item_total = item["price"] * item["quantity"]
total_price += item_total
cart_html += f"""
{item['name']}
{item['quantity']}
${item_total:.2f}
"""
cart_html += f"
Total: ${total_price:.2f}
"
cart_html += ''
return cart_html
# Gradio Interface
with gr.Blocks() as demo:
gr.Markdown("# Dynamic Food Menu")
menu_html = gr.HTML(value=display_menu())
cart_summary = gr.HTML(value="No items added to cart.")
# Hidden input for item name
item_name_input = gr.Textbox(visible=False)
# Add Button Logic
add_button = gr.Button("Add to Cart", visible=False)
add_button.click(fn=add_to_cart, inputs=item_name_input, outputs=[menu_html, cart_summary])
# Display Menu and Cart Summary
menu_html.render()
cart_summary.render()
demo.launch()