Update app.py

app.py

@@ -3,28 +3,14 @@ import qrcode
 from PIL import Image
 
 # Sample Data
-CUSTOMERS = {}
+CUSTOMERS = {
+    "9876543210": {"name": "John Doe", "email": "john@example.com", "password": "password123"}
+}
+
 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"]},
-        {"name": "Veg Biryani", "description": "Flavourful veg biryani", "image": "veg_biryani.jpg", "spice_levels": ["Mild", "Medium"], "portion_sizes": ["Small", "Medium", "Large"]},
-        {"name": "Butter Naan", "description": "Soft buttery naan", "image": "butter_naan.jpg", "spice_levels": [], "portion_sizes": ["Single", "Double"]},
-        {"name": "Tandoori Chicken", "description": "Chargrilled chicken", "image": "tandoori_chicken.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Quarter", "Half", "Full"]},
-        {"name": "Dal Makhani", "description": "Rich lentil curry", "image": "dal_makhani.jpg", "spice_levels": ["Mild", "Medium"], "portion_sizes": ["Small", "Large"]},
-        {"name": "Chole Bhature", "description": "Spicy chickpeas with fried bread", "image": "chole_bhature.jpg", "spice_levels": ["Mild", "Medium", "Spicy"], "portion_sizes": ["Single", "Double"]},
-        {"name": "Egg Curry", "description": "Eggs in spicy gravy", "image": "egg_curry.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Small", "Medium", "Large"]},
-        {"name": "Gulab Jamun", "description": "Sweet milk dumplings", "image": "gulab_jamun.jpg", "spice_levels": [], "portion_sizes": ["Single", "Double"]},
-        {"name": "Mutton Biryani", "description": "Fragrant rice with mutton", "image": "mutton_biryani.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Small", "Medium", "Large"]},
-        {"name": "Fish Fry", "description": "Crispy fried fish", "image": "fish_fry.jpg", "spice_levels": ["Mild", "Medium"], "portion_sizes": ["Single", "Double"]},
-        {"name": "Prawns Curry", "description": "Spicy prawn curry", "image": "prawns_curry.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Small", "Medium", "Large"]},
-        {"name": "Malai Kofta", "description": "Rich and creamy kofta curry", "image": "malai_kofta.jpg", "spice_levels": ["Mild", "Medium"], "portion_sizes": ["Small", "Medium", "Large"]},
-        {"name": "Vegetable Soup", "description": "Healthy mixed vegetable soup", "image": "vegetable_soup.jpg", "spice_levels": [], "portion_sizes": ["Small", "Medium"]},
-        {"name": "Paneer Tikka", "description": "Grilled paneer skewers", "image": "paneer_tikka.jpg", "spice_levels": ["Mild", "Medium"], "portion_sizes": ["Single", "Double"]}
-    ],
-    "VEGAN": [
-        {"name": "Veg Biryani", "description": "Flavourful veg biryani", "image": "veg_biryani.jpg", "spice_levels": ["Mild", "Medium"], "portion_sizes": ["Small", "Medium", "Large"]},
-        {"name": "Dal Makhani", "description": "Rich lentil curry", "image": "dal_makhani.jpg", "spice_levels": ["Mild", "Medium"], "portion_sizes": ["Small", "Large"]}
+        {"name": "Paneer Butter Masala", "description": "Creamy paneer dish", "image": "paneer_butter.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Small", "Medium", "Large"]}
     ]
 }
 
@@ -37,59 +23,34 @@ def generate_qr(table_id):
     img = qr.make_image(fill="black", back_color="white")
     return img
 
-# Login / Signup
+# Login or Signup
+
+def login_or_signup(phone_number, name=None, email=None, password=None):
+    """
+    Handles login or sign-up for customers.
+
+    Args:
+        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).
 
-def login_or_signup(name, phone_number, email):
+    Returns:
+        str: Status message about login or sign-up.
+    """
+    # Check if phone number exists in the database
     if phone_number in CUSTOMERS:
-        return f"Welcome back, {CUSTOMERS[phone_number]['name']}!"
+        # 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:
-        CUSTOMERS[phone_number] = {"name": name, "email": email, "preferences": []}
-        return f"Account created for {name}. Please set your preferences."
-
-# 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(preference):
-    return MENU.get(preference.upper(), MENU["ALL"])
-
-# Menu Display as Images and Names
-
-def display_menu(preference):
-    dishes = filter_menu(preference)
-    display_items = []
-    for dish in dishes:
-        display_items.append({"label": dish["name"], "value": dish["image"]})
-    return display_items
-
-# 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"]
-    return "Dish not found.", None
-
-# Recommendations
-
-def recommend_dishes(selected_dish):
-    recommendations = {
-        "Chicken Curry": ["Rotis", "Rice"],
-        "Paneer Butter Masala": ["Naan", "Rice"],
-    }
-    return recommendations.get(selected_dish, [])
+        # New customer, proceed with sign-up
+        if not all([name, email, password]):
+            return "Missing details for sign-up. Please provide Name, Email, and Password."
+        CUSTOMERS[phone_number] = {"name": name, "email": email, "password": password}
+        return f"Account created successfully for {name}!"
 
 # Gradio Interfaces
 
@@ -106,42 +67,19 @@ def main():
 
         # Login or Signup
         with gr.Tab("Login / Signup"):
-            name_input = gr.Textbox(label="Name", placeholder="Enter your name")
-            phone_input = gr.Textbox(label="Phone Number", placeholder="Enter phone number")
-            email_input = gr.Textbox(label="Email", placeholder="Enter email")
-            login_button = gr.Button("Login / Signup")
-            login_output = gr.Text(label="Login Status")
-            login_button.click(login_or_signup, inputs=[name_input, phone_input, email_input], outputs=login_output)
-
-        # Preferences
-        with gr.Tab("Preferences"):
-            phone_input_pref = gr.Textbox(label="Phone Number", placeholder="Enter phone number")
-            preferences = gr.CheckboxGroup(["Vegan", "Halal", "Indian", "Chinese"], label="Select Preferences")
-            pref_button = gr.Button("Save Preferences")
-            pref_output = gr.Text(label="Preference Status")
-            pref_button.click(save_preferences, inputs=[phone_input_pref, preferences], outputs=pref_output)
-
-        # Menu Display
-        with gr.Tab("Menu Display"):
-            menu_pref = gr.Dropdown(["ALL", "VEGAN", "HALAL"], label="Menu Type")
-            menu_button = gr.Button("Show Menu")
-            menu_output = gr.Gallery(label="Filtered Menu")
-            menu_button.click(display_menu, inputs=menu_pref, outputs=menu_output)
-
-        # Dish Details
-        with gr.Tab("Dish Details"):
-            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])
-
-        # Recommendations
-        with gr.Tab("Recommendations"):
-            selected_dish = gr.Textbox(label="Selected Dish", placeholder="Enter dish name")
-            rec_button = gr.Button("Get Recommendations")
-            rec_output = gr.List(label="Recommendations")
-            rec_button.click(recommend_dishes, inputs=selected_dish, outputs=rec_output)
+            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")
+            password_input = gr.Textbox(label="Password", placeholder="Enter your password", type="password")
+
+            action_button = gr.Button("Submit")
+            result_output = gr.Text(label="Status")
+
+            action_button.click(
+                login_or_signup,
+                inputs=[phone_input, name_input, email_input, password_input],
+                outputs=result_output
+            )
 
     demo.launch()