Update app.py

app.py

@@ -1,115 +1,119 @@
 import gradio as gr
+import pandas as pd
+import qrcode
+import os
 
-# Data Module
-menu_data = [
-    {
-        "name": "Veg Burger",
-        "category": "VEGAN",
-        "image": "https://upload.wikimedia.org/wikipedia/commons/6/6e/Veggie_burger.jpg",
-        "description": "A delicious vegan burger with plant-based patty, lettuce, and tomato."
-    },
-    {
-        "name": "Chicken Biryani",
-        "category": "HALAL",
-        "image": "https://upload.wikimedia.org/wikipedia/commons/7/79/Chicken_Biryani.jpg",
-        "description": "Spicy chicken biryani with aromatic basmati rice and tender chicken pieces."
-    },
-    {
-        "name": "Paneer Butter Masala",
-        "category": "VEGAN",
-        "image": "https://upload.wikimedia.org/wikipedia/commons/3/3d/Paneer_Butter_Masala.jpg",
-        "description": "Paneer cooked in a rich and creamy tomato-based gravy."
-    },
-]
-
-# HTML Rendering Module
-def render_dish_grid(category):
-    filtered_dishes = [
-        dish for dish in menu_data if category == "ALL" or dish["category"] == category
-    ]
-    html_content = "<div style='display: flex; flex-wrap: wrap; justify-content: center;'>"
-    for dish in filtered_dishes:
-        html_content += f"""
-        <div style='margin: 10px; text-align: center; cursor: pointer; border: 1px solid #ddd; 
-                    padding: 10px; border-radius: 10px; width: 200px; background-color: #f9f9f9;'
-             onclick='selectDish("{dish['name']}")'>
-            <img src='{dish['image']}' alt='{dish['name']}' 
-                 style='width: 180px; height: 120px; object-fit: cover; border-radius: 10px;'>
-            <h4 style='margin-top: 10px; font-size: 16px; color: #444;'>{dish['name']}</h4>
-        </div>
-        """
-    html_content += "</div>"
-    return html_content
-
-# Business Logic Module
-def get_dish_details(dish_name):
-    for dish in menu_data:
-        if dish["name"] == dish_name:
-            return gr.update(visible=True), dish["image"], dish["name"], dish["description"]
-    return gr.update(visible=False), "", "Dish Not Found", "No description available."
-
-def close_popup():
-    return gr.update(visible=False)
-
-# Gradio App Module
-def create_app():
-    with gr.Blocks() as demo:
-        gr.HTML("<h1 style='text-align: center;'>🍛 Biryani Menu 🍛</h1>")
-
-        # JavaScript for dish selection
-        gr.HTML("""
-        <script>
-        function selectDish(dishName) {
-            let input = document.querySelector('input[aria-label="selected_dish_name"]');
-            if (input) {
-                input.value = dishName;
-                input.dispatchEvent(new Event('input', { bubbles: true }));
-            }
-        }
-        </script>
-        """)
-
-        gr.Markdown("### Select a Category")
+# Data storage (for prototyping; switch to Salesforce or database for production)
+customer_data_path = "customer_data.xlsx"
+order_data_path = "order_data.xlsx"
+
+# Ensure data files exist
+if not os.path.exists(customer_data_path):
+    pd.DataFrame(columns=["Name", "Email", "Phone", "Preferences", "Allergies"]).to_excel(customer_data_path, index=False)
+
+if not os.path.exists(order_data_path):
+    pd.DataFrame(columns=["Table", "Customer", "Order", "Status"]).to_excel(order_data_path, index=False)
+
+# Sample menu (can be dynamically loaded from an external file)
+menu = {
+    "Bread": ["Roti", "Naan", "Paratha"],
+    "Veg Curries": ["Paneer Butter Masala", "Mix Veg", "Dal Tadka"],
+    "Non-Veg Curries": ["Chicken Curry", "Mutton Korma", "Fish Curry"],
+    "Biryanis": ["Veg Biryani", "Chicken Biryani", "Mutton Biryani"]
+}
+
+# Generate QR Code for each table
+def generate_qr(table_number):
+    qr = qrcode.make(f"https://your-huggingface-space/{table_number}")
+    qr_path = f"table_{table_number}_qr.png"
+    qr.save(qr_path)
+    return qr_path
+
+# Sign-Up functionality
+def sign_up(name, email, phone, preferences, allergies):
+    customer_data = pd.read_excel(customer_data_path)
+    if email in customer_data["Email"].values:
+        return "Email already exists! Please log in."
+    new_entry = {"Name": name, "Email": email, "Phone": phone, "Preferences": preferences, "Allergies": allergies}
+    customer_data = customer_data.append(new_entry, ignore_index=True)
+    customer_data.to_excel(customer_data_path, index=False)
+    return "Sign-Up successful! Please log in."
+
+# Login functionality
+def login(email):
+    customer_data = pd.read_excel(customer_data_path)
+    if email in customer_data["Email"].values:
+        customer = customer_data[customer_data["Email"] == email].iloc[0]
+        return f"Welcome back, {customer['Name']}!", customer.to_dict()
+    return "Email not found! Please sign up.", None
+
+# Display menu based on preferences
+def display_menu(preferences, allergies):
+    filtered_menu = {}
+    for category, items in menu.items():
+        filtered_menu[category] = [item for item in items if item not in allergies]
+    return filtered_menu
+
+# Add item to order
+def add_to_order(table, customer, item):
+    order_data = pd.read_excel(order_data_path)
+    new_order = {"Table": table, "Customer": customer, "Order": item, "Status": "Pending"}
+    order_data = order_data.append(new_order, ignore_index=True)
+    order_data.to_excel(order_data_path, index=False)
+    return "Item added to order."
+
+# Gradio Interface
+with gr.Blocks() as app:
+    gr.Markdown("# QR-Based Restaurant Ordering System")
+
+    # QR Code Section
+    with gr.Tab("Generate QR Code"):
+        table_input = gr.Number(label="Table Number", value=1)
+        qr_output = gr.Image(label="QR Code")
+        generate_button = gr.Button("Generate QR Code")
+
+        generate_button.click(generate_qr, inputs=table_input, outputs=qr_output)
+
+    # Sign-Up and Login Section
+    with gr.Tab("Sign Up / Login"):
         with gr.Row():
