Update app.py

app.py

@@ -6,10 +6,9 @@ def load_menu_data():
     try:
         with open("menu.json", "r") as file:
             data = json.load(file)
-        return data.get("menu", [])
+        return data["menu"]
     except Exception as e:
-        print(f"Error loading menu: {e}")
-        return []
+        return f"Error loading menu: {e}"
 
 menu_data = load_menu_data()
 
@@ -17,21 +16,16 @@ menu_data = load_menu_data()
 current_order = []
 preferences = {}
 
-
 # Functions
 def login_user(name, phone):
-    if not name.strip():  # Ensure name is not empty
-        return "Name cannot be empty. Please enter your name.", gr.Tabs.update(selected=0)
-    if len(phone) != 10 or not phone.isdigit():  # Ensure phone number is valid
-        return "Invalid phone number. Please enter a valid 10-digit number.", gr.Tabs.update(selected=0)
-    return f"Welcome, {name}! Please proceed to set your preferences.", gr.Tabs.update(selected=1)
-
+    if len(phone) != 10 or not phone.isdigit():
+        return "Invalid phone number. Please enter a 10-digit number."
+    return f"Welcome, {name}! Please proceed to set your preferences."
 
 def save_preferences(pref_type, allergies):
     preferences["type"] = pref_type
     preferences["allergies"] = allergies
-    return "Preferences saved successfully! Proceed to the menu.", gr.Tabs.update(selected=2)
-
+    return "Preferences saved successfully! Proceed to the menu."
 
 def get_filtered_menu():
     if not preferences.get("type"):
@@ -42,101 +36,83 @@ def get_filtered_menu():
     ]
     return filtered_menu
 
+def generate_menu_buttons():
+    menu_items = get_filtered_menu()
+    if isinstance(menu_items, str):
+        return [gr.Label(value=menu_items)]
+    
+    buttons = []
+    for item in menu_items:
+        buttons.append(
+            gr.Button(
+                value=f"{item['name']} - ₹{item['price']}",
+                elem_id=f"menu_item_{item['id']}",
+                variant="primary"
+            )
+        )
+    return buttons
 
 def add_to_order(item_name):
     current_order.append(item_name)
     return "\n".join(current_order)
 
-
 def place_order():
     if not current_order:
-        return "No items in the order. Please add items to your order before placing it.", gr.Tabs.update(selected=2)
-    return f"Order placed successfully! Items: {', '.join(current_order)}.", gr.Tabs.update(selected=4)
-
-
-def load_menu():
-    filtered_menu = get_filtered_menu()
-    if isinstance(filtered_menu, str):  # Check for error message
-        return gr.update(value=filtered_menu)
-    return [
-        f"{item['name']} - ₹{item['price']}" for item in filtered_menu
-    ]
-
+        return "No items in the order. Please add items to your order before placing it."
+    return f"Order placed successfully! Items: {', '.join(current_order)}."
 
 # App Interface
 with gr.Blocks() as app:
-    # Tabs for navigation
-    with gr.Tabs() as tabs:
-        # Login Screen
-        with gr.Tab("Login"):
-            with gr.Row():
-                name = gr.Textbox(label="Name", placeholder="Enter your name")
-                phone = gr.Textbox(label="Phone", placeholder="Enter your 10-digit phone number")
-            login_button = gr.Button("Login")
-            login_output = gr.Label()
-            login_button.click(
-                login_user,
-                inputs=[name, phone],
-                outputs=[login_output, tabs]
-            )
-
-        # Preferences Screen
-        with gr.Tab("Preferences"):
-            pref_type = gr.Radio(
-                ["Vegetarian", "Non-Vegetarian", "Complete Menu"],
-                label="Select Preference"
-            )
-            allergies = gr.CheckboxGroup(
-                ["Peanuts", "Cheese", "Pork", "Custom"],
-                label="Allergies (if any)"
+    # Login Screen
+    with gr.Tab("Login"):
+        name = gr.Textbox(label="Name", placeholder="Enter your name")
+        phone = gr.Textbox(label="Phone", placeholder="Enter your 10-digit phone number")
+        login_button = gr.Button("Login")
+        login_output = gr.Label()
+        login_button.click(login_user, inputs=[name, phone], outputs=login_output)
+
+    # Preferences Screen
+    with gr.Tab("Preferences"):
+        pref_type = gr.Radio(
+            ["Vegetarian", "Non-Vegetarian", "Complete Menu"],
+            label="Select Preference"
+        )
+        allergies = gr.CheckboxGroup(
+            ["Peanuts", "Cheese", "Pork", "Custom"],
+            label="Allergies (if any)"
+        )
+        save_pref_button = gr.Button("Save Preferences")
+        pref_output = gr.Label()
+        save_pref_button.click(save_preferences, inputs=[pref_type, allergies], outputs=pref_output)
+
+    # Menu Screen
+    with gr.Tab("Menu"):
+        menu_buttons = gr.Column(elem_id="menu_buttons")
+        current_order_display = gr.Textbox(label="Current Order")
+        proceed_to_summary = gr.Button("Proceed to Order Summary")
+        
+        # Dynamically generate menu buttons after saving preferences
+        save_pref_button.click(generate_menu_buttons, outputs=menu_buttons)
+        
+        # Handle adding items to order
+        for item in menu_data:
+            menu_buttons.append(
+                gr.Button(
+                    value=f"{item['name']} - ₹{item['price']}",
+                    elem_id=f"item-{item['id']}"
+                ).click(
+                    add_to_order,
+                    inputs=None,
+                    outputs=current_order_display
+                )
             )
-            save_pref_button = gr.Button("Save Preferences")
-            pref_output = gr.Label()
-            save_pref_button.click(
-                save_preferences,
-                inputs=[pref_type, allergies],
-                outputs=[pref_output, tabs]
-            )
-
-        # Menu Screen
-        with gr.Tab("Menu"):
-            menu_list = gr.Textbox(label="Menu Items", interactive=False, lines=10)
-            load_menu_button = gr.Button("Load Menu")
-            load_menu_button.click(
-                lambda: "\n".join(load_menu()),
-                inputs=None,
-                outputs=menu_list
-            )
-
-            current_order_display = gr.Textbox(label="Current Order", interactive=False)
-            item_name = gr.Textbox(label="Enter Item Name to Add")
-            add_item_button = gr.Button("Add to Order")
-            add_item_button.click(
-                add_to_order,
-                inputs=item_name,
-                outputs=current_order_display
-            )
-
-            proceed_to_summary = gr.Button("Proceed to Order Summary")
-            proceed_to_summary.click(
-                lambda: gr.Tabs.update(selected=3),
-                inputs=None,
-                outputs=[tabs]
-            )
-
-        # Order Summary Screen
-        with gr.Tab("Order Summary"):
-            order_summary = gr.Textbox(label="Your Order", interactive=False)
-            place_order_button = gr.Button("Place Order")
-            order_output = gr.Label()
-            place_order_button.click(
-                place_order,
-                inputs=None,
-                outputs=[order_output, tabs]
-            )
-
-        # Thank You Screen
-        with gr.Tab("Thank You"):
-            thank_you_message = gr.Label("Thank you for placing your order!")
-
+    
+    # Order Summary Screen
+    with gr.Tab("Order Summary"):
+        order_summary = gr.Textbox(label="Your Order", interactive=False)
+        place_order_button = gr.Button("Place Order")
+        order_output = gr.Label()
+        place_order_button.click(place_order, inputs=None, outputs=order_output)
+
+# Launch App
 app.launch()