Update app.py

app.py

@@ -1,15 +1,36 @@
 import gradio as gr
+import qrcode
+from PIL import Image
 
-# Sample in-memory database for demonstration
+# Sample Data
 CUSTOMERS = {
     "9876543210": {"name": "John Doe", "email": "john@example.com", "password": "password123"}
 }
 
-def login_or_signup(phone_number, name=None, email=None, password=None):
+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"]}
+    ]
+}
+
+# Generate QR Code
+
+def generate_qr(table_id):
+    qr = qrcode.QRCode(box_size=10, border=5)
+    qr.add_data(f"http://restaurant.com/table/{table_id}")
+    qr.make(fit=True)
+    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).
@@ -18,43 +39,71 @@ def login_or_signup(phone_number, name=None, email=None, password=None):
     Returns:
         str: Status message about login or sign-up.
     """
-    # Check if phone number exists in the database
-    if phone_number in CUSTOMERS:
-        # Customer exists, check for login
-        if CUSTOMERS[phone_number]["password"] == password:
-            return f"Welcome back, {CUSTOMERS[phone_number]['name']}!"
-        else:
-            return "Incorrect password. Please try again."
-    else:
+    if is_new_customer:
         # New customer, proceed with sign-up
-        if not all([name, email, password]):
-            return "Missing details for sign-up. Please provide Name, Email, and Password."
+        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}!"
+    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."
+
+# Gradio Interfaces
 
-# Gradio Interface
 def main():
     with gr.Blocks() as demo:
-        gr.Markdown("### Login or Sign-Up")
-
-        # Input fields for phone number
-        phone_input = gr.Textbox(label="Phone Number", placeholder="Enter your phone number")
-
-        # New customer details
-        name_input = gr.Textbox(label="Name (For New Customers)", placeholder="Enter your name")
-        email_input = gr.Textbox(label="Email (For New Customers)", placeholder="Enter your email")
-        password_input = gr.Textbox(label="Password", placeholder="Enter your password", type="password")
-
-        # Action button
-        action_button = gr.Button("Submit")
-        result_output = gr.Text(label="Status")
-
-        # Function binding
-        action_button.click(
-            login_or_signup,
-            inputs=[phone_input, name_input, email_input, password_input],
-            outputs=result_output
-        )
+        gr.Markdown("# Welcome to Biryani Hub")
+
+        # QR Code Generator
+        with gr.Tab("QR Code Generator"):
+            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="New Customer", value=True)
+            phone_input = gr.Textbox(label="Phone Number", placeholder="Enter your phone number")
+            name_input = gr.Textbox(label="Name", placeholder="Enter your name", visible=True)
+            email_input = gr.Textbox(label="Email", placeholder="Enter your email", visible=True)
+            password_input = gr.Textbox(label="Password", placeholder="Enter your password", type="password")
+            username_input = gr.Textbox(label="Username", placeholder="Enter your username", visible=False)
+
+            def toggle_fields(is_new):
+                return (
+                    is_new,  # Show name input
+                    is_new,  # Show email input
+                    not is_new  # Show username input for existing customers
+                )
+
+            is_new_customer.change(
+                toggle_fields,
+                inputs=[is_new_customer],
+                outputs=[name_input, email_input, username_input]
+            )
+
+            action_button = gr.Button("Submit")
+            result_output = gr.Text(label="Status")
+
+            def handle_input(is_new, phone, name, email, password, username):
+                if not is_new:
+                    name, email = None, None  # Ignore name and email for existing customers
+                return login_or_signup(is_new, phone, name, email, password)
+
+            action_button.click(
+                handle_input,
+                inputs=[is_new_customer, phone_input, name_input, email_input, password_input, username_input],
+                outputs=result_output
+            )
 
     demo.launch()