Update app.py

app.py

@@ -1,118 +1,148 @@
 import gradio as gr
-import json
-
-# Load menu data
-def load_menu_data():
-    try:
-        with open("menu.json", "r") as file:
-            data = json.load(file)
-        return data["menu"]
-    except Exception as e:
-        return f"Error loading menu: {e}"
-
-menu_data = load_menu_data()
-
-# Globals
-current_order = []
-preferences = {}
-
-# Functions
-def login_user(name, phone):
-    if len(phone) != 10 or not phone.isdigit():
-        return "Invalid phone number. Please enter a 10-digit number."
-    return f"Welcome, {name}! Please proceed to set your preferences."
-
-def save_preferences(pref_type, allergies):
-    preferences["type"] = pref_type
-    preferences["allergies"] = allergies
-    return "Preferences saved successfully! Proceed to the menu."
-
-def get_filtered_menu():
-    if not preferences.get("type"):
-        return "Please set preferences first."
-    filtered_menu = [
-        item for item in menu_data
-        if item["type"] == preferences["type"] or preferences["type"] == "Complete Menu"
-    ]
-    return filtered_menu
-
-def generate_menu_buttons():
-    menu_items = get_filtered_menu()
-    if isinstance(menu_items, str):
-        return [gr.Label(value=menu_items)]
-    
-    buttons = []
-    for item in menu_items:
-        buttons.append(
-            gr.Button(
-                value=f"{item['name']} - ₹{item['price']}",
-                elem_id=f"menu_item_{item['id']}",
-                variant="primary"
-            )
-        )
-    return buttons
+import pandas as pd
+import os
+
+# Initialize order and customer database
+DATA_FILE = "customer_data.xlsx"
+if not os.path.exists(DATA_FILE):
+    df = pd.DataFrame(columns=["Name", "Phone", "Preferences", "Allergies", "Order"])
+    df.to_excel(DATA_FILE, index=False)
+
+ORDER_NOTIFICATION_FILE = "orders.xlsx"
+if not os.path.exists(ORDER_NOTIFICATION_FILE):
+    orders_df = pd.DataFrame(columns=["Name", "Phone", "Order"])
+    orders_df.to_excel(ORDER_NOTIFICATION_FILE, index=False)
+
+
+# Save customer data
+def save_customer_data(name, phone, preferences, allergies):
+    df = pd.read_excel(DATA_FILE)
+    new_entry = {"Name": name, "Phone": phone, "Preferences": preferences, "Allergies": allergies, "Order": ""}
+    df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)
+    df.to_excel(DATA_FILE, index=False)
+
+
+# Save order for notification
+def save_order_notification(name, phone, order):
+    orders_df = pd.read_excel(ORDER_NOTIFICATION_FILE)
+    new_order = {"Name": name, "Phone": phone, "Order": order}
+    orders_df = pd.concat([orders_df, pd.DataFrame([new_order])], ignore_index=True)
+    orders_df.to_excel(ORDER_NOTIFICATION_FILE, index=False)
 
-def add_to_order(item_name):
-    current_order.append(item_name)
-    return "\n".join(current_order)
 
-def place_order():
-    if not current_order:
-        return "No items in the order. Please add items to your order before placing it."
-    return f"Order placed successfully! Items: {', '.join(current_order)}."
+# Menu data
+menu = [
+    {"name": "Paneer Butter Masala", "type": "Vegetarian"},
+    {"name": "Chicken Biryani", "type": "Non-Vegetarian"},
+    {"name": "Dal Tadka", "type": "Vegetarian"},
+    {"name": "Fish Curry", "type": "Non-Vegetarian"}
+]
 
-# App Interface
+# Get filtered menu based on preferences
+def get_filtered_menu(preferences):
+    if preferences == "Vegetarian":
+        return [item["name"] for item in menu if item["type"] == "Vegetarian"]
+    elif preferences == "Non-Vegetarian":
+        return [item["name"] for item in menu if item["type"] == "Non-Vegetarian"]
+    return [item["name"] for item in menu]
+
+
+# Gradio app
 with gr.Blocks() as app:
+    # Variables to store user data across screens
+    user_name = gr.State()
+    user_phone = gr.State()
+    user_preferences = gr.State()
+    user_allergies = gr.State()
+    user_order = gr.State([])
+
     # Login Screen
     with gr.Tab("Login"):
-        name = gr.Textbox(label="Name", placeholder="Enter your name")
-        phone = gr.Textbox(label="Phone", placeholder="Enter your 10-digit phone number")
-        login_button = gr.Button("Login")
-        login_output = gr.Label()
-        login_button.click(login_user, inputs=[name, phone], outputs=login_output)
+        gr.Markdown("# Welcome to BH Restaurant")
+        name_input = gr.Textbox(label="Name", placeholder="Enter your name")
+        phone_input = gr.Textbox(label="Phone Number", placeholder="Enter your 10-digit phone number")
+        login_btn = gr.Button("Login")
+        login_output = gr.Textbox(visible=False)
+
+        def login(name, phone):
+            if not name or not phone:
+                return "Please enter both Name and Phone Number.", None, None
+            return "Login successful! Redirecting to preferences...", name, phone
+
+        login_btn.click(login, inputs=[name_input, phone_input], outputs=[login_output, user_name, user_phone])
 
     # Preferences Screen
     with gr.Tab("Preferences"):
