Update app.py

app.py

@@ -3,14 +3,14 @@ import qrcode
 from PIL import Image
 
 # Sample Data
-CUSTOMERS = {
-    "9876543210": {"name": "John Doe", "email": "john@example.com", "password": "password123"}
-}
-
+CUSTOMERS = {"9876543210": {"name": "John Doe", "preferences": ["Halal", "Indian"]}}
 MENU = {
     "ALL": [
         {"name": "Chicken Curry", "description": "Spicy chicken curry", "image": "chicken_curry.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Small", "Medium", "Large"]},
         {"name": "Paneer Butter Masala", "description": "Creamy paneer dish", "image": "paneer_butter.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Small", "Medium", "Large"]}
+    ],
+    "VEGAN": [
+        {"name": "Veg Biryani", "description": "Flavourful veg biryani", "image": "veg_biryani.jpg", "spice_levels": ["Mild", "Medium"], "portion_sizes": ["Small", "Medium", "Large"]}
     ]
 }
 
@@ -23,37 +23,40 @@ def generate_qr(table_id):
     img = qr.make_image(fill="black", back_color="white")
     return img
 
-# Login or Signup
-
-def login_or_signup(is_new_customer, phone_number, name=None, email=None, password=None):
-    """
-    Handles login or sign-up for customers.
-
-    Args:
-        is_new_customer (bool): Whether the user is a new customer.
-        phone_number (str): Customer's phone number.
-        name (str, optional): Customer's name (required for sign-up).
-        email (str, optional): Customer's email (required for sign-up).
-        password (str, optional): Customer's password (required for both login and sign-up).
-
-    Returns:
-        str: Status message about login or sign-up.
-    """
-    if is_new_customer:
-        # New customer, proceed with sign-up
-        if not all([phone_number, name, email, password]):
-            return "Missing details for sign-up. Please provide Name, Email, Phone Number, and Password."
-        CUSTOMERS[phone_number] = {"name": name, "email": email, "password": password}
-        return f"Account created successfully for {name}!"
+# Login / Signup
+
+def login_or_signup(phone_number):
+    if phone_number in CUSTOMERS:
+        return f"Welcome back, {CUSTOMERS[phone_number]['name']}!", True
     else:
-        # Existing customer, proceed with login
-        if phone_number in CUSTOMERS:
-            if CUSTOMERS[phone_number]["password"] == password:
-                return f"Welcome back, {CUSTOMERS[phone_number]['name']}!"
-            else:
-                return "Incorrect password. Please try again."
-        else:
-            return "Phone number not found. Please sign up."
+        return "Please sign up to continue.", False
+
+# Save Preferences
+
+def save_preferences(phone_number, preferences):
+    if phone_number not in CUSTOMERS:
+        return "Customer not found. Sign up first."
+    CUSTOMERS[phone_number]["preferences"] = preferences
+    return f"Preferences saved: {preferences}"
+
+# Filter Menu
+
+def filter_menu(preferences):
+    filtered_menu = MENU.get(preferences[0].upper(), MENU["ALL"])
+    return filtered_menu
+
+# Dish Details
+
+def dish_details(dish_name):
+    for category in MENU.values():
+        for dish in category:
+            if dish["name"] == dish_name:
+                return {
+                    "Name": dish["name"],
+                    "Description": dish["description"],
+                    "Spice Levels": dish["spice_levels"],
+                    "Portion Sizes": dish["portion_sizes"]
+                }, dish["image"]
 
 # Gradio Interfaces
 
@@ -62,50 +65,40 @@ def main():
         gr.Markdown("# Restaurant QR Code Ordering System")
 
         # QR Code Generator
-        with gr.Tab("QR Code Generator"):
+        with gr.Row():
             table_id = gr.Textbox(label="Table ID", placeholder="Enter table ID")
             qr_button = gr.Button("Generate QR Code")
             qr_output = gr.Image(label="QR Code")
             qr_button.click(lambda table_id: generate_qr(table_id), inputs=table_id, outputs=qr_output)
 
         # Login or Signup
-        with gr.Tab("Login / Signup"):
-            gr.Markdown("## Are you a new or existing customer?")
-            is_new_customer = gr.Checkbox(label="Check if you are a new customer")
-
-            # Input fields
-            phone_input = gr.Textbox(label="Phone Number", placeholder="Enter your phone number")
-            name_input = gr.Textbox(label="Name (For New Customers)", placeholder="Enter your name", visible=False)
-            email_input = gr.Textbox(label="Email (For New Customers)", placeholder="Enter your email", visible=False)
-            password_input_new = gr.Textbox(label="Password (For New Customers)", placeholder="Enter your password", type="password", visible=False)
-            username_input = gr.Textbox(label="Username (For Existing Customers)", placeholder="Enter your username", visible=False)
-            password_input_existing = gr.Textbox(label="Password (For Existing Customers)", placeholder="Enter your password", type="password", visible=False)
-
-            # Visibility logic
-            def toggle_fields(is_new):
-                return (is_new, is_new, is_new, not is_new, not is_new)
-
-            is_new_customer.change(
-                toggle_fields,
-                inputs=[is_new_customer],
-                outputs=[name_input, email_input, password_input_new, username_input, password_input_existing]
-            )
-
-            # Action button
-            action_button = gr.Button("Submit")
-            result_output = gr.Text(label="Status")
-
-            def handle_input(is_new, phone, name, email, password, username, existing_password):
-                if is_new:
-                    return login_or_signup(is_new, phone, name, email, password)
-                else:
-                    return login_or_signup(is_new, phone, name=username, password=existing_password)
-
-            action_button.click(
-                handle_input,
-                inputs=[is_new_customer, phone_input, name_input, email_input, password_input_new, username_input, password_input_existing],
-                outputs=result_output
-            )
+        with gr.Row():
+            phone_input = gr.Textbox(label="Phone Number", placeholder="Enter phone number")
+            login_button = gr.Button("Login / Signup")
+            login_output = gr.Text(label="Login Status")
+            login_button.click(login_or_signup, inputs=phone_input, outputs=login_output)
+
+        # Preferences
+        with gr.Row():
+            preferences = gr.CheckboxGroup(["Vegan", "Halal", "Indian", "Chinese"], label="Preferences")
+            pref_button = gr.Button("Save Preferences")
+            pref_output = gr.Text(label="Preference Status")
+            pref_button.click(save_preferences, inputs=[phone_input, preferences], outputs=pref_output)
+
+        # Menu Display
+        with gr.Row():
+            menu_pref = gr.Dropdown(["ALL", "VEGAN", "HALAL"], label="Menu Type")
+            menu_button = gr.Button("Show Menu")
+            menu_output = gr.JSON(label="Menu")
+            menu_button.click(filter_menu, inputs=menu_pref, outputs=menu_output)
+
+        # Dish Details
+        with gr.Row():
+            dish_name = gr.Textbox(label="Dish Name", placeholder="Enter dish name")
+            dish_button = gr.Button("Get Dish Details")
+            dish_output = gr.JSON(label="Dish Details")
+            dish_image = gr.Image(label="Dish Image")
+            dish_button.click(dish_details, inputs=dish_name, outputs=[dish_output, dish_image])
 
     demo.launch()