-            btn_all = gr.Button("ALL")
-            btn_vegan = gr.Button("VEGAN")
-            btn_halal = gr.Button("HALAL")
-
-        gr.Markdown("### Available Dishes")
-        dish_display = gr.HTML(value=render_dish_grid("ALL"))
-
-        # Pop-up for dish details
-        popup = gr.Group(visible=False)
-        with popup:
-            with gr.Row():
-                dish_image = gr.Image()
-                with gr.Column():
-                    dish_name = gr.Textbox(label="Dish Name", interactive=False)
-                    dish_description = gr.Textbox(label="Description", interactive=False)
-            close_button = gr.Button("Close")
-
-        # Hidden textbox to handle JavaScript -> Python communication
-        selected_dish_name = gr.Textbox(visible=False)
-        selected_dish_name.change(
-            fn=get_dish_details,
-            inputs=selected_dish_name,
-            outputs=[popup, dish_image, dish_name, dish_description]
-        )
-
-        # Close button functionality
-        close_button.click(fn=close_popup, outputs=popup)
-
-        # Update category buttons
-        def update_dish_grid(category):
-            return gr.update(value=render_dish_grid(category))
-
-        btn_all.click(fn=lambda: update_dish_grid("ALL"), outputs=dish_display)
-        btn_vegan.click(fn=lambda: update_dish_grid("VEGAN"), outputs=dish_display)
-        btn_halal.click(fn=lambda: update_dish_grid("HALAL"), outputs=dish_display)
-
-    return demo
-
-# Main Execution
-if __name__ == "__main__":
-    app = create_app()
-    app.launch()
+            with gr.Column():
+                gr.Markdown("### Sign Up")
+                name_input = gr.Textbox(label="Name")
+                email_input = gr.Textbox(label="Email")
+                phone_input = gr.Textbox(label="Phone")
+                preferences_input = gr.Textbox(label="Preferences (e.g., Vegetarian)")
+                allergies_input = gr.Textbox(label="Allergies (e.g., Peanuts)")
+                sign_up_button = gr.Button("Sign Up")
+                sign_up_output = gr.Textbox(label="Sign-Up Status")
+
+                sign_up_button.click(sign_up, 
+                                    inputs=[name_input, email_input, phone_input, preferences_input, allergies_input],
+                                    outputs=sign_up_output)
+
+            with gr.Column():
+                gr.Markdown("### Login")
+                login_email_input = gr.Textbox(label="Email")
+                login_button = gr.Button("Login")
+                login_output = gr.Textbox(label="Login Status")
+
+                login_button.click(login, inputs=login_email_input, outputs=login_output)
+
+    # Menu and Ordering Section
+    with gr.Tab("Menu & Order"):
+        preferences_input = gr.Textbox(label="Preferences (from Login)", interactive=False)
+        allergies_input = gr.Textbox(label="Allergies (from Login)", interactive=False)
+        menu_display = gr.JSON(label="Filtered Menu")
+        display_menu_button = gr.Button("Show Menu")
+
+        display_menu_button.click(display_menu, inputs=[preferences_input, allergies_input], outputs=menu_display)
+
+        item_input = gr.Textbox(label="Item to Add")
+        table_input = gr.Number(label="Table Number", value=1)
+        customer_input = gr.Textbox(label="Customer Name")
+        add_order_button = gr.Button("Add to Order")
+        add_order_output = gr.Textbox(label="Order Status")
+
+        add_order_button.click(add_to_order, inputs=[table_input, customer_input, item_input], outputs=add_order_output)
+
+app.launch()