import gradio as gr
import pandas as pd

# Function to load the menu data from Excel
def load_menu():
    menu_file = "menu.xlsx"  # Make sure this file exists in the same directory
    try:
        return pd.read_excel(menu_file)
    except Exception as e:
        raise ValueError(f"Error loading menu file: {e}")

# Function to filter menu items based on preference
def filter_menu(preference):
    # Load menu data
    menu_data = load_menu()

    # Check for the required columns
    required_columns = ["Category", "Dish Name", "Price ($)", "Image URL", "Description"]
    for col in required_columns:
        if col not in menu_data.columns:
            raise ValueError(f"Missing column in Excel file: {col}")

    # Filter data based on preference
    if preference != "All":
        filtered_data = menu_data[menu_data["Category"].str.contains(preference, case=False, na=False)]
    else:
        filtered_data = menu_data

    # Generate HTML for the menu
    html_content = ""
    for _, item in filtered_data.iterrows():
        html_content += f"""
        <div style="display: flex; align-items: center; border: 1px solid #ddd; border-radius: 8px; padding: 15px; margin-bottom: 10px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);">
            <div style="flex: 1; margin-right: 15px;">
                <h3 style="margin: 0; font-size: 18px;">{item['Dish Name']}</h3>
                <p style="margin: 5px 0; font-size: 16px; color: #888;">${item['Price ($)']}</p>
                <p style="margin: 5px 0; font-size: 14px; color: #555;">{item['Description']}</p>
            </div>
            <div style="flex-shrink: 0; text-align: center;">
                <img src="{item['Image URL']}" alt="{item['Dish Name']}" style="width: 100px; height: 100px; border-radius: 8px; object-fit: cover; margin-bottom: 10px;">
                <button style="background-color: #28a745; color: white; border: none; padding: 8px 15px; font-size: 14px; border-radius: 5px; cursor: pointer;">Add</button>
            </div>
        </div>
        """
    return html_content

# Gradio app definition
def app():
    with gr.Blocks(title="Dynamic Menu with Preferences") as demo:
        # Radio button for selecting preference
        selected_preference = gr.Radio(
            choices=["All", "Vegan", "Halal", "Guilt-Free"],
            value="All",
            label="Choose a Preference",
        )

        # Output area for menu items
        menu_output = gr.HTML(value=filter_menu("All"))

        # Define interactivity
        selected_preference.change(filter_menu, inputs=[selected_preference], outputs=[menu_output])

        # Layout
        gr.Markdown("## Dynamic Menu with Preferences")
        with gr.Row():
            selected_preference.render()
        menu_output.render()

    return demo

# Run the app
if __name__ == "__main__":
    demo = app()
    demo.launch()