Update app.py

app.py

@@ -1,129 +1,148 @@
 import gradio as gr
 import pandas as pd
-import json
 import os
 
-# Load menu data
-with open("menu.json", "r") as f:
-    menu = json.load(f)
-
-# Initialize customer database
+# 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", "Table", "Order"])
+    df = pd.DataFrame(columns=["Name", "Phone", "Preferences", "Allergies", "Order"])
     df.to_excel(DATA_FILE, index=False)
 
-# Function to save customer data
-def save_customer_data(name, phone, preferences, allergies, table):
+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,
-        "Table": table,
-        "Order": ""
-    }
+    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)
 
-# Function to get menu based on preferences
+
+# 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)
+
+
+# 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"}
+]
+
+# Get filtered menu based on preferences
 def get_filtered_menu(preferences):
-    if preferences == "Complete Menu":
-        return menu
-    return [item for item in menu if item["type"] == preferences]
-
-# Function to display item details
-def show_item_details(item_name):
-    item = next((x for x in menu if x["name"] == item_name), None)
-    if item:
-        return (
-            item["image"],  # Image
-            f"**{item['name']}**\nPortion: {item['portion']}\nNutrition: {item['nutrition']}"
+    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"):
+        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"):
+        gr.Markdown("# Set Your Preferences")
+        preferences_input = gr.Radio(
+            ["Vegetarian", "Non-Vegetarian", "Complete Menu"], label="Food Preferences"
+        )
+        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],
         )
-    return None, "Item not found."
-
-# Main application workflow
-def main_workflow(table_id):
-    # Define Gradio blocks
-    with gr.Blocks() as app:
-        login_tab = gr.Tab("Login")
-        preferences_tab = gr.Tab("Preferences")
-        menu_tab = gr.Tab("Menu")
-        order_tab = gr.Tab("Order Summary")
-
-        # Login Tab
-        with login_tab:
-            gr.Markdown("# Welcome to BH Restaurant")
-            with gr.Row():
-                name = gr.Textbox(label="Name", placeholder="Enter your name")  # Removed required=True
-                phone = gr.Textbox(label="Phone Number", placeholder="Enter your 10-digit phone number")  # Removed required=True
-            login_btn = gr.Button("Login")
-            login_output = gr.Textbox(visible=False)
-
-            # Update the login function to manually check if the fields are filled
-            def login(name, phone):
-                if not name or not phone:
-                    return "Please enter both Name and Phone Number.", None  # Check if inputs are empty
-                # Check for existing user
-                df = pd.read_excel(DATA_FILE)
-                if not df[df["Phone"] == phone].empty:
-                    return "Welcome back! Proceeding to the menu...", preferences_tab
-                return "New user detected. Proceed to preferences.", preferences_tab
-
-            login_btn.click(login, inputs=[name, phone], outputs=[login_output, preferences_tab])
-
-        # Preferences Tab
-        with preferences_tab:
-            gr.Markdown("# Preferences")
-            preferences = gr.Radio(["Vegetarian", "Vegan", "Halal", "Complete Menu"], label="Food Preferences")
-            allergies = gr.Textbox(label="Any Allergies?")
-            save_pref_btn = gr.Button("Save Preferences")
-
-            def save_preferences(name, phone, preferences, allergies):
-                save_customer_data(name, phone, preferences, allergies, table_id)
-                return f"Preferences saved for {name}. Proceeding to menu..."
-
-            save_pref_btn.click(save_preferences, inputs=[name, phone, preferences, allergies], outputs=[menu_tab])
-
-        # Menu Tab
-        with menu_tab:
-            gr.Markdown("# Menu")
-            menu_dropdown = gr.Dropdown(label="Menu", choices=[])
-            menu_details_btn = gr.Button("Show Details")
-            menu_image = gr.Image()
-            menu_info = gr.Markdown()
-            menu_add_btn = gr.Button("Add to Order")
-            order_summary = gr.Textbox(placeholder="Order summary")
-
-            def display_menu(preferences):
-                filtered_menu = get_filtered_menu(preferences)
-                return gr.update(choices=[item["name"] for item in filtered_menu])
-
-            menu_details_btn.click(show_item_details, inputs=menu_dropdown, outputs=[menu_image, menu_info])
-            menu_add_btn.click(lambda selected_items: order_summary.update(value=", ".join(selected_items)), inputs=menu_dropdown, outputs=order_summary)
-
-        # Order Tab
-        with order_tab:
-            gr.Markdown("# Order Summary")
-            order_output = gr.Textbox(label="Your Order", interactive=False)
-            place_order_btn = gr.Button("Place Order")
-            order_status = gr.Textbox(visible=False)
-
-            def place_order(selected_items):
-                # Save order to the customer data file
-                df = pd.read_excel(DATA_FILE)
-                customer_data = df[df["Table"] == table_id].iloc[0]
-                order = selected_items
-                df.loc[df["Table"] == table_id, "Order"] = ", ".join(order)
-                df.to_excel(DATA_FILE, index=False)
-                return "Order placed successfully! The kitchen has been notified."
-
-            place_order_btn.click(place_order, inputs=order_summary, outputs=order_status)
-
-    return app
-
-# Create Gradio app for table 1
-if __name__ == "__main__":
-    app = main_workflow(table_id=1)
-    app.launch()
+
+    # Menu Screen
+    with gr.Tab("Menu"):
+        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"):
+        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 the app
+app.launch()