Update app.py

app.py

@@ -25,11 +25,12 @@ def generate_qr(table_id):
 
 # Login or Signup
 
-def login_or_signup(phone_number, name=None, email=None, password=None):
+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).
@@ -38,19 +39,21 @@ 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
 
@@ -67,6 +70,8 @@ def main():
 
         # Login or Signup
         with gr.Tab("Login / Signup"):
+            gr.Markdown("## Are you a new or existing customer?")
+            is_new_customer = gr.Radio(["New Customer", "Existing Customer"], label="Select Customer Type")
             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")
             email_input = gr.Textbox(label="Email (For New Customers)", placeholder="Enter your email")
@@ -75,9 +80,13 @@ def main():
             action_button = gr.Button("Submit")
             result_output = gr.Text(label="Status")
 
+            def handle_input(customer_type, phone, name, email, password):
+                is_new = customer_type == "New Customer"
+                return login_or_signup(is_new, phone, name, email, password)
+
             action_button.click(
-                login_or_signup,
-                inputs=[phone_input, name_input, email_input, password_input],
+                handle_input,
+                inputs=[is_new_customer, phone_input, name_input, email_input, password_input],
                 outputs=result_output
             )