-        pref_type = gr.Radio(
-            ["Vegetarian", "Non-Vegetarian", "Complete Menu"],
-            label="Select Preference"
+        gr.Markdown("# Set Your Preferences")
+        preferences_input = gr.Radio(
+            ["Vegetarian", "Non-Vegetarian", "Complete Menu"], label="Food Preferences"
         )
-        allergies = gr.CheckboxGroup(
-            ["Peanuts", "Cheese", "Pork", "Custom"],
-            label="Allergies (if any)"
+        allergies_input = gr.Textbox(label="Any Allergies?")
+        save_pref_btn = gr.Button("Save Preferences")
+        pref_output = gr.Textbox(visible=False)
+
+        def save_preferences(preferences, allergies, name, phone):
+            save_customer_data(name, phone, preferences, allergies)
+            return "Preferences saved! Redirecting to menu...", preferences, allergies
+
+        save_pref_btn.click(
+            save_preferences,
+            inputs=[preferences_input, allergies_input, user_name, user_phone],
+            outputs=[pref_output, user_preferences, user_allergies],
         )
-        save_pref_button = gr.Button("Save Preferences")
-        pref_output = gr.Label()
-        save_pref_button.click(save_preferences, inputs=[pref_type, allergies], outputs=pref_output)
 
     # Menu Screen
     with gr.Tab("Menu"):
-        menu_buttons = gr.Column(elem_id="menu_buttons")
-        current_order_display = gr.Textbox(label="Current Order")
-        proceed_to_summary = gr.Button("Proceed to Order Summary")
-        
-        # Dynamically generate menu buttons after saving preferences
-        save_pref_button.click(generate_menu_buttons, outputs=menu_buttons)
-        
-        # Handle adding items to order
-        for item in menu_data:
-            menu_buttons.append(
-                gr.Button(
-                    value=f"{item['name']} - ₹{item['price']}",
-                    elem_id=f"item-{item['id']}"
-                ).click(
-                    add_to_order,
-                    inputs=None,
-                    outputs=current_order_display
-                )
-            )
-    
+        gr.Markdown("# Menu")
+        menu_dropdown = gr.Dropdown(choices=[], label="Select an Item")
+        add_to_order_btn = gr.Button("Add to Order")
+        order_summary = gr.Textbox(label="Current Order", interactive=False)
+
+        def load_menu(preferences):
+            return gr.update(choices=get_filtered_menu(preferences))
+
+        def add_to_order(item, order_list):
+            if item:
+                order_list.append(item)
+            return ", ".join(order_list), order_list
+
+        menu_dropdown.change(load_menu, inputs=[user_preferences], outputs=[menu_dropdown])
+        add_to_order_btn.click(add_to_order, inputs=[menu_dropdown, user_order], outputs=[order_summary, user_order])
+
     # Order Summary Screen
     with gr.Tab("Order Summary"):
-        order_summary = gr.Textbox(label="Your Order", interactive=False)
-        place_order_button = gr.Button("Place Order")
-        order_output = gr.Label()
-        place_order_button.click(place_order, inputs=None, outputs=order_output)
+        gr.Markdown("# Order Summary")
+        final_order_summary = gr.Textbox(label="Your Final Order", interactive=False)
+        place_order_btn = gr.Button("Place Order")
+        thank_you_output = gr.Textbox(visible=False)
+
+        def place_order(order_list, name, phone):
+            order_str = ", ".join(order_list)
+            save_order_notification(name, phone, order_str)
+            return "Thank you for your order! Redirecting to Thank You page."
+
+        place_order_btn.click(
+            place_order,
+            inputs=[user_order, user_name, user_phone],
+            outputs=[thank_you_output],
+        )
+
+    # Thank You Page
+    with gr.Tab("Thank You"):
+        gr.Markdown("# Thank You!")
+        gr.Markdown("Your order has been placed successfully. Please wait while the chef prepares your meal.")
+
+    # Chef/Waiter Notification Screen
+    with gr.Tab("Chef/Waiter Notifications"):
+        gr.Markdown("# Order Notifications")
+        refresh_btn = gr.Button("Refresh Orders")
+        order_list = gr.Textbox(label="Orders", interactive=False)
+
+        def load_orders():
+            orders_df = pd.read_excel(ORDER_NOTIFICATION_FILE)
+            return orders_df.to_string(index=False)
+
+        refresh_btn.click(load_orders, outputs=[order_list])
 
-# Launch App
+# Launch the app
 app.launch()