Update app.py

app.py

@@ -3,7 +3,7 @@ import qrcode
 from PIL import Image
 
 # Sample Data
-CUSTOMERS = {"9876543210": {"name": "John Doe", "preferences": ["Halal", "Indian"]}}
+CUSTOMERS = {}
 MENU = {
     "ALL": [
         {"name": "Chicken Curry", "description": "Spicy chicken curry", "image": "chicken_curry.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Small", "Medium", "Large"]},
@@ -25,11 +25,12 @@ def generate_qr(table_id):
 
 # Login / Signup
 
-def login_or_signup(phone_number):
+def login_or_signup(name, phone_number, email):
     if phone_number in CUSTOMERS:
         return f"Welcome back, {CUSTOMERS[phone_number]['name']}!", True
     else:
-        return "Please sign up to continue.", False
+        CUSTOMERS[phone_number] = {"name": name, "email": email, "preferences": []}
+        return f"Account created for {name}. Please set your preferences.", False
 
 # Save Preferences
 
@@ -41,9 +42,8 @@ def save_preferences(phone_number, preferences):
 
 # Filter Menu
 
-def filter_menu(preferences):
-    filtered_menu = MENU.get(preferences[0].upper(), MENU["ALL"])
-    return filtered_menu
+def filter_menu(preference):
+    return MENU.get(preference.upper(), MENU["ALL"])
 
 # Dish Details
 
@@ -57,6 +57,16 @@ def dish_details(dish_name):
                     "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, [])
 
 # Gradio Interfaces
 
@@ -65,41 +75,52 @@ def main():
         gr.Markdown("# Restaurant QR Code Ordering System")
 
         # QR Code Generator
-        with gr.Row():
+        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.Row():
+        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=phone_input, outputs=login_output)
+            existing_customer = gr.Text(label="Is Existing Customer?")
+            login_button.click(login_or_signup, inputs=[name_input, phone_input, email_input], outputs=[login_output, existing_customer])
 
         # Preferences
-        with gr.Row():
-            preferences = gr.CheckboxGroup(["Vegan", "Halal", "Indian", "Chinese"], label="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, preferences], outputs=pref_output)
+            pref_button.click(save_preferences, inputs=[phone_input_pref, preferences], outputs=pref_output)
 
         # Menu Display
-        with gr.Row():
+        with gr.Tab("Menu Display"):
             menu_pref = gr.Dropdown(["ALL", "VEGAN", "HALAL"], label="Menu Type")
             menu_button = gr.Button("Show Menu")
-            menu_output = gr.JSON(label="Menu")
+            menu_output = gr.JSON(label="Filtered Menu")
             menu_button.click(filter_menu, inputs=menu_pref, outputs=menu_output)
 
         # Dish Details
-        with gr.Row():
+        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)
+
     demo.launch()
 
 if __name__ == "__main